addresses PR review

RocketPy-Team · Oct 13, 2024 · a419057 · a419057
1 parent 0c1a979
commit a419057
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/lib/models/environment.py b/lib/models/environment.py
@@ -8,7 +8,6 @@ class AtmosphericModelTypes(str, Enum):
     STANDARD_ATMOSPHERE: str = "STANDARD_ATMOSPHERE"
     CUSTOM_ATMOSPHERE: str = "CUSTOM_ATMOSPHERE"
     WYOMING_SOUNDING: str = "WYOMING_SOUNDING"
-    NOAARUCSOUNDING: str = "NOAARUCSOUNDING"
     FORECAST: str = "FORECAST"
     REANALYSIS: str = "REANALYSIS"
     ENSEMBLE: str = "ENSEMBLE"
@@ -17,7 +16,7 @@ class AtmosphericModelTypes(str, Enum):
 class Env(BaseModel):
     latitude: float
     longitude: float
-    elevation: Optional[int] = None
+    elevation: Optional[int] = 0
 
     # Optional parameters
     atmospheric_model_type: Optional[AtmosphericModelTypes] = None

diff --git a/lib/models/motor.py b/lib/models/motor.py
@@ -31,6 +31,9 @@ class InterpolationMethods(str, Enum):
     LINEAR: str = "LINEAR"
     SPLINE: str = "SPLINE"
     AKIMA: str = "AKIMA"
+    POLYNOMIAL: str = "POLYNOMIAL"
+    SHEPARD: str = "SHEPARD"
+    RBF: str = "RBF"
 
 
 class MotorTank(BaseModel):