6
6
from numpy import isnan
7
7
from pandas import DataFrame , concat
8
8
9
- from .csv_importer import CsvRowValues
9
+ from .csv_importer import CsvImporter
10
10
from ...server .utils .dates import date_to_time_value , time_value_to_date
11
11
12
12
@@ -79,7 +79,7 @@ def _sanity_check_fields(self, test_mode: bool = True):
79
79
return self
80
80
81
81
@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 ):
83
83
if row_value is None :
84
84
return None
85
85
return CovidcastRow (
@@ -100,7 +100,7 @@ def fromCsvRowValue(row_value: Optional[CsvRowValues], source: str, signal: str,
100
100
)
101
101
102
102
@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 ):
104
104
# NOTE: returns a generator, as row_values is expected to be a generator
105
105
return (CovidcastRow .fromCsvRowValue (row_value , source , signal , time_type , geo_type , time_value , issue , lag ) for row_value in row_values )
106
106
0 commit comments