remove random from tests

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>

Author: kylesayrs

tests/test_transform/factory/test_correctness.py

@@ -25,7 +25,7 @@
 from tests.testing_utils import requires_accelerate, requires_gpu
 
 
-@pytest.mark.parametrize("type", TransformFactory.registered_names())
+@pytest.mark.parametrize("type", ("hadamard", "random-hadamard"))
 @pytest.mark.parametrize("randomized", (True, False))
 def test_correctness_linear(type, randomized):
     size = (4, 8)
@@ -54,7 +54,7 @@ def test_correctness_linear(type, randomized):
     assert torch.allclose(true_output, output, atol=1e-5, rtol=0.0)
 
 
-@pytest.mark.parametrize("type", TransformFactory.registered_names())
+@pytest.mark.parametrize("type", ("hadamard", "random-hadamard"))
 @pytest.mark.parametrize("randomized", (True, False))
 def test_correctness_model(type, randomized, model_apply, offload=False):
     # load model
@@ -83,7 +83,7 @@ def test_correctness_model(type, randomized, model_apply, offload=False):
 
 @requires_gpu
 @requires_accelerate()
-@pytest.mark.parametrize("type", TransformFactory.registered_names())
+@pytest.mark.parametrize("type", ("hadamard", "random-hadamard"))
 @pytest.mark.parametrize("randomized", (True, False))
 def test_correctness_model_offload(type, randomized, model_apply):
     test_correctness_model(type, randomized, model_apply, offload=True)

tests/test_transform/factory/test_memory.py

@@ -20,7 +20,6 @@
     TransformArgs,
     TransformBase,
     TransformConfig,
-    TransformFactory,
     TransformScheme,
     apply_transform_config,
 )
@@ -29,7 +28,7 @@
 from tests.testing_utils import requires_accelerate, requires_gpu
 
 
-@pytest.mark.parametrize("type", TransformFactory.registered_names())
+@pytest.mark.parametrize("type", ("hadamard", "random-hadamard"))
 @pytest.mark.parametrize("randomized", (True, False))
 @pytest.mark.parametrize("requires_grad", (True, False))
 def test_memory_sharing(type, randomized, requires_grad, offload=False):
@@ -84,7 +83,7 @@ def test_memory_sharing(type, randomized, requires_grad, offload=False):
 
 @requires_gpu
 @requires_accelerate()
-@pytest.mark.parametrize("type", TransformFactory.registered_names())
+@pytest.mark.parametrize("type", ("hadamard", "random-hadamard"))
 @pytest.mark.parametrize("randomized", (True, False))
 def test_memory_sharing_offload(
     type,