We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c9d8c70 commit af92441Copy full SHA for af92441
tests/trainer/test_trainer.py
@@ -2398,7 +2398,7 @@ def test_bnb_adam8bit_no_bnb(self):
2398
2399
# Pretend that bnb does not exist, even if installed. By setting bnb to None, importing
2400
# bnb will fail even if bnb is installed.
2401
- with patch.dict("sys.modules", {"bnb.optim": None}):
+ with patch.dict("sys.modules", {"bitsandbytes.optim": None}):
2402
with self.assertRaises(ValueError):
2403
Trainer.get_optimizer_cls_and_kwargs(args)
2404
0 commit comments