Bug fix

microsoft · Nov 10, 2024 · 4f3dfc3 · 4f3dfc3
1 parent d618a1a
commit 4f3dfc3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/intelligence_toolkit/generate_mock_data/data_generator.py b/intelligence_toolkit/generate_mock_data/data_generator.py
@@ -55,7 +55,7 @@ async def generate_data(
         required = remainder / records_per_batch
         if not required.is_integer():
             required += 1
-        batches = min(parallel_batches, required)
+        batches = min(parallel_batches, int(required))
         sample_records = sample_from_record_array(
             current_object_json, primary_record_array, batches
         )