data-designer 0.3.2__py3-none-any.whl → 0.3.3__py3-none-any.whl

data_designer/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.3.2'
-__version_tuple__ = version_tuple = (0, 3, 2)
+__version__ = version = '0.3.3'
+__version_tuple__ = version_tuple = (0, 3, 3)
 
 __commit_id__ = commit_id = None

data_designer/engine/dataset_builders/utils/concurrency.py

@@ -108,6 +108,26 @@ class ConcurrentThreadExecutor:
         self._shutdown_window_size = shutdown_error_window
         self._results = ExecutorResults(failure_threshold=shutdown_error_rate)
 
+    @property
+    def results(self) -> ExecutorResults:
+        return self._results
+
+    @property
+    def max_workers(self) -> int:
+        return self._max_workers
+
+    @property
+    def shutdown_error_rate(self) -> float:
+        return self._shutdown_error_rate
+
+    @property
+    def shutdown_window_size(self) -> int:
+        return self._shutdown_window_size
+
+    @property
+    def semaphore(self) -> Semaphore:
+        return self._semaphore
+
     def __enter__(self) -> ConcurrentThreadExecutor:
         self._executor = ThreadPoolExecutor(
             max_workers=self._max_workers,
@@ -145,14 +165,16 @@ class ConcurrentThreadExecutor:
             self._raise_task_error()
 
         def _handle_future(future: Future) -> None:
-            self._results.completed_count += 1
             try:
                 result = future.result()
                 if self._result_callback is not None:
                     self._result_callback(result, context=context)
-                self._results.success_count += 1
+                with self._lock:
+                    self._results.completed_count += 1
+                    self._results.success_count += 1
             except Exception as err:
                 with self._lock:
+                    self._results.completed_count += 1
                     self._results.error_trap.handle_error(err)
                     if self._results.is_error_rate_exceeded(self._shutdown_window_size):
                         # Signal to shutdown early on the next submission (if received).

{data_designer-0.3.2.dist-info → data_designer-0.3.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: data-designer
-Version: 0.3.2
+Version: 0.3.3
 Summary: General framework for synthetic data generation
 License-Expression: Apache-2.0
 License-File: LICENSE
@@ -37,7 +37,7 @@ Requires-Dist: pygments<3,>=2.19.2
 Requires-Dist: python-json-logger<4,>=3
 Requires-Dist: pyyaml<7,>=6.0.1
 Requires-Dist: requests<3,>=2.32.2
-Requires-Dist: rich<14,>=13.7.1
+Requires-Dist: rich<15,>=13.7.1
 Requires-Dist: ruff<1,>=0.14.10
 Requires-Dist: scipy<2,>=1.11.0
 Requires-Dist: sqlfluff<4,>=3.2.0

{data_designer-0.3.2.dist-info → data_designer-0.3.3.dist-info}/RECORD

@@ -1,5 +1,5 @@
 data_designer/__init__.py,sha256=xYZCBRleSswcNCARfHQzFy_Edag3Nmmwfa6A5C1d5B8,466
-data_designer/_version.py,sha256=e8NqPtZ8fggRgk3GPrqZ_U_BDV8aSULw1u_Gn9NNbnk,704
+data_designer/_version.py,sha256=lemL_4Kl75FgrO6lVuFrrtw6-Dcf9wtXBalKkXuzkO4,704
 data_designer/errors.py,sha256=BjnxDxwtTGscY3lZbi1RQ666j5PseoxNJRobeMXAJiI,184
 data_designer/logging.py,sha256=2ToJzPPb6zF-QR-yQnrLVcBiBGL_QVNsnCHSQC0nPpQ,5346
 data_designer/plugin_manager.py,sha256=C2ZkZiXlcMRiaxfrrho5Shz6DKdExVeBha7ch-d4CnU,2695
@@ -103,7 +103,7 @@ data_designer/engine/dataset_builders/column_wise_builder.py,sha256=y068Eza-MabP
 data_designer/engine/dataset_builders/errors.py,sha256=ov9cTRvLtLieIFkUGZdk1n_iabdc904ZATwgT5u9uzY,364
 data_designer/engine/dataset_builders/multi_column_configs.py,sha256=bxRildX3SfzSFKv_rqwwsUeFn_RoyhfFneSj4qBi-Q4,1624
 data_designer/engine/dataset_builders/utils/__init__.py,sha256=XLO09Ei8g0lU7hYlzKCvhvQhLFBe5CBwE4v2PqK9xWY,142
-data_designer/engine/dataset_builders/utils/concurrency.py,sha256=DtC0ZNjvbdi9LW4po4pPu4bbcGzmdgDqVAJ3ikxEXrQ,7341
+data_designer/engine/dataset_builders/utils/concurrency.py,sha256=SQlT9Tu3UgATllm_fwle4kkro09NQBVU_nLSZj7lBDQ,7893
 data_designer/engine/dataset_builders/utils/config_compiler.py,sha256=iAbaLiDNBPyjZwSVK1a83KFKln9LACjKdVSpIiZRemw,2405
 data_designer/engine/dataset_builders/utils/dag.py,sha256=L3-sla2s8oqcy3V-WSxXAWZVakkmb3cvmTh5L6phC1M,2474
 data_designer/engine/dataset_builders/utils/dataset_batch_manager.py,sha256=c-6ER96oP-0oZ6CncN8sjXy1S1ZjzTreeYBr1oOX8Yg,7774
@@ -186,8 +186,8 @@ data_designer/plugins/registry.py,sha256=1vNlmDyFMCIY8D_z1RbnWMfKnHZ02g8sAvJg1nE
 data_designer/plugins/testing/__init__.py,sha256=lSuWzt1AVTJg4gV9wh6BUU1Az7IjQ-9FgADdUAtm_qQ,260
 data_designer/plugins/testing/stubs.py,sha256=E8bovwU4zpNaJM4b1i4biCDXyqvthe8vmsrZ9M0M9vM,4277
 data_designer/plugins/testing/utils.py,sha256=OLoQzW8-qbA-91wTUWtwNHrMDyGz1-ma0f6a-3NBUNI,937
-data_designer-0.3.2.dist-info/METADATA,sha256=MxCBgRT_p5AKQwJYXthjMIqJ4adlPHHyKd-gOft42bQ,7648
-data_designer-0.3.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-data_designer-0.3.2.dist-info/entry_points.txt,sha256=NWWWidyDxN6CYX6y664PhBYMhbaYTQTyprqfYAgkyCg,57
-data_designer-0.3.2.dist-info/licenses/LICENSE,sha256=cSWJDwVqHyQgly8Zmt3pqXJ2eQbZVYwN9qd0NMssxXY,11336
-data_designer-0.3.2.dist-info/RECORD,,
+data_designer-0.3.3.dist-info/METADATA,sha256=zJ5wtn9YBRb9qEXBHT3_93rZxo7RT5P6JhSsAdILimQ,7648
+data_designer-0.3.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+data_designer-0.3.3.dist-info/entry_points.txt,sha256=NWWWidyDxN6CYX6y664PhBYMhbaYTQTyprqfYAgkyCg,57
+data_designer-0.3.3.dist-info/licenses/LICENSE,sha256=cSWJDwVqHyQgly8Zmt3pqXJ2eQbZVYwN9qd0NMssxXY,11336
+data_designer-0.3.3.dist-info/RECORD,,