From 3a07e7d367cadf22be7684850ec97c6f5067e1c3 Mon Sep 17 00:00:00 2001
From: William Falcon <waf2107@columbia.edu>
Date: Sat, 13 Feb 2021 20:58:54 -0500
Subject: [PATCH] added .csv image loading utils

---
 flash/data/data_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/flash/data/data_utils.py b/flash/data/data_utils.py
index 0529884b34..50e6b3bf63 100644
--- 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.