Skip to content

Commit 00d0cf9

Browse files
committed
fix input types
1 parent e00bb69 commit 00d0cf9

1 file changed

Lines changed: 11 additions & 11 deletions

File tree

dataretrieval/waterdata/api.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1649,18 +1649,18 @@ def get_samples(
16491649
def get_statistics(
16501650
service: STATISTICS_SERVICES = "observationNormals",
16511651
approval_status: Optional[str] = None,
1652-
computation_type: Optional[str] = None,
1653-
country_code: Optional[str] = None,
1654-
state_code: Optional[str] = None,
1655-
county_code: Optional[str] = None,
1656-
start_date: Optional[Union[str, pd.datetime]] = None,
1657-
end_date: Optional[Union[str, pd.datetime]] = None,
1658-
monitoring_location_id: Optional[str] = None,
1652+
computation_type: Optional[Union[str, list[str]]] = None,
1653+
country_code: Optional[Union[str, list[str]]] = None,
1654+
state_code: Optional[Union[str, list[str]]] = None,
1655+
county_code: Optional[Union[str, list[str]]] = None,
1656+
start_date: Optional[str] = None,
1657+
end_date: Optional[str] = None,
1658+
monitoring_location_id: Optional[Union[str, list[str]]] = None,
16591659
page_size: int = 1000,
1660-
parent_timeseries_id: Optional[str] = None,
1661-
site_type_code: Optional[str] = None,
1662-
site_type_name: Optional[str] = None,
1663-
parameter_code: Optional[str] = None,
1660+
parent_timeseries_id: Optional[Union[str, list[str]]] = None,
1661+
site_type_code: Optional[Union[str, list[str]]] = None,
1662+
site_type_name: Optional[Union[str, list[str]]] = None,
1663+
parameter_code: Optional[Union[str, list[str]]] = None,
16641664
) -> Tuple[pd.DataFrame, BaseMetadata]:
16651665
"""Get water data statistics from the USGS Water Data API.
16661666
This service provides endpoints for access to computations on the

0 commit comments

Comments
 (0)