Skip to content

Commit c0dff23

Browse files
committed
Update a few type hints
1 parent 5bdabb0 commit c0dff23

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/acquisition/covidcast/covidcast_row.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from numpy import isnan
77
from pandas import DataFrame, concat
88

9-
from .csv_importer import CsvRowValues
9+
from .csv_importer import CsvImporter
1010
from ...server.utils.dates import date_to_time_value, time_value_to_date
1111

1212

@@ -79,7 +79,7 @@ def _sanity_check_fields(self, test_mode: bool = True):
7979
return self
8080

8181
@staticmethod
82-
def fromCsvRowValue(row_value: Optional[CsvRowValues], source: str, signal: str, time_type: str, geo_type: str, time_value: int, issue: int, lag: int):
82+
def fromCsvRowValue(row_value: Optional[CsvImporter.RowValues], source: str, signal: str, time_type: str, geo_type: str, time_value: int, issue: int, lag: int):
8383
if row_value is None:
8484
return None
8585
return CovidcastRow(
@@ -100,7 +100,7 @@ def fromCsvRowValue(row_value: Optional[CsvRowValues], source: str, signal: str,
100100
)
101101

102102
@staticmethod
103-
def fromCsvRows(row_values: Iterable[Optional[CsvRowValues]], source: str, signal: str, time_type: str, geo_type: str, time_value: int, issue: int, lag: int):
103+
def fromCsvRows(row_values: Iterable[Optional[CsvImporter.RowValues]], source: str, signal: str, time_type: str, geo_type: str, time_value: int, issue: int, lag: int):
104104
# NOTE: returns a generator, as row_values is expected to be a generator
105105
return (CovidcastRow.fromCsvRowValue(row_value, source, signal, time_type, geo_type, time_value, issue, lag) for row_value in row_values)
106106

0 commit comments

Comments
 (0)