Fix DataframeInput error when microbatch enabled (#802)

bentoml · Jun 15, 2020 · 5b6bd29 · 5b6bd29
1 parent 6156a96
commit 5b6bd29
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/bentoml/adapters/dataframe_input.py b/bentoml/adapters/dataframe_input.py
@@ -294,7 +294,10 @@ def handle_batch_request(
         ]
         # TODO: check content_type
 
-        df_conc, slices = read_dataframes_from_json_n_csv(datas, content_types)
+        df_conc, slices_generator = read_dataframes_from_json_n_csv(
+            datas, content_types
+        )
+        slices = list(slices_generator)
 
         result_conc = func(df_conc)