pc

mosaicml · Jun 19, 2024 · 0ffd73b · 0ffd73b
1 parent dc7169a
commit 0ffd73b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/llmfoundry/utils/huggingface_hub_utils.py b/llmfoundry/utils/huggingface_hub_utils.py
@@ -278,8 +278,7 @@ def edit_files_for_hf_compatibility(
     }
     # Filter out __init__ files
     all_relative_imports = {
-        relative_import
-        for relative_import in all_relative_imports
+        relative_import for relative_import in all_relative_imports
         if relative_import != '__init__'
     }
     for entrypoint in entrypoint_files: