Skip to content

Commit 9c260b1

Browse files
committed
update: disable coverage from certain line
1 parent 98186fc commit 9c260b1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch_optimizer/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -245,7 +245,7 @@ def load_optimizer(optimizer: str) -> OPTIMIZER:
245245

246246
if optimizer.startswith('bnb'):
247247
if HAS_BNB and torch.cuda.is_available():
248-
return load_bnb_optimizer(optimizer)
248+
return load_bnb_optimizer(optimizer) # pragma: no cover
249249
raise ImportError(f'[-] bitsandbytes and CUDA required for bnb optimizers : {optimizer}')
250250
if optimizer not in OPTIMIZERS:
251251
raise NotImplementedError(f'[-] not implemented optimizer : {optimizer}')

0 commit comments

Comments
 (0)