refactor: put all types in _model

cmu-delphi · Jul 15, 2024 · 871c832 · 871c832
1 parent b92ccbe
commit 871c832
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/epidatpy/_covidcast.py b/epidatpy/_covidcast.py
@@ -25,12 +25,11 @@
     EpidataFieldInfo,
     EpidataFieldType,
     EpiRangeParam,
+    GeoType,
     InvalidArgumentException,
+    TimeType,
 )
 
-GeoType = Literal["nation", "msa", "hrr", "hhs", "state", "county"]
-TimeType = Literal["day", "week"]
-
 
 @dataclass
 class WebLink:

diff --git a/epidatpy/_model.py b/epidatpy/_model.py
@@ -4,6 +4,7 @@
 from typing import (
     Final,
     List,
+    Literal,
     Mapping,
     Optional,
     Sequence,
@@ -24,6 +25,8 @@
     parse_user_date_or_week,
 )
 
+GeoType = Literal["nation", "msa", "hrr", "hhs", "state", "county"]
+TimeType = Literal["day", "week"]
 EpiDateLike = Union[int, str, date, Week]
 EpiRangeDict = TypedDict("EpiRangeDict", {"from": EpiDateLike, "to": EpiDateLike})
 EpiRangeLike = Union[int, str, "EpiRange", EpiRangeDict, date, Week]