BUG: [mktplace]: fix sanitize_df to handle 1-row DFs

linmingren · Mar 9, 2018 · 8f7d678 · 8f7d678
1 parent 1de084a
commit 8f7d678
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/catalyst/marketplace/utils/bundle_utils.py b/catalyst/marketplace/utils/bundle_utils.py
@@ -47,8 +47,10 @@ def sanitize_df(df):
     # large datasets
     if len(df) > 100:
         indexes = random.sample(range(0, len(df) - 1), 100)
-    else:
+    elif len(df) > 1:
         indexes = range(0, len(df) - 1)
+    else:
+        indexes = [0, ]
 
     for column in df.columns:
         is_date = False