def save_model_info(model_info): model = MLModel() model.f_create_time = current_timestamp() for k, v in model_info.items(): attr_name = 'f_%s' % k if hasattr(MLModel, attr_name): setattr(model, attr_name, v) elif hasattr(MLModel, k): setattr(model, k, v) try: rows = model.save(force_insert=True) if rows != 1: raise Exception("Create {} failed".format(MLModel)) if RuntimeConfig.zk_client is not None: ServiceUtils.register( RuntimeConfig.zk_client, gen_party_model_id(role=model.f_role, party_id=model.f_party_id, model_id=model.f_model_id), model.f_model_version) return model except peewee.IntegrityError as e: if e.args[0] == 1062: sql_logger(job_id=model_info.get("job_id", "fate_flow")).warning(e) else: raise Exception("Create {} failed:\n{}".format(MLModel, e)) except Exception as e: raise Exception("Create {} failed:\n{}".format(MLModel, e))
def save_model_info(model_info): model = MLModel() model.f_create_time = current_timestamp() for k, v in model_info.items(): attr_name = 'f_%s' % k if hasattr(MLModel, attr_name): setattr(model, attr_name, v) elif hasattr(MLModel, k): setattr(model, k, v) try: rows = model.save(force_insert=True) if rows != 1: raise Exception("Save to database failed") except peewee.IntegrityError as e: if e.args[0] != 1062: raise Exception("Create {} failed:\n{}".format(MLModel, e)) sql_logger(job_id=model_info.get("job_id", "fate_flow")).warning(e) return except Exception as e: raise Exception("Create {} failed:\n{}".format(MLModel, e)) RuntimeConfig.SERVICE_DB.register_model( gen_party_model_id(role=model.f_role, party_id=model.f_party_id, model_id=model.f_model_id), model.f_model_version) return model
def save_model_info(model_info): model = MLModel() model.f_create_time = current_timestamp() for k, v in model_info.items(): attr_name = 'f_%s' % k if hasattr(MLModel, attr_name): setattr(model, attr_name, v) elif hasattr(MLModel, k): setattr(model, k, v) try: rows = model.save(force_insert=True) if rows != 1: raise Exception("Create {} failed".format(MLModel)) return model except peewee.IntegrityError as e: if e.args[0] == 1062: sql_logger(job_id=model_info.get("job_id", "fate_flow")).warning(e) else: raise Exception("Create {} failed:\n{}".format(MLModel, e)) except Exception as e: raise Exception("Create {} failed:\n{}".format(MLModel, e))