Skip to content

Commit 34e3f97

Browse files
Fix test
1 parent 6bb694b commit 34e3f97

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_mixed.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -660,7 +660,7 @@ def test_modules_to_save(self):
660660
# TODO: theoretically, we could allow this if it's the same target layer
661661
config1 = LoHaConfig(target_modules=["lin0"], modules_to_save=["lin1"])
662662
peft_model.add_adapter("adapter1", config1)
663-
with pytest.raises(ValueError, match="Only one adapter can be set at a time for modules_to_save"):
663+
with pytest.raises(ValueError, match="Only one adapter can be set at a time for ModulesToSaveWrapper"):
664664
peft_model.set_adapter(["adapter0", "adapter1"])
665665

666666
def test_get_nb_trainable_parameters(self):

0 commit comments

Comments
 (0)