Skip to content

Commit e8050a0

Browse files
committed
ruff format
Signed-off-by: Ananth Subramaniam <ansubramania@nvidia.com>
1 parent 9003803 commit e8050a0

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

‎tests/unit_tests/utils/test_import_utils.py‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)