added .csv image loading utils

Lightning-Universe · Feb 14, 2021 · 3a07e7d · 3a07e7d
1 parent 70b42a9
commit 3a07e7d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flash/data/data_utils.py b/flash/data/data_utils.py
@@ -3,7 +3,7 @@
 import pandas as pd
 
 
-def labels_from_categorical_csv(csv: str, index_col: str, return_dict: dict = True) -> Union[Dict, List]:
+def labels_from_categorical_csv(csv: str, index_col: str, return_dict: bool = True) -> Union[Dict, List]:
     """
     Returns a dictionary with {index_col: label} for each entry in the csv.