From dfcac50bc8641d8cb9a4521ef0d7fdaa7e0bb97c Mon Sep 17 00:00:00 2001
From: pcain-usgs <pcain@usgs.gov>
Date: Mon, 15 Jun 2020 14:39:16 -0600
Subject: [PATCH] Refine defaults/return values
---
geomagio/api/ws/DataApiQuery.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/geomagio/api/ws/DataApiQuery.py b/geomagio/api/ws/DataApiQuery.py
index 9c894c46f..dc5dcb8ec 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
--
GitLab