wip

mivanit · Jun 18, 2024 · 542a439 · 542a439
1 parent 6a81a6d
commit 542a439
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 9 deletions.
diff --git a/tests/unit/json_serialize/serializable_dataclass/test_sdc_defaults.py b/tests/unit/json_serialize/serializable_dataclass/test_sdc_defaults.py
@@ -14,12 +14,12 @@
 
 # pylint: disable=missing-class-docstring
 
-BELOW_PY_3_9: bool = sys.version_info < (3, 9)
+BELOW_PY_3_10: bool = sys.version_info < (3, 10)
 
 
 def _loading_test_wrapper(cls, data, assert_record_len: int | None = None) -> Any:
     """wrapper for testing the load function, which accounts for version differences"""
-    if BELOW_PY_3_9:
+    if BELOW_PY_3_10:
         with pytest.warns(UserWarning) as record:
             loaded = cls.load(data)
         print([x.message for x in record])

diff --git a/tests/unit/json_serialize/serializable_dataclass/test_sdc_properties_nested.py b/tests/unit/json_serialize/serializable_dataclass/test_sdc_properties_nested.py
@@ -11,12 +11,12 @@
 
 print(f"{SUPPORS_KW_ONLY = }")
 
-BELOW_PY_3_9: bool = sys.version_info < (3, 9)
+BELOW_PY_3_10: bool = sys.version_info < (3, 10)
 
 
 def _loading_test_wrapper(cls, data, assert_record_len: int | None = None) -> Any:
     """wrapper for testing the load function, which accounts for version differences"""
-    if BELOW_PY_3_9:
+    if BELOW_PY_3_10:
         with pytest.warns(UserWarning) as record:
             loaded = cls.load(data)
         print([x.message for x in record])

diff --git a/tests/unit/json_serialize/serializable_dataclass/test_serializable_dataclass.py b/tests/unit/json_serialize/serializable_dataclass/test_serializable_dataclass.py
@@ -14,12 +14,12 @@
 # pylint: disable=missing-class-docstring, unused-variable
 
 
-BELOW_PY_3_9: bool = sys.version_info < (3, 9)
+BELOW_PY_3_10: bool = sys.version_info < (3, 10)
 
 
 def _loading_test_wrapper(cls, data, assert_record_len: int | None = None) -> Any:
     """wrapper for testing the load function, which accounts for version differences"""
-    if BELOW_PY_3_9:
+    if BELOW_PY_3_10:
         with pytest.warns(UserWarning) as record:
             loaded = cls.load(data)
         print([x.message for x in record])

diff --git a/tests/unit/test_mlutils.py b/tests/unit/test_mlutils.py
@@ -24,7 +24,7 @@ def test_get_checkpoint_paths_for_run():
     assert checkpoint_paths == [(123, checkpoint1_path), (456, checkpoint2_path)]
 
 
-BELOW_PY_3_9: bool = sys.version_info < (3, 9)
+BELOW_PY_3_10: bool = sys.version_info < (3, 9)
 
 
 def test_register_method():
@@ -51,14 +51,14 @@ class TestEvalsB:
             def other_eval_function():
                 pass
 
-    if BELOW_PY_3_9:
+    if BELOW_PY_3_10:
         assert len(record) == 2
     else:
         assert len(record) == 0
 
     evalsA = TestEvalsA.evals
     evalsB = TestEvalsB.evals
-    if BELOW_PY_3_9:
+    if BELOW_PY_3_10:
         assert len(evalsA) == 1
         assert len(evalsB) == 1
     else: