small change (#18584)

2022-08-12 20:04:38 +02:00
parent b3ff7c680c
commit 1ccd2515ed
1 changed files with 1 additions and 1 deletions
--- a/tests/trainer/test_trainer.py
+++ b/tests/trainer/test_trainer.py
@@ -2398,7 +2398,7 @@ class TrainerOptimizerChoiceTest(unittest.TestCase):

        # 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)