diff --git a/tests/metrics/test_randomization_metrics.py b/tests/metrics/test_randomization_metrics.py index db266fe0..8375277a 100644 --- a/tests/metrics/test_randomization_metrics.py +++ b/tests/metrics/test_randomization_metrics.py @@ -21,6 +21,7 @@ def parameter_randomization_test(model, request): assert torch.norm(param1.data - param2.data) > 1e3 # norm of the difference in parameters should be significant +@pytest.mark.parametrize() def model_randomization_test(): assert torch.__version__ == "2.0.0" gen = torch.Generator()