fix error with test_fit_does_not_overwrite_hyper_params

sktime · Jun 13, 2024 · 7c79b15 · 7c79b15
1 parent 18ba482
commit 7c79b15
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sktime/forecasting/hf_transformers_forecaster.py b/sktime/forecasting/hf_transformers_forecaster.py
@@ -276,7 +276,8 @@ def _fit(self, y, X, fh):
         elif self.fit_strategy == "peft":
             if _check_soft_dependencies("peft", severity="none"):
                 from peft import get_peft_model
-            self.model = get_peft_model(self.model, self.peft_config)
+            peft_config = deepcopy(self.peft_config)
+            self.model = get_peft_model(self.model, peft_config)
         else:
             raise ValueError("Unknown fit strategy")