Skip to content

Commit 86e466f

Browse files
edit pyproject.toml due to build failure
1 parent 44e88bf commit 86e466f

File tree

2 files changed

+17
-15
lines changed

2 files changed

+17
-15
lines changed

pyaqsapi/bystate/bystate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ def monitors(
1616
cbdate: date | None = None,
1717
cedate: date | None = None,
1818
return_header: bool | None = False,
19-
) -> AQSAPI_V2 | DataFrame:
19+
) -> list[DataFrame] | AQSAPI_V2 | DataFrame:
2020
"""
2121
Return a table of monitors.
2222

pyaqsapi/helperfunctions.py

Lines changed: 16 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def __init__(self) -> None:
7070
self._request_time: str | None = None
7171
self._rows: str | None = None
7272
self._url: str | None = None
73-
self._status_code: str = None
73+
self._status_code: str | None = None
7474

7575
def set_header(self, Header: DataFrame) -> None:
7676
"""
@@ -177,7 +177,7 @@ def _set_status_code(self, status_code: str) -> None:
177177
None
178178
179179
"""
180-
self._status_code: str = status_code
180+
self._status_code = status_code
181181

182182
def get_request_time(self) -> str:
183183
"""
@@ -940,7 +940,7 @@ def _aqs_services_by_MA(
940940
def _aqs_list_services(
941941
self,
942942
aqsfilter: str,
943-
countycode: date | None = None,
943+
countycode: str | None = None,
944944
stateFIPS: str | None = None,
945945
cbsa_code: str | None = None,
946946
MA_code: str | None = None,
@@ -1111,7 +1111,7 @@ def aqs_credentials(username: str | None = None, key: str | None = None) -> None
11111111

11121112

11131113
def aqs_removeheader(
1114-
aqsobject: list[AQSAPI_V2],
1114+
aqsobject: None | DataFrame | AQSAPI_V2 | list[DataFrame] | list[AQSAPI_V2],
11151115
) -> DataFrame | AQSAPI_V2:
11161116
"""
11171117
Coerces a single AQS_Data_Mart_APIv2 instance or a list of
@@ -1232,11 +1232,13 @@ def _aqsmultiyearcall(
12321232
# until end date (including the year of bdate and not the year of
12331233
# edate) with edate appended to the end.
12341234
bdatelist = [
1235-
date(year=x, month=1, day=1) for x in range(bdate.year + 1, edate.year + 1)
1235+
date(year=x, month=1, day=1) for x in range(bdate.year + 1,
1236+
edate.year + 1)
12361237
]
12371238
bdatelist.insert(0, bdate)
12381239
edatelist = [
1239-
date(year=y, month=12, day=31) for y in range(bdate.year, edate.year)
1240+
date(year=y, month=12, day=31) for y in range(bdate.year,
1241+
edate.year)
12401242
]
12411243
edatelist.append(edate)
12421244

@@ -1294,33 +1296,33 @@ def _aqsmultiyearcall(
12941296
# RuntimeError("invalid function sent to _aqsmultiyearcall")
12951297
if fun == "_aqs_services_by_site":
12961298
returnvalue = list(
1297-
starmap(aqsresult._aqs_services_by_site, params)
1299+
starmap(aqsresult._aqs_services_by_site, cast(Iterable[Any], params))
12981300
) # type: ignore
12991301
elif fun == "_aqs_services_by_county":
13001302
returnvalue = list(
1301-
starmap(aqsresult._aqs_services_by_county, params)
1303+
starmap(aqsresult._aqs_services_by_county, cast(Iterable[Any], params))
13021304
) # type: ignore
13031305
elif fun == "_aqs_services_by_state":
13041306
returnvalue = list(
1305-
starmap(aqsresult._aqs_services_by_state, params)
1307+
starmap(aqsresult._aqs_services_by_state, cast(Iterable[Any], params))
13061308
) # type: ignore
13071309
elif fun == "_aqs_services_by_MA":
13081310
returnvalue = list(
1309-
starmap(aqsresult._aqs_services_by_MA, params)
1311+
starmap(aqsresult._aqs_services_by_MA,cast(Iterable[Any], params))
13101312
) # type: ignore
13111313
elif fun == "_aqs_services_by_pqao":
13121314
returnvalue = list(
1313-
starmap(aqsresult._aqs_services_by_pqao, params)
1315+
starmap(aqsresult._aqs_services_by_pqao, cast(Iterable[Any], params))
13141316
) # type: ignore
13151317
elif fun == "_aqs_services_by_cbsa":
13161318
returnvalue = list(
1317-
starmap(aqsresult._aqs_services_by_cbsa, params)
1319+
starmap(aqsresult._aqs_services_by_cbsa, cast(Iterable[Any], params))
13181320
) # type: ignore
13191321
elif fun == "_aqs_services_by_box":
13201322
returnvalue = list(
1321-
starmap(aqsresult._aqs_services_by_box, params)
1323+
starmap(aqsresult._aqs_services_by_box, cast(Iterable[Any], params))
13221324
) # type: ignore
13231325
else:
13241326
returnvalue = None
13251327

1326-
return returnvalue
1328+
return cast(list[DataFrame] | None, returnvalue)

0 commit comments

Comments
 (0)