precommit

EleutherAI · Oct 8, 2024 · a06973b · a06973b
1 parent 3bb9128
commit a06973b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/tests/neox_args/test_neoxargs_usage.py b/tests/neox_args/test_neoxargs_usage.py
@@ -66,7 +66,9 @@ def test_neoxargs_usage():
 
         # find args matches
         matches = list(
-            re.findall(r"(?<=neox_args\.).{2,}?(?=[\s\n(){}+-/*;:,=,[,\]])", file_contents)
+            re.findall(
+                r"(?<=neox_args\.).{2,}?(?=[\s\n(){}+-/*;:,=,[,\]])", file_contents
+            )
         )
         if len(matches) == 0:
             continue

diff --git a/tests/unit/test_format_conversion_scripts.py b/tests/unit/test_format_conversion_scripts.py
@@ -3,6 +3,7 @@
 from tests.common import simulate_deepy_env, save_random_model
 from megatron.neox_arguments.neox_args import NeoXArgsTokenizer
 
+
 @pytest.mark.skip(
     reason="Conversion test is skipped until we fix the CUDA + torch multiprocessing issue."
 )