Skip to content

Commit 638c329

Browse files
Update generative-proof-of-concept-CPU-preprocessing-in-memory.py
Made MLFLOW_PORT a param
1 parent c1f5f7d commit 638c329

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

generative-proof-of-concept-CPU-preprocessing-in-memory.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,9 @@
44
from datetime import datetime
55
from subprocess import run
66

7-
answer = run("mlflow server --host 127.0.0.1 --port 5000 &",
7+
MLFLOW_PORT = 5000
8+
9+
answer = run(f"mlflow server --host 127.0.0.1 --port {MLFLOW_PORT} &",
810
shell=True,
911
)
1012
print(answer.stdout)
@@ -15,8 +17,7 @@
1517
N_TRIALS = 30
1618

1719

18-
19-
mlflow.set_tracking_uri(uri="http://127.0.0.1:5000")
20+
mlflow.set_tracking_uri(uri="http://127.0.0.1:{MLFLOW_PORT}")
2021

2122
mlflow.set_experiment(f"single-worker-1st-pass-tuning-{EXPERIMENT_ITERATION}-a")
2223

0 commit comments

Comments
 (0)