From a06973b3767d6d2a771397458baa56d2ae23291f Mon Sep 17 00:00:00 2001 From: Quentin Anthony Date: Tue, 8 Oct 2024 12:26:48 -0700 Subject: [PATCH] precommit --- tests/neox_args/test_neoxargs_usage.py | 4 +++- tests/unit/test_format_conversion_scripts.py | 1 + 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/neox_args/test_neoxargs_usage.py b/tests/neox_args/test_neoxargs_usage.py index 9c709d918..5f8ba7bd2 100644 --- 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 index 32e0357b2..6935e480a 100644 --- 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." )