@@ -58,7 +58,7 @@ class Image(BaseModel):
5858
5959 @field_validator ("node_requirements" , mode = "before" )
6060 @classmethod
61- def migrate_from_requirements (cls , v , info : ValidationInfo ):
61+ def _migrate_from_requirements (cls , v , info : ValidationInfo ):
6262 if v is None :
6363 # NOTE: 'node_requirements' field's default=None although is NOT declared as nullable.
6464 # Then this validator with `pre=True, always=True` is used to create a default
@@ -147,12 +147,12 @@ class CompTaskAtDB(BaseModel):
147147 created : datetime .datetime
148148 modified : datetime .datetime
149149 # Additional information about price and hardware (ex. AWS EC2 instance type)
150- pricing_info : dict | None = None
150+ pricing_info : dict | None
151151 hardware_info : HardwareInfo
152152
153153 @field_validator ("state" , mode = "before" )
154154 @classmethod
155- def convert_state_from_state_type_enum_if_needed (cls , v ):
155+ def _convert_state_from_state_type_enum_if_needed (cls , v ):
156156 if isinstance (v , str ):
157157 # try to convert to a StateType, if it fails the validations will continue
158158 # and pydantic will try to convert it to a RunninState later on
@@ -164,14 +164,14 @@ def convert_state_from_state_type_enum_if_needed(cls, v):
164164
165165 @field_validator ("start" , "end" , "submit" )
166166 @classmethod
167- def ensure_utc (cls , v : datetime .datetime | None ) -> datetime .datetime | None :
167+ def _ensure_utc (cls , v : datetime .datetime | None ) -> datetime .datetime | None :
168168 if v is not None and v .tzinfo is None :
169169 v = v .replace (tzinfo = datetime .UTC )
170170 return v
171171
172172 @field_validator ("hardware_info" , mode = "before" )
173173 @classmethod
174- def backward_compatible_null_value (cls , v : HardwareInfo | None ) -> HardwareInfo :
174+ def _backward_compatible_null_value (cls , v : HardwareInfo | None ) -> HardwareInfo :
175175 if v is None :
176176 return HardwareInfo (aws_ec2_instances = [])
177177 return v
0 commit comments