rename mlflow model

zenml-io · Oct 19, 2023 · ceae88b · ceae88b
1 parent 3605a88
commit ceae88b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/template/inference_config.yaml b/template/inference_config.yaml
@@ -13,7 +13,7 @@ settings:
       - sklearn
       - slack
 extra:
-  mlflow_model_name: e2e_use_case_model
+  mlflow_model_name: {{ product_name }}
 {%- if target_environment == 'production' %}
   target_env: Production
 {%- else %}

diff --git a/template/train_config.yaml b/template/train_config.yaml
@@ -13,7 +13,7 @@ settings:
       - sklearn
       - slack
 extra:
-  mlflow_model_name: e2e_use_case_model
+  mlflow_model_name: {{ product_name }}
 {%- if target_environment == 'production' %}
   target_env: Production
 {%- else %}