@@ -249,7 +249,7 @@ def test_import_from_with_alternative(self):
249249 def test_fallback_module (self ):
250250 """Test safe_import_from with a fallback module."""
251251 # First import fails, but fallback succeeds
252- with patch (' importlib.import_module' ) as mock_import :
252+ with patch (" importlib.import_module" ) as mock_import :
253253 # Mock the first import to fail as AttributeError
254254 def side_effect (name ):
255255 if name == "primary_module" :
@@ -346,25 +346,25 @@ def test_get_torch_version_returns_version(self):
346346
347347 def test_get_torch_version_fallback (self ):
348348 """Test that get_torch_version falls back to '0.0.0' when torch.__version__ is unavailable."""
349- with patch (' torch.__version__' , side_effect = AttributeError ("No version attribute" )):
349+ with patch (" torch.__version__" , side_effect = AttributeError ("No version attribute" )):
350350 version = get_torch_version ()
351351 assert version == PkgVersion ("0.0.0" )
352352
353353 def test_is_torch_min_version_true (self ):
354354 """Test is_torch_min_version returns True when torch version meets minimum."""
355- with patch (' nemo_lm.utils.import_utils.get_torch_version' , return_value = PkgVersion ("2.0.0" )):
355+ with patch (" nemo_lm.utils.import_utils.get_torch_version" , return_value = PkgVersion ("2.0.0" )):
356356 assert is_torch_min_version ("1.9.0" ) is True
357357 assert is_torch_min_version ("2.0.0" ) is True
358358
359359 def test_is_torch_min_version_false (self ):
360360 """Test is_torch_min_version returns False when torch version is below minimum."""
361- with patch (' nemo_lm.utils.import_utils.get_torch_version' , return_value = PkgVersion ("1.8.0" )):
361+ with patch (" nemo_lm.utils.import_utils.get_torch_version" , return_value = PkgVersion ("1.8.0" )):
362362 assert is_torch_min_version ("1.9.0" ) is False
363363 assert is_torch_min_version ("2.0.0" ) is False
364364
365365 def test_is_torch_min_version_without_equality (self ):
366366 """Test is_torch_min_version with check_equality=False."""
367- with patch (' nemo_lm.utils.import_utils.get_torch_version' , return_value = PkgVersion ("2.0.0" )):
367+ with patch (" nemo_lm.utils.import_utils.get_torch_version" , return_value = PkgVersion ("2.0.0" )):
368368 assert is_torch_min_version ("1.9.0" , check_equality = False ) is True
369369 assert is_torch_min_version ("2.0.0" , check_equality = False ) is False
370370 assert is_torch_min_version ("2.1.0" , check_equality = False ) is False
0 commit comments