fix code quanlity check

Signed-off-by: weimingc <17592131+meenchen@users.noreply.github.com>

Author: meenchen

modelopt/torch/utils/dataset_utils.py

@@ -19,7 +19,7 @@
 import json
 import os
 from collections.abc import Callable, Iterator
-from contextlib import contextmanager
+from contextlib import contextmanager, suppress
 from pathlib import Path
 from typing import TYPE_CHECKING, Any
 from warnings import warn
@@ -464,10 +464,8 @@ def _disable_use_cache(model: torch.nn.Module) -> Iterator[None]:
         if had_attr:
             config.use_cache = prev
         else:
-            try:
+            with suppress(AttributeError):
                 delattr(config, "use_cache")
-            except AttributeError:
-                pass
 
 
 def get_max_batch_size(

tests/unit/torch/utils/test_dataset_utils.py

@@ -195,10 +195,9 @@ def test_disable_use_cache_restores_on_exception():
     model.config = _Config()
     model.config.use_cache = True
 
-    with pytest.raises(RuntimeError, match="boom"):
-        with _disable_use_cache(model):
-            assert model.config.use_cache is False
-            raise RuntimeError("boom")
+    with pytest.raises(RuntimeError, match="boom"), _disable_use_cache(model):
+        assert model.config.use_cache is False
+        raise RuntimeError("boom")
 
     assert model.config.use_cache is True