@@ -97,7 +97,6 @@ class ModelCreationRequest(BaseModel):
9797 # valid = False, means the model is not ready for use.
9898 valid : bool = False
9999 schema_configs : list [list [SchemaConfigObj ]]
100- framework : str | None = None
101100
102101
103102class ModelInfo (BaseModel ):
@@ -217,11 +216,6 @@ def create_model(
217216 created_by = str_to_uuid (current_user .user_id ),
218217 valid = req .valid ,
219218 schema_configs = jsonpickle .encode (req .schema_configs ),
220- framework = (
221- f
222- if (f := (req .framework or "" ).strip ().lower ()) in {"sklearn" , "h2o" }
223- else "sklearn"
224- ),
225219 )
226220 local_session .get ().add (model )
227221 local_session .get ().commit ()
@@ -259,7 +253,6 @@ def create_model(
259253 "created_by" : uuid_to_str (query_result [0 ][0 ].created_by ),
260254 "deleted" : query_result [0 ][0 ].deleted ,
261255 "valid" : query_result [0 ][0 ].valid ,
262- "framework" : query_result [0 ][0 ].framework ,
263256 }
264257
265258
@@ -307,7 +300,6 @@ def read_inst_model(
307300 "created_by" : uuid_to_str (query_result [0 ][0 ].created_by ),
308301 "deleted" : query_result [0 ][0 ].deleted ,
309302 "valid" : query_result [0 ][0 ].valid ,
310- "framework" : query_result [0 ][0 ].framework ,
311303 }
312304
313305
0 commit comments