|
23 | 23 | from typing import Any, ClassVar, Dict, List, Optional, Union
|
24 | 24 | from typing_extensions import Annotated
|
25 | 25 | from jaqpot_api_client.models.doa import Doa
|
| 26 | +from jaqpot_api_client.models.docker_config import DockerConfig |
26 | 27 | from jaqpot_api_client.models.feature import Feature
|
27 | 28 | from jaqpot_api_client.models.library import Library
|
28 | 29 | from jaqpot_api_client.models.model_scores import ModelScores
|
@@ -67,9 +68,10 @@ class Model(BaseModel):
|
67 | 68 | legacy_prediction_service: Optional[StrictStr] = Field(default=None, alias="legacyPredictionService")
|
68 | 69 | scores: Optional[ModelScores] = None
|
69 | 70 | r_pbpk_config: Optional[RPbpkConfig] = Field(default=None, alias="rPbpkConfig")
|
| 71 | + docker_config: Optional[DockerConfig] = Field(default=None, alias="dockerConfig") |
70 | 72 | created_at: Optional[datetime] = Field(default=None, description="The date and time when the feature was created.", alias="createdAt")
|
71 | 73 | updated_at: Optional[datetime] = Field(default=None, description="The date and time when the model was last updated.", alias="updatedAt")
|
72 |
| - __properties: ClassVar[List[str]] = ["id", "name", "description", "type", "jaqpotpyVersion", "doas", "libraries", "dependentFeatures", "independentFeatures", "sharedWithOrganizations", "visibility", "task", "archived", "archivedAt", "torchConfig", "preprocessors", "featurizers", "rawPreprocessor", "rawModel", "creator", "canEdit", "isAdmin", "selectedFeatures", "tags", "legacyPredictionService", "scores", "rPbpkConfig", "createdAt", "updatedAt"] |
| 74 | + __properties: ClassVar[List[str]] = ["id", "name", "description", "type", "jaqpotpyVersion", "doas", "libraries", "dependentFeatures", "independentFeatures", "sharedWithOrganizations", "visibility", "task", "archived", "archivedAt", "torchConfig", "preprocessors", "featurizers", "rawPreprocessor", "rawModel", "creator", "canEdit", "isAdmin", "selectedFeatures", "tags", "legacyPredictionService", "scores", "rPbpkConfig", "dockerConfig", "createdAt", "updatedAt"] |
73 | 75 |
|
74 | 76 | model_config = ConfigDict(
|
75 | 77 | populate_by_name=True,
|
@@ -168,6 +170,9 @@ def to_dict(self) -> Dict[str, Any]:
|
168 | 170 | # override the default output from pydantic by calling `to_dict()` of r_pbpk_config
|
169 | 171 | if self.r_pbpk_config:
|
170 | 172 | _dict['rPbpkConfig'] = self.r_pbpk_config.to_dict()
|
| 173 | + # override the default output from pydantic by calling `to_dict()` of docker_config |
| 174 | + if self.docker_config: |
| 175 | + _dict['dockerConfig'] = self.docker_config.to_dict() |
171 | 176 | return _dict
|
172 | 177 |
|
173 | 178 | @classmethod
|
@@ -207,6 +212,7 @@ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
|
207 | 212 | "legacyPredictionService": obj.get("legacyPredictionService"),
|
208 | 213 | "scores": ModelScores.from_dict(obj["scores"]) if obj.get("scores") is not None else None,
|
209 | 214 | "rPbpkConfig": RPbpkConfig.from_dict(obj["rPbpkConfig"]) if obj.get("rPbpkConfig") is not None else None,
|
| 215 | + "dockerConfig": DockerConfig.from_dict(obj["dockerConfig"]) if obj.get("dockerConfig") is not None else None, |
210 | 216 | "createdAt": obj.get("createdAt"),
|
211 | 217 | "updatedAt": obj.get("updatedAt")
|
212 | 218 | })
|
|
0 commit comments