diff --git a/geomagio/api/ws/DataApiQuery.py b/geomagio/api/ws/DataApiQuery.py index 9c894c46f47864e5384254d71d22c2c3738c9bfc..dc5dcb8ec37612f4e52d28e877bf0d6443756e3e 100644 --- a/geomagio/api/ws/DataApiQuery.py +++ b/geomagio/api/ws/DataApiQuery.py @@ -65,7 +65,7 @@ class DataApiQuery(BaseModel): elements: List[str] = DEFAULT_ELEMENTS sampling_period: SamplingPeriod = SamplingPeriod.MINUTE data_type: Union[DataType, str] = DataType.VARIATION - dbdt: list = [] + dbdt: List[str] = [] format: OutputFormat = OutputFormat.IAGA2002 @validator("data_type") @@ -125,11 +125,11 @@ class DataApiQuery(BaseModel): return endtime @validator("dbdt", always=True) - def validate_dbdt(cls, dbdt: list,) -> list: + def validate_dbdt(cls, dbdt: List[str],) -> List[str]: """Default dbdt based on valid elements. """ for channel in dbdt: - if channel not in DEFAULT_ELEMENTS: + if channel not in ELEMENTS: raise ValueError("Specified channel not found in valid elements.") return dbdt