dycw-utilities 0.133.2__tar.gz → 0.133.6__tar.gz

{dycw_utilities-0.133.2 → dycw_utilities-0.133.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.133.2
+Version: 0.133.6
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12
@@ -17,6 +17,7 @@ Requires-Dist: pytest-asyncio<1.1,>=1.0.0; extra == 'test'
 Requires-Dist: pytest-cov<6.2,>=6.1.1; extra == 'test'
 Requires-Dist: pytest-instafail<0.6,>=0.5.0; extra == 'test'
 Requires-Dist: pytest-lazy-fixtures<1.2,>=1.1.4; extra == 'test'
+Requires-Dist: pytest-randomly<3.17,>=3.16.0; extra == 'test'
 Requires-Dist: pytest-regressions<2.9,>=2.8.0; extra == 'test'
 Requires-Dist: pytest-rerunfailures<16,>=15.1; extra == 'test'
 Requires-Dist: pytest-rng<1.1,>=1.0.0; extra == 'test'

{dycw_utilities-0.133.2 → dycw_utilities-0.133.6}/pyproject.toml

@@ -53,7 +53,7 @@ memory-profiler = ["memory-profiler >= 0.61.0, < 0.62"]
 more-itertools = ["more-itertools >= 10.7.0, < 10.8"]
 numpy = ["numpy >= 2.3.0, < 2.4"]
 operator = ["polars-lts-cpu"]
-optuna = ["optuna >= 4.3.0, < 4.4"]
+optuna = ["optuna >= 4.4.0, < 4.5"]
 orjson = ["orjson >= 3.10.18, < 3.11"]
 orjson-test = ["polars-lts-cpu"]
 polars = ["polars-lts-cpu >= 1.30.0, < 1.31"]
@@ -104,7 +104,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.133.2"
+version = "0.133.6"
 
 [project.optional-dependencies]
 logging = [
@@ -118,6 +118,7 @@ test = [
   "pytest-cov >= 6.1.1, < 6.2",
   "pytest-instafail >= 0.5.0, < 0.6",
   "pytest-lazy-fixtures >= 1.1.4, < 1.2",
+  "pytest-randomly >= 3.16.0, < 3.17",
   "pytest-regressions >= 2.8.0, < 2.9",
   "pytest-rerunfailures >= 15.1, < 16",
   "pytest-rng >= 1.0.0, < 1.1",
@@ -130,7 +131,7 @@ test = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.133.2"
+current_version = "0.133.6"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.133.2 → dycw_utilities-0.133.6}/src/tests/test_sqlalchemy.py

@@ -274,17 +274,17 @@ class TestEnsureTablesDropped:
 
 
 class TestGetChunkSize:
-    @given(chunk_size_frac=floats(0.0, 1.0), scaling=floats(0.1, 10.0))
+    @given(chunk_size_frac=floats(0.0, 1.0), scaling=integers(0, 10))
     @settings(
         max_examples=1,
         phases={Phase.generate},
         suppress_health_check={HealthCheck.function_scoped_fixture},
     )
     async def test_main(
-        self, *, chunk_size_frac: float, scaling: float, test_engine: AsyncEngine
+        self, *, chunk_size_frac: float, scaling: int, test_engine: AsyncEngine
     ) -> None:
         result = get_chunk_size(
-            test_engine, chunk_size_frac=chunk_size_frac, scaling=scaling
+            test_engine, chunk_size_frac=chunk_size_frac, max_length=scaling
         )
         assert result >= 1
 

{dycw_utilities-0.133.2 → dycw_utilities-0.133.6}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.133.2"
+__version__ = "0.133.6"

{dycw_utilities-0.133.2 → dycw_utilities-0.133.6}/src/utilities/sqlalchemy.py

@@ -298,11 +298,13 @@ def get_chunk_size(
     /,
     *,
     chunk_size_frac: float = CHUNK_SIZE_FRAC,
-    scaling: float = 1.0,
+    max_length: int = 1,
 ) -> int:
     """Get the maximum chunk size for an engine."""
     max_params = _get_dialect_max_params(engine_or_conn)
-    return max(floor(chunk_size_frac * max_params / scaling), 1)
+    scaling = max(max_length, 1)
+    size = floor(chunk_size_frac * max_params / scaling)
+    return max(size, 1)
 
 
 ##
@@ -1098,7 +1100,7 @@ def _prepare_insert_or_upsert_items(
     }
     max_length = max(lengths, default=1)
     chunk_size = get_chunk_size(
-        engine, chunk_size_frac=chunk_size_frac, scaling=max_length
+        engine, chunk_size_frac=chunk_size_frac, max_length=max_length
     )
 
     def yield_pairs() -> Iterator[tuple[Insert, None]]:

{dycw_utilities-0.133.2 → dycw_utilities-0.133.6}/src/utilities/sqlalchemy_polars.py

@@ -433,7 +433,7 @@ def _select_to_dataframe_yield_selects_with_in_clauses(
     in_col, in_values = in_clauses
     if in_clauses_chunk_size is None:
         chunk_size = get_chunk_size(
-            engine, chunk_size_frac=chunk_size_frac, scaling=max_length
+            engine, chunk_size_frac=chunk_size_frac, max_length=max_length
         )
     else:
         chunk_size = in_clauses_chunk_size

{dycw_utilities-0.133.2 → dycw_utilities-0.133.6}/src/utilities/types.py

@@ -82,6 +82,9 @@ TCallable2 = TypeVar("TCallable2", bound=Callable[..., Any])
 TCallableCoroutine1 = TypeVar(
     "TCallableCoroutine1", bound=Callable[..., Coroutine1[Any]]
 )
+TCallableCoroutine1None = TypeVar(
+    "TCallableCoroutine1None", bound=Callable[..., Coroutine1[None]]
+)
 TCallableMaybeCoroutine1None = TypeVar(
     "TCallableMaybeCoroutine1None", bound=Callable[..., MaybeCoroutine1[None]]
 )
@@ -352,6 +355,7 @@ __all__ = [
     "TCallable1",
     "TCallable2",
     "TCallableCoroutine1",
+    "TCallableCoroutine1None",
     "TCallableMaybeCoroutine1None",
     "TDataclass",
     "TEnum",