Skip to content

Commit bea0425

Browse files
committed
fix: removed db mapping framework
1 parent edff369 commit bea0425

File tree

1 file changed

+0
-8
lines changed

1 file changed

+0
-8
lines changed

src/webapp/routers/models.py

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

103102
class 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

Comments
 (0)