add b param to irt model

2021-11-30 21:12:08 +00:00
parent ebccd9522a
commit b2c3f0df79
4 changed files with 6 additions and 6 deletions
--- a/app/models/constraint.py
+++ b/app/models/constraint.py
@ -1,5 +1,4 @@
 from pydantic import BaseModel
-from typing import Optional

 from models.attribute import Attribute

--- a/app/models/irt_model.py
+++ b/app/models/irt_model.py
@ -1,7 +1,8 @@
 from pydantic import BaseModel
+from typing import Dict

 class IRTModel(BaseModel):
  a_param: float
-  b_param: float
+  b_param: Dict = {"schema_bson_id": str, "field_bson_id": str}
  c_param: float
  model: str