formatted

Author: rithwik-db

tests/fixtures/fixtures.py

@@ -138,13 +138,15 @@ def uc_volume_read_only(request: pytest.FixtureRequest):
     else:
         return os.environ.get('UC_VOLUME_READ_ONLY_PATH', 'Volumes/main/regression_testing/composer_artifacts')
 
+
 @pytest.fixture
 def uc_volume_ephemeral(request: pytest.FixtureRequest):
     if request.node.get_closest_marker('remote') is None:
         return 'my-volume'
     else:
         return os.environ.get('UC_VOLUME_EPHEMERAL_PATH', 'Volumes/main/regression_testing/composer_ephemeral')
 
+
 ## MODEL HELPERS ##
 def causal_lm_model_helper(config):  # type: ignore
     transformers = pytest.importorskip('transformers')

tests/test_notebooks.py

@@ -137,11 +137,11 @@ def test_notebook(notebook: str, device: str, uc_volume_read_only: str):
 
     # Download CIFAR-10 data from UC Volumes
     uc_store = UCObjectStore(uc_volume_read_only)
-    files = uc_store.list_objects(prefix="CIFAR-10")
+    files = uc_store.list_objects(prefix='CIFAR-10')
 
     for file_path in files:
-        file_relative_path = file_path[len(f"/{uc_volume_read_only}/"):]
-        target = os.path.join(os.getcwd(), 'data', file_relative_path[len("CIFAR-10/"):])
+        file_relative_path = file_path[len(f'/{uc_volume_read_only}/'):]
+        target = os.path.join(os.getcwd(), 'data', file_relative_path[len('CIFAR-10/'):])
         if os.path.exists(target):
             continue
         os.makedirs(os.path.dirname(target), exist_ok=True)

tests/trainer/test_fsdp_checkpoint.py

@@ -953,6 +953,7 @@ def test_fsdp_partitioned_state_dict_load(
     trainer2.fit()
     trainer2.close()
 
+
 @pytest.mark.gpu
 @pytest.mark.remote
 @world_size(2)
@@ -1070,7 +1071,7 @@ def test_cleanup_sharded_checkpoints(
     tmp_path: pathlib.Path,
     num_ckpts_to_keep: int,
 ):
-    del world_size # not needed
+    del world_size  # not needed
     run_name = None
     batches_to_train = 3