BUG: check that end and start are not none before normalizing them

LuCeHe · May 23, 2018 · 2767128 · 2767128
1 parent 8f1dac6
commit 2767128
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/catalyst/utils/run_algo.py b/catalyst/utils/run_algo.py
@@ -283,8 +283,8 @@ def choose_loader(column):
         # We still need to support bundles for other misc data, but we
         # can handle this later.
 
-        if start != pd.tslib.normalize_date(start) or \
-                end != pd.tslib.normalize_date(end):
+        if (start and start != pd.tslib.normalize_date(start)) or \
+                (end and end != pd.tslib.normalize_date(end)):
             # todo: add to Sim_Params the option to
             # start & end at specific times
             log.warn(