Skip to content

Commit 8442a35

Browse files
kiihne-noaailaflott
authored andcommitted
adding fmt to save_refmodel default
1 parent 8fb4711 commit 8442a35

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

src/epmt/epmt_query.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1351,7 +1351,7 @@ def create_refmodel(jobs, name=None, tag={}, op_tags=[],
13511351
info_dict['pca'] = {'inp_features': orig_features, 'out_features': pca_features}
13521352

13531353
# now save the ref model
1354-
save_refmodel(ReferenceModel, jobs=jobs, computed=computed, info_dict = info_dict, enabled=enabled, name=name, tags=tag, op_tags=op_tags)
1354+
save_refmodel(ReferenceModel, jobs=jobs, computed=computed, info_dict = info_dict, enabled=enabled, name=name, tags=tag, op_tags=op_tags, fmt=fmt)
13551355
'''
13561356
r = orm_create(ReferenceModel, jobs=jobs, name=name, tags=tag, op_tags=op_tags, computed=computed, info_dict = info_dict, enabled=enabled)
13571357
orm_commit()
@@ -1364,8 +1364,8 @@ def create_refmodel(jobs, name=None, tag={}, op_tags=[],
13641364
'''
13651365
# returns the number of models deleted.
13661366
#jobs, computed, infodict, and emabled can all be retrieved from the ReferenceModel object
1367-
def save_refmodel(ReferenceModel, jobs, computed, info_dict, enabled, name=None, tags={}, op_tags=[]):
1368-
r = orm_create(ReferenceModel, jobs=jobs, computed=computed, info_dict = info_dict, enabled=enabled, name=name, tags=tags, op_tags=op_tags)
1367+
def save_refmodel(ReferenceModel, jobs, computed, info_dict, enabled, name=None, tags={}, op_tags=[], fmt='dict'):
1368+
r = orm_create(ReferenceModel, jobs=jobs, computed=computed, info_dict = info_dict, enabled=enabled, name=name, tags=tags, op_tags=op_tags, fmt=fmt)
13691369
orm_commit()
13701370
if fmt == 'orm':
13711371
return r

0 commit comments

Comments
 (0)