@@ -89,7 +89,7 @@ def create_test_featureset(project, label_type='class'):
8989 fset_path = pjoin (cfg ['paths' ]['features_folder' ],
9090 '{}.nc' .format (str (uuid .uuid4 ())))
9191 fset_data .to_netcdf (fset_path )
92- f , created = m .File .create_or_get (uri = fset_path )
92+ f , created = m .File .get_or_create (uri = fset_path )
9393 fset = m .Featureset .create (name = 'test_featureset' , file = f , project = project ,
9494 features_list = features_to_use ,
9595 custom_features_script = None ,
@@ -133,7 +133,7 @@ def create_test_model(fset, model_type='RandomForestClassifier'):
133133 model_path = pjoin (cfg ['paths' ]['models_folder' ],
134134 '{}.pkl' .format (str (uuid .uuid4 ())))
135135 joblib .dump (model_data , model_path )
136- f , created = m .File .create_or_get (uri = model_path )
136+ f , created = m .File .get_or_create (uri = model_path )
137137 model = m .Model .create (name = 'test_model' ,
138138 file = f , featureset = fset , project = fset .project ,
139139 params = model_params [model_type ], type = model_type ,
@@ -163,7 +163,7 @@ def create_test_prediction(dataset, model):
163163 pred_path = pjoin (cfg ['paths' ]['predictions_folder' ],
164164 '{}.nc' .format (str (uuid .uuid4 ())))
165165 pred_data .to_netcdf (pred_path )
166- f , created = m .File .create_or_get (uri = pred_path )
166+ f , created = m .File .get_or_create (uri = pred_path )
167167 pred = m .Prediction .create (file = f , dataset = dataset , project = dataset .project ,
168168 model = model , finished = datetime .datetime .now ())
169169 pred .save ()
0 commit comments