diff --git a/data/maker/__init__.py b/data/maker/__init__.py index 78bc08d..527d245 100644 --- a/data/maker/__init__.py +++ b/data/maker/__init__.py @@ -211,7 +211,7 @@ def generate(**args): # else: # values = df[col].dropna().unique().tolist() msize = args['matrix_size'] if 'matrix_size' in args else 128 - values = bhandler.get_column_values(df[col]) + values = bhandler.get_column_values(df[col],msize)