Fix test

statelesshz · Feb 14, 2024 · 79016eb · 79016eb
1 parent 164193f
commit 79016eb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/accelerate/test_utils/scripts/external_deps/test_metrics.py b/src/accelerate/test_utils/scripts/external_deps/test_metrics.py
@@ -82,7 +82,7 @@ def collate_fn(examples):
 
 def get_mrpc_setup(dispatch_batches, split_batches):
     dataloader_config = DataLoaderConfiguration(dispatch_batches=dispatch_batches, split_batches=split_batches)
-    accelerator = Accelerator(data_loader_config=dataloader_config)
+    accelerator = Accelerator(dataloader_config=dataloader_config)
     dataloader = get_dataloader(accelerator, not dispatch_batches)
     model = AutoModelForSequenceClassification.from_pretrained(
         "hf-internal-testing/mrpc-bert-base-cased", return_dict=True

diff --git a/tests/test_multigpu.py b/tests/test_multigpu.py
@@ -41,7 +41,9 @@ 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", "external_deps", "test_pippy.py"])
+        self.pippy_file_path = os.path.sep.join(
+            mod_file.split(os.path.sep)[:-1] + ["scripts", "external_deps", "test_pippy.py"]
+        )
 
     @require_multi_device
     def test_multi_device(self):