From 8f7d67817056e6ec9fba2a292f2c43ecb215c49c Mon Sep 17 00:00:00 2001 From: Victor Grau Serrat Date: Fri, 9 Mar 2018 12:44:01 -0700 Subject: [PATCH] BUG: [mktplace]: fix sanitize_df to handle 1-row DFs --- catalyst/marketplace/utils/bundle_utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/catalyst/marketplace/utils/bundle_utils.py b/catalyst/marketplace/utils/bundle_utils.py index 3fc9a45a..e489c0b5 100644 --- 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