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