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