syntax

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>
NVIDIA · Aug 27, 2024 · 3ce7df8 · 3ce7df8
1 parent 0176b31
commit 3ce7df8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/collections/llm/gpt/model/test_model_import.py b/tests/collections/llm/gpt/model/test_model_import.py
@@ -80,7 +80,7 @@ def import_from_hf(config_name, hf_path):
     import_ckpt(model=model, source=hf_path)
 
 if __name__ == '__main__':
-for config_name, hf_id in config_name_to_hf_id.items():
-    src = f'hf:///mnt/datadrive/TestData/nemo2_ckpt/{config_name}'
-    import_from_hf(config_name, src)
+    for config_name, hf_id in config_name_to_hf_id.items():
+        src = f'hf:///mnt/datadrive/TestData/nemo2_ckpt/{config_name}'
+        import_from_hf(config_name, src)