@@ -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