Fix missing data folder case in data_utils

2015-03-29 16:15:18 -07:00 · 2015-03-29 16:15:18 -07:00 · c589c524d7
commit c589c524d7
parent 6b8dce4779
1 changed files with 2 additions and 0 deletions
--- a/keras/datasets/data_utils.py
+++ b/keras/datasets/data_utils.py
@ -5,6 +5,8 @@ from ..utils.generic_utils import Progbar
 def get_file(fname, origin, untar=False):
    datadir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
    datadir = os.path.join(datadir, 'data')
+    if not os.path.exists(datadir):
+        os.makedirs(datadir)

    if untar:
        untar_fpath = os.path.join(datadir, fname)