diff --git a/src/repositories/flight.py b/src/repositories/flight.py index 0df540c..a9587da 100644 --- a/src/repositories/flight.py +++ b/src/repositories/flight.py @@ -28,7 +28,7 @@ async def read_flight_by_id(self, flight_id: str) -> Optional[FlightModel]: @repository_exception_handler async def update_flight_by_id(self, flight_id: str, flight: FlightModel): await self.update_by_id( - flight.model_dump(exclude_none=True), data_id=flight_id + flight.model_dump(exclude_none=False), data_id=flight_id ) @repository_exception_handler diff --git a/src/repositories/motor.py b/src/repositories/motor.py index 52c016b..023b31f 100644 --- a/src/repositories/motor.py +++ b/src/repositories/motor.py @@ -28,7 +28,7 @@ async def read_motor_by_id(self, motor_id: str) -> Optional[MotorModel]: @repository_exception_handler async def update_motor_by_id(self, motor_id: str, motor: MotorModel): await self.update_by_id( - motor.model_dump(exclude_none=True), data_id=motor_id + motor.model_dump(exclude_none=False), data_id=motor_id ) @repository_exception_handler diff --git a/src/repositories/rocket.py b/src/repositories/rocket.py index a3849d7..1c72959 100644 --- a/src/repositories/rocket.py +++ b/src/repositories/rocket.py @@ -28,7 +28,7 @@ async def read_rocket_by_id(self, rocket_id: str) -> Optional[RocketModel]: @repository_exception_handler async def update_rocket_by_id(self, rocket_id: str, rocket: RocketModel): await self.update_by_id( - rocket.model_dump(exclude_none=True), data_id=rocket_id + rocket.model_dump(exclude_none=False), data_id=rocket_id ) @repository_exception_handler diff --git a/src/utils.py b/src/utils.py index c964e3f..e028736 100644 --- a/src/utils.py +++ b/src/utils.py @@ -54,8 +54,10 @@ def for_flight(cls) -> 'DiscretizeConfig': class InfinityEncoder(RocketPyEncoder): - def default(self, o): - obj = o + def __init__(self, *args, **kwargs): + super().__init__(*args, **kwargs) + + def default(self, obj): if ( isinstance(obj, Function) and not callable(obj.source) @@ -73,7 +75,7 @@ def default(self, o): mutate_self=False, ) if isinstance(obj, Flight): - obj._Flight__evaluate_post_process() + obj._Flight__evaluate_post_process solution = np.array(obj.solution) size = len(solution) if size > 25: diff --git a/src/views/environment.py b/src/views/environment.py index 5147d35..a04603f 100644 --- a/src/views/environment.py +++ b/src/views/environment.py @@ -43,7 +43,7 @@ class EnvironmentSimulation(ApiBaseView): initial_east: Optional[float] = None initial_hemisphere: Optional[str] = None initial_ew: Optional[str] = None - max_expected_height: Optional[int] = None + max_expected_height: Optional[float] = None date: Optional[datetime] = Field(default_factory=_default_future_datetime) local_date: Optional[datetime] = Field( default_factory=_default_future_datetime