@@ -70,7 +70,7 @@ def __init__(self) -> None:
70
70
self ._request_time : str | None = None
71
71
self ._rows : str | None = None
72
72
self ._url : str | None = None
73
- self ._status_code : str = None
73
+ self ._status_code : str | None = None
74
74
75
75
def set_header (self , Header : DataFrame ) -> None :
76
76
"""
@@ -177,7 +177,7 @@ def _set_status_code(self, status_code: str) -> None:
177
177
None
178
178
179
179
"""
180
- self ._status_code : str = status_code
180
+ self ._status_code = status_code
181
181
182
182
def get_request_time (self ) -> str :
183
183
"""
@@ -940,7 +940,7 @@ def _aqs_services_by_MA(
940
940
def _aqs_list_services (
941
941
self ,
942
942
aqsfilter : str ,
943
- countycode : date | None = None ,
943
+ countycode : str | None = None ,
944
944
stateFIPS : str | None = None ,
945
945
cbsa_code : str | None = None ,
946
946
MA_code : str | None = None ,
@@ -1111,7 +1111,7 @@ def aqs_credentials(username: str | None = None, key: str | None = None) -> None
1111
1111
1112
1112
1113
1113
def aqs_removeheader (
1114
- aqsobject : list [AQSAPI_V2 ],
1114
+ aqsobject : None | DataFrame | AQSAPI_V2 | list [ DataFrame ] | list [AQSAPI_V2 ],
1115
1115
) -> DataFrame | AQSAPI_V2 :
1116
1116
"""
1117
1117
Coerces a single AQS_Data_Mart_APIv2 instance or a list of
@@ -1232,11 +1232,13 @@ def _aqsmultiyearcall(
1232
1232
# until end date (including the year of bdate and not the year of
1233
1233
# edate) with edate appended to the end.
1234
1234
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 )
1236
1237
]
1237
1238
bdatelist .insert (0 , bdate )
1238
1239
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 )
1240
1242
]
1241
1243
edatelist .append (edate )
1242
1244
@@ -1294,33 +1296,33 @@ def _aqsmultiyearcall(
1294
1296
# RuntimeError("invalid function sent to _aqsmultiyearcall")
1295
1297
if fun == "_aqs_services_by_site" :
1296
1298
returnvalue = list (
1297
- starmap (aqsresult ._aqs_services_by_site , params )
1299
+ starmap (aqsresult ._aqs_services_by_site , cast ( Iterable [ Any ], params ))
1298
1300
) # type: ignore
1299
1301
elif fun == "_aqs_services_by_county" :
1300
1302
returnvalue = list (
1301
- starmap (aqsresult ._aqs_services_by_county , params )
1303
+ starmap (aqsresult ._aqs_services_by_county , cast ( Iterable [ Any ], params ) )
1302
1304
) # type: ignore
1303
1305
elif fun == "_aqs_services_by_state" :
1304
1306
returnvalue = list (
1305
- starmap (aqsresult ._aqs_services_by_state , params )
1307
+ starmap (aqsresult ._aqs_services_by_state , cast ( Iterable [ Any ], params ) )
1306
1308
) # type: ignore
1307
1309
elif fun == "_aqs_services_by_MA" :
1308
1310
returnvalue = list (
1309
- starmap (aqsresult ._aqs_services_by_MA , params )
1311
+ starmap (aqsresult ._aqs_services_by_MA ,cast ( Iterable [ Any ], params ) )
1310
1312
) # type: ignore
1311
1313
elif fun == "_aqs_services_by_pqao" :
1312
1314
returnvalue = list (
1313
- starmap (aqsresult ._aqs_services_by_pqao , params )
1315
+ starmap (aqsresult ._aqs_services_by_pqao , cast ( Iterable [ Any ], params ) )
1314
1316
) # type: ignore
1315
1317
elif fun == "_aqs_services_by_cbsa" :
1316
1318
returnvalue = list (
1317
- starmap (aqsresult ._aqs_services_by_cbsa , params )
1319
+ starmap (aqsresult ._aqs_services_by_cbsa , cast ( Iterable [ Any ], params ) )
1318
1320
) # type: ignore
1319
1321
elif fun == "_aqs_services_by_box" :
1320
1322
returnvalue = list (
1321
- starmap (aqsresult ._aqs_services_by_box , params )
1323
+ starmap (aqsresult ._aqs_services_by_box , cast ( Iterable [ Any ], params ) )
1322
1324
) # type: ignore
1323
1325
else :
1324
1326
returnvalue = None
1325
1327
1326
- return returnvalue
1328
+ return cast ( list [ DataFrame ] | None , returnvalue )
0 commit comments