Skip to content

Commit

Permalink
Use LazyDataFrame
Browse files Browse the repository at this point in the history
Signed-off-by: Bram Stoeller <[email protected]>
  • Loading branch information
bramstoeller committed Mar 2, 2023
1 parent 947c77f commit dd15250
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/power_grid_model_io/data_stores/csv_dir_store.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ def load(self) -> TabularData:
Create a lazy loader for all CSV files in a directory and store them in a TabularData instance.
"""

def lazy_csv_loader(csv_path: Path) -> Callable[[], pd.DataFrame]:
def lazy_csv_loader(csv_path: Path) -> LazyDataFrame:
def csv_loader():
return pd.read_csv(filepath_or_buffer=csv_path, header=self._header_rows, **self._csv_kwargs)

Expand Down

0 comments on commit dd15250

Please sign in to comment.