Include pippy_file_path (#2444)

huggingface · Feb 14, 2024 · d7de8d1 · d7de8d1
1 parent b443be7
commit d7de8d1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_multigpu.py b/tests/test_multigpu.py
@@ -41,6 +41,7 @@ def setUp(self):
             mod_file.split(os.path.sep)[:-1] + ["scripts", "test_distributed_data_loop.py"]
         )
         self.operation_file_path = os.path.sep.join(mod_file.split(os.path.sep)[:-1] + ["scripts", "test_ops.py"])
+        self.pippy_file_path = os.path.sep.join(mod_file.split(os.path.sep)[:-1] + ["scripts", "test_pippy.py"])
 
     @require_multi_device
     def test_multi_device(self):