Skip to content

Commit 65b4610

Browse files
committed
update: disable coverage from certain lines
1 parent f7ad37d commit 65b4610

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pytorch_optimizer/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@
115115
try:
116116
import bitsandbytes as bnb
117117

118-
HAS_BNB: bool = True
118+
HAS_BNB: bool = True # pragma: no cover
119119
except ImportError:
120120
HAS_BNB: bool = False
121121

@@ -219,7 +219,7 @@
219219
}
220220

221221

222-
def load_bnb_optimizer(optimizer: str) -> OPTIMIZER:
222+
def load_bnb_optimizer(optimizer: str) -> OPTIMIZER: # pragma: no cover
223223
r"""load bnb optimizer instance."""
224224
if 'sgd8bit' in optimizer:
225225
return bnb.optim.SGD8bit

0 commit comments

Comments
 (0)