update: disable coverage

kozistr · Jul 14, 2024 · 4662970 · 4662970
1 parent 9baf82d
commit 4662970
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytorch_optimizer/optimizer/utils.py b/pytorch_optimizer/optimizer/utils.py
@@ -15,11 +15,11 @@
 
 HAS_TRANSFORMERS: bool = find_spec('transformers') is not None
 
-if HAS_TRANSFORMERS:
+if HAS_TRANSFORMERS:  # pragma: no cover
     try:
         from transformers.integrations.deepspeed import is_deepspeed_zero3_enabled
     except ImportError:
-        from transformers.deepspeed import is_deepspeed_zero3_enabled  # pragma: no cover
+        from transformers.deepspeed import is_deepspeed_zero3_enabled
 else:
 
     def is_deepspeed_zero3_enabled() -> bool: