Skip to content

Commit af92441

Browse files
younesbelkadaamyeroberts
authored andcommitted
small change (huggingface#18584)
1 parent c9d8c70 commit af92441

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/trainer/test_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2398,7 +2398,7 @@ def test_bnb_adam8bit_no_bnb(self):
23982398

23992399
# Pretend that bnb does not exist, even if installed. By setting bnb to None, importing
24002400
# bnb will fail even if bnb is installed.
2401-
with patch.dict("sys.modules", {"bnb.optim": None}):
2401+
with patch.dict("sys.modules", {"bitsandbytes.optim": None}):
24022402
with self.assertRaises(ValueError):
24032403
Trainer.get_optimizer_cls_and_kwargs(args)
24042404

0 commit comments

Comments
 (0)