mostlyai-mock 0.2.0__tar.gz → 0.2.1__tar.gz

{mostlyai_mock-0.2.0 → mostlyai_mock-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mostlyai-mock
-Version: 0.2.0
+Version: 0.2.1
 Summary: Synthetic Mock Data
 Project-URL: homepage, https://github.com/mostly-ai/mostlyai-mock
 Project-URL: repository, https://github.com/mostly-ai/mostlyai-mock

{mostlyai_mock-0.2.0 → mostlyai_mock-0.2.1}/mostlyai/mock/__init__.py

@@ -15,4 +15,4 @@
 from mostlyai.mock.core import sample
 
 __all__ = ["sample"]
-__version__ = "0.2.0"  # Do not set this manually. Use poetry version [params].
+__version__ = "0.2.1"  # Do not set this manually. Use poetry version [params].

{mostlyai_mock-0.2.0 → mostlyai_mock-0.2.1}/mostlyai/mock/core.py

@@ -880,6 +880,10 @@ async def _create_table_rows_generator(
         # +2 because LLM may not always count the rows correctly
         batch_sizes[-1] = sample_size - sum(batch_sizes[:-1]) + 2
 
+    # emit initial progress message right away
+    if progress_callback:
+        await progress_callback(table=name, progress=0, total=n_total_batches, rows=0, elapsed_time=0)
+
     # initialize queues for async communication
     batch_queue = asyncio.PriorityQueue()
     result_queue = asyncio.Queue()
@@ -944,6 +948,7 @@ async def _create_table_rows_generator(
 
     n_completed_batches = 0
     n_yielded_sequences = 0
+    n_generated_rows = 0
     table_start_time = time.time()
     while n_yielded_sequences < sample_size:
         if n_completed_batches >= n_total_batches:
@@ -968,6 +973,7 @@ async def _create_table_rows_generator(
         rows = result
         for row_idx, row in enumerate(rows):
             yield (batch_idx, row)
+            n_generated_rows += 1
             if context_batches is None or row_idx == len(rows) - 1:
                 # in case of flat table, each row is considered a single sequence
                 # in case of linked table, all rows are considered a single sequence
@@ -982,7 +988,7 @@ async def _create_table_rows_generator(
                 table=name,
                 progress=n_completed_batches,
                 total=n_total_batches,
-                rows=n_yielded_sequences,
+                rows=n_generated_rows,
                 elapsed_time=round(elapsed_time, 2),
             )
         result_queue.task_done()

{mostlyai_mock-0.2.0 → mostlyai_mock-0.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mostlyai-mock"
-version = "0.2.0"
+version = "0.2.1"
 description = "Synthetic Mock Data"
 authors = [{ name = "MOSTLY AI", email = "dev@mostly.ai" }]
 requires-python = ">=3.10"