Fix test

pytorch · Jan 10, 2025 · f93e330 · f93e330
1 parent b1616f0
commit f93e330
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_transforms.py b/test/test_transforms.py
@@ -12703,7 +12703,7 @@ def test_transform_inverse(self):
     )
     def test_reward_spec(
         self,
-        weights: list[float] | None,
+        weights,
         reward_spec: TensorSpec,
         expected_spec: TensorSpec,
     ) -> None: