TestMap.fn

pytorch · Nov 22, 2023 · 8ca93b6 · 8ca93b6
1 parent 152612c
commit 8ca93b6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_tensordict.py b/test/test_tensordict.py
@@ -6574,15 +6574,15 @@ def test_map_seed(self):
         # making sure that no worker does more than any other.
         generator.manual_seed(0)
         td_out_0 = td.map(
-            self.get_rand_incr,
+            TestMap.get_rand_incr,
             num_workers=4,
             generator=generator,
             chunksize=1,
             max_tasks_per_child=5,
         )
         generator.manual_seed(0)
         td_out_1 = td.map(
-            self.get_rand_incr,
+            TestMap.get_rand_incr,
             num_workers=4,
             generator=generator,
             chunksize=1,