Skip to content

Commit 94a46ce

Browse files
committed
Replace obselete pw.Model.create_or_get with get_or_create
1 parent b40affc commit 94a46ce

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

cesium_app/models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ def add(name, project, file_uris=[], file_names=[], meta_features=[]):
9696
d = Dataset.create(name=name, project=project,
9797
meta_features=meta_features)
9898
for fname, uri in zip(file_names, file_uris):
99-
f, created = File.create_or_get(name=fname, uri=uri)
99+
f, created = File.get_or_create(name=fname, uri=uri)
100100
DatasetFile.create(dataset=d, file=f)
101101
return d
102102

cesium_app/tests/fixtures.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)