fix typo, change class, remove long test case

Author: dsikka

src/compressed_tensors/transforms/base.py

@@ -27,10 +27,10 @@
 # first or second matirx in torch.matmul depending on dimensions, can be inferred
 # by the layer time likely.
 
-MATIRX_TRANSFORMS = ["matrix-mul", "hadamard", "random-hadamard"]
+MATRIX_TRANSFORMS = ["matrix-mul", "hadamard", "random-hadamard"]
 
 
-class Transforms(RegistryMixin):
+class Transforms(torch.nn.Parameter, RegistryMixin):
     def __new__(
         cls,
         transform: torch.Tensor,
@@ -69,7 +69,7 @@ def __new__(
 
     @classmethod
     def fetch_apply(cls, name: str):
-        if name in MATIRX_TRANSFORMS:
+        if name in MATRIX_TRANSFORMS:
             return apply_matrix_transform
         raise NotImplementedError("Only matrix transforms are supported")
 

tests/test_transforms/test_hadamards.py

@@ -51,7 +51,7 @@ def test_random_hadamard_matrix_compliant(size):
 
 @pytest.mark.parametrize(
     "size",
-    [1024, 2048],
+    [1024],
 )
 def test_deterministic_hadamard_compliant(size):
     had_matrix = deterministic_hadamard_matrix(size)