Skip to content

Commit f112fad

Browse files
committed
refactor(prod_model): ordering
1 parent 0c6b977 commit f112fad

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

github_dagger_workflow_project/04_prod_model.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,6 @@
44
from github_dagger_workflow_project.config import PROD_BEST_EXPERIMENT_PATH
55
from github_dagger_workflow_project.mlflow_client import client
66

7-
artifact_path = "model"
8-
model_name = "lead_model"
9-
experiment_best = pd.read_pickle(PROD_BEST_EXPERIMENT_PATH)
10-
117

128
def get_production_model(model_name):
139
return [
@@ -29,6 +25,11 @@ def register_and_wait_model(run_id, artifact_path, model_name):
2925
return dict(model_details)
3026

3127

28+
artifact_path = "model"
29+
model_name = "lead_model"
30+
experiment_best = pd.read_pickle(PROD_BEST_EXPERIMENT_PATH)
31+
32+
3233
train_model_score = experiment_best["metrics.f1_score"]
3334
run_id = None
3435
prod_model = get_production_model(model_name)

0 commit comments

Comments
 (0)