Copy model_key_override when copying model specs (#2776)

Summary: Pull Request resolved: #2776 Reviewed By: saitcakmak Differential Revision: D63268113 fbshipit-source-id: e7c03004d29263994fd8dfd46d01ee1bbe415e9d
facebook · Sep 23, 2024 · d738685 · d738685
1 parent 34ad4d0
commit d738685
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ax/modelbridge/model_spec.py b/ax/modelbridge/model_spec.py
@@ -255,6 +255,7 @@ def copy(self) -> ModelSpec:
             model_kwargs=deepcopy(self.model_kwargs),
             model_gen_kwargs=deepcopy(self.model_gen_kwargs),
             model_cv_kwargs=deepcopy(self.model_cv_kwargs),
+            model_key_override=self.model_key_override,
         )
 
     def _safe_to_update(