@@ -4427,7 +4427,7 @@ def test_bnb_adam8bit_no_bnb(self):
44274427 args = TrainingArguments (optim = OptimizerNames .ADAMW_BNB , output_dir = "None" )
44284428
44294429 # Pretend that bnb does not exist, even if installed. By setting bnb to None, importing
4430- # bnb will fail even if bnb is installed.
4430+ # bnb will fail even if `bitsandbytes` is installed.
44314431 with patch .dict ("sys.modules" , {"bitsandbytes.optim" : None }):
44324432 with self .assertRaises (ValueError ):
44334433 Trainer .get_optimizer_cls_and_kwargs (args )
@@ -4436,7 +4436,7 @@ def test_bnb_paged_adam_no_bnb(self):
44364436 args = TrainingArguments (optim = OptimizerNames .PAGED_ADAMW , output_dir = "None" )
44374437
44384438 # Pretend that bnb does not exist, even if installed. By setting bnb to None, importing
4439- # bnb will fail even if bnb is installed.
4439+ # bnb will fail even if `bitsandbytes` is installed.
44404440 with patch .dict ("sys.modules" , {"bitsandbytes.optim" : None }):
44414441 with self .assertRaises (ValueError ):
44424442 Trainer .get_optimizer_cls_and_kwargs (args )
@@ -4445,7 +4445,7 @@ def test_bnb_paged_adam8bit_no_bnb(self):
44454445 args = TrainingArguments (optim = OptimizerNames .PAGED_ADAMW_8BIT , output_dir = "None" )
44464446
44474447 # Pretend that bnb does not exist, even if installed. By setting bnb to None, importing
4448- # bnb will fail even if bnb is installed.
4448+ # bnb will fail even if `bitsandbytes` is installed.
44494449 with patch .dict ("sys.modules" , {"bitsandbytes.optim" : None }):
44504450 with self .assertRaises (ValueError ):
44514451 Trainer .get_optimizer_cls_and_kwargs (args )
@@ -4454,7 +4454,7 @@ def test_bnb_paged_lion_no_bnb(self):
44544454 args = TrainingArguments (optim = OptimizerNames .PAGED_LION , output_dir = "None" )
44554455
44564456 # Pretend that bnb does not exist, even if installed. By setting bnb to None, importing
4457- # bnb will fail even if bnb is installed.
4457+ # bnb will fail even if `bitsandbytes` is installed.
44584458 with patch .dict ("sys.modules" , {"bitsandbytes.optim" : None }):
44594459 with self .assertRaises (ValueError ):
44604460 Trainer .get_optimizer_cls_and_kwargs (args )
@@ -4463,7 +4463,7 @@ def test_bnb_paged_lion8bit_no_bnb(self):
44634463 args = TrainingArguments (optim = OptimizerNames .PAGED_LION_8BIT , output_dir = "None" )
44644464
44654465 # Pretend that bnb does not exist, even if installed. By setting bnb to None, importing
4466- # bnb will fail even if bnb is installed.
4466+ # bnb will fail even if `bitsandbytes` is installed.
44674467 with patch .dict ("sys.modules" , {"bitsandbytes.optim" : None }):
44684468 with self .assertRaises (ValueError ):
44694469 Trainer .get_optimizer_cls_and_kwargs (args )
0 commit comments