Skip to content

Commit d5c46f9

Browse files
Update generative-proof-of-concept-CPU-preprocessing-in-memory.py
subprocess.run called to avoid naming collision with mlflow.run
1 parent bb206ca commit d5c46f9

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,12 @@
22
import os
33
import mlflow
44
from datetime import datetime
5-
from subprocess import run
5+
import subprocess
66
from warnings import warn
77

88
MLFLOW_PORT = 7777
99

10-
answer = run(f"mlflow server --host 127.0.0.1 --port {MLFLOW_PORT} &",
10+
answer = subprocess.run(f"mlflow server --host 127.0.0.1 --port {MLFLOW_PORT} &",
1111
shell=True,
1212
)
1313
print(answer.stdout)
@@ -769,7 +769,7 @@ def create_dataset(raw_text_samples, tokenizer, sample_expansion_batch_size=50,
769769
# Test inter - module serialization
770770

771771
print(f"🧪 Running serialization test for trial {trial_number}...")
772-
result = run(
772+
result = subprocess.run(
773773
f"python3 test_llm_serialization.py {TOKENIZER_SAVE_PATH} {MODEL_SAVE_PATH}",
774774
capture_output=True,
775775
shell=True

0 commit comments

Comments
 (0)