small change (huggingface#18584)

oneraghavan · Sep 26, 2022 · 1bcdadd · 1bcdadd
1 parent fc58b3b
commit 1bcdadd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py
@@ -2398,7 +2398,7 @@ def test_bnb_adam8bit_no_bnb(self):
 
         # Pretend that bnb does not exist, even if installed. By setting bnb to None, importing
         # bnb will fail even if bnb is installed.
-        with patch.dict("sys.modules", {"bnb.optim": None}):
+        with patch.dict("sys.modules", {"bitsandbytes.optim": None}):
             with self.assertRaises(ValueError):
                 Trainer.get_optimizer_cls_and_kwargs(args)