added .csv image loading utils

Lightning-Universe · Feb 14, 2021 · 891a6e9 · 891a6e9
1 parent bc33414
commit 891a6e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flash/data/data_utils.py b/flash/data/data_utils.py
@@ -8,7 +8,7 @@ def labels_from_categorical_csv(
         feature_cols: List,
         return_dict: bool = True,
         index_col_collate_fn: Any = None
-    ) -> Union[Dict, List]:
+) -> Union[Dict, List]:
     """
     Returns a dictionary with {index_col: label} for each entry in the csv.