Skip to content

Commit ef99df2

Browse files
committed
revert get_or_create_project
1 parent 1c2323c commit ef99df2

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

functions/src/auto_trainer/test_auto_trainer.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def test_train(model: Tuple[str, str]):
8282
dataset, label_columns = _get_dataset(model[1])
8383
is_test_passed = True
8484

85-
project = mlrun.get_or_create_project("auto-trainer-test", context="./")
85+
project = mlrun.new_project("auto-trainer-test", context="./")
8686
fn = project.set_function("function.yaml", "train", kind="job", image="mlrun/mlrun")
8787

8888
train_run = None
@@ -119,7 +119,7 @@ def test_train_evaluate(model: Tuple[str, str]):
119119
dataset, label_columns = _get_dataset(model[1])
120120
is_test_passed = True
121121
# Importing function:
122-
project = mlrun.get_or_create_project("auto-trainer-test", context="./")
122+
project = mlrun.new_project("auto-trainer-test", context="./")
123123
fn = project.set_function("function.yaml", "train", kind="job", image="mlrun/mlrun")
124124
temp_dir = tempfile.mkdtemp()
125125

@@ -172,7 +172,7 @@ def test_train_predict(model: Tuple[str, str]):
172172
df = pd.read_csv(dataset)
173173
sample = df.head().drop("labels", axis=1).values.tolist()
174174
# Importing function:
175-
project = mlrun.get_or_create_project("auto-trainer-test", context="./")
175+
project = mlrun.new_project("auto-trainer-test", context="./")
176176
fn = project.set_function("function.yaml", "train", kind="job", image="mlrun/mlrun")
177177
temp_dir = tempfile.mkdtemp()
178178

0 commit comments

Comments
 (0)