chalkpy 2.94.0__tar.gz → 2.94.1__tar.gz

{chalkpy-2.94.0 → chalkpy-2.94.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chalkpy
-Version: 2.94.0
+Version: 2.94.1
 Summary: Python SDK for Chalk
 Author: Chalk AI, Inc.
 Project-URL: Homepage, https://chalk.ai

chalkpy-2.94.1/chalk/_version.py

@@ -0,0 +1 @@
+__version__ = "2.94.1"

{chalkpy-2.94.0 → chalkpy-2.94.1}/chalk/importer.py

@@ -304,6 +304,19 @@ def _parse_agg_function_call(expr: Underscore | None) -> Tuple[str, Underscore,
                 f"expecting 'int' type argument for 'k', but received arg of type '{type(call_expr._chalk__kwargs.get('k'))}'"
             )
         opts = FrozenOrderedSet(call_expr._chalk__kwargs.items())
+    elif aggregation == "approx_percentile":
+        if len(call_expr._chalk__args) > 0:
+            raise ChalkParseError("should not have any positional arguments")
+        elif {"quantile"} != call_expr._chalk__kwargs.keys():
+            raise ChalkParseError("expecting exactly one required keyword argument 'quantile'")
+        elif not isinstance(call_expr._chalk__kwargs.get("quantile"), float):
+            raise ChalkParseError(
+                f"expecting 'float' type argument for 'quantile', but received arg of type '{type(call_expr._chalk__kwargs.get('quantile'))}'"
+            )
+        # TODO: expand proto definition to accept kwargs that are not necessarily `k`
+        quantile = call_expr._chalk__kwargs["quantile"]
+        nano_quantile = int(round(quantile * 1_000_000_000))
+        opts = FrozenOrderedSet([("k", nano_quantile)])
     elif aggregation in ("min_by_n", "max_by_n"):
         if len(call_expr._chalk__kwargs) > 0:
             raise ChalkParseError("should not have any keyword arguments")
@@ -570,39 +583,51 @@ def parse_grouped_window(f: Feature) -> WindowConfigResolved:
         aggregation_kwargs=aggregation_kwargs,
         pyarrow_dtype=pyarrow_dtype,
         filters=parsed_filters,
-        backfill_resolver=_try_parse_resolver_fqn(
-            "backfill_resolver",
-            f.window_materialization.get("backfill_resolver", None),
-        )
-        if isinstance(f.window_materialization, dict)
-        else None,
-        backfill_schedule=f.window_materialization.get("backfill_schedule", None)
-        if isinstance(f.window_materialization, dict)
-        else None,
-        backfill_lookback_duration_seconds=_try_parse_duration(
-            "backfill_lookback_duration",
-            f.window_materialization.get("backfill_lookback_duration", None),
-        )
-        if isinstance(f.window_materialization, dict)
-        else None,
-        backfill_start_time=_try_parse_datetime(
-            "backfill_start_time",
-            f.window_materialization.get("backfill_start_time", None),
-        )
-        if isinstance(f.window_materialization, dict)
-        else None,
-        continuous_resolver=_try_parse_resolver_fqn(
-            "continuous_resolver",
-            f.window_materialization.get("continuous_resolver", None),
-        )
-        if isinstance(f.window_materialization, dict)
-        else None,
-        continuous_buffer_duration_seconds=_try_parse_duration(
-            "continuous_buffer_duration",
-            f.window_materialization.get("continuous_buffer_duration", None),
-        )
-        if isinstance(f.window_materialization, dict)
-        else None,
+        backfill_resolver=(
+            _try_parse_resolver_fqn(
+                "backfill_resolver",
+                f.window_materialization.get("backfill_resolver", None),
+            )
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
+        backfill_schedule=(
+            f.window_materialization.get("backfill_schedule", None)
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
+        backfill_lookback_duration_seconds=(
+            _try_parse_duration(
+                "backfill_lookback_duration",
+                f.window_materialization.get("backfill_lookback_duration", None),
+            )
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
+        backfill_start_time=(
+            _try_parse_datetime(
+                "backfill_start_time",
+                f.window_materialization.get("backfill_start_time", None),
+            )
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
+        continuous_resolver=(
+            _try_parse_resolver_fqn(
+                "continuous_resolver",
+                f.window_materialization.get("continuous_resolver", None),
+            )
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
+        continuous_buffer_duration_seconds=(
+            _try_parse_duration(
+                "continuous_buffer_duration",
+                f.window_materialization.get("continuous_buffer_duration", None),
+            )
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
     )
 
     return cfg
@@ -798,39 +823,51 @@ def parse_windowed_materialization(f: Feature) -> WindowConfigResolved | None:
         aggregation_kwargs=aggregation_kwargs,
         pyarrow_dtype=f.converter.pyarrow_dtype,
         filters=parsed_filters,
-        backfill_resolver=_try_parse_resolver_fqn(
-            "backfill_resolver",
-            f.window_materialization.get("backfill_resolver", None),
-        )
-        if isinstance(f.window_materialization, dict)
-        else None,
-        backfill_schedule=f.window_materialization.get("backfill_schedule", None)
-        if isinstance(f.window_materialization, dict)
-        else None,
-        backfill_lookback_duration_seconds=_try_parse_duration(
-            "backfill_lookback_duration",
-            f.window_materialization.get("backfill_lookback_duration", None),
-        )
-        if isinstance(f.window_materialization, dict)
-        else None,
-        backfill_start_time=_try_parse_datetime(
-            "backfill_start_time",
-            f.window_materialization.get("backfill_start_time", None),
-        )
-        if isinstance(f.window_materialization, dict)
-        else None,
-        continuous_resolver=_try_parse_resolver_fqn(
-            "continuous_resolver",
-            f.window_materialization.get("continuous_resolver", None),
-        )
-        if isinstance(f.window_materialization, dict)
-        else None,
-        continuous_buffer_duration_seconds=_try_parse_duration(
-            "continuous_buffer_duration",
-            f.window_materialization.get("continuous_buffer_duration", None),
-        )
-        if isinstance(f.window_materialization, dict)
-        else None,
+        backfill_resolver=(
+            _try_parse_resolver_fqn(
+                "backfill_resolver",
+                f.window_materialization.get("backfill_resolver", None),
+            )
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
+        backfill_schedule=(
+            f.window_materialization.get("backfill_schedule", None)
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
+        backfill_lookback_duration_seconds=(
+            _try_parse_duration(
+                "backfill_lookback_duration",
+                f.window_materialization.get("backfill_lookback_duration", None),
+            )
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
+        backfill_start_time=(
+            _try_parse_datetime(
+                "backfill_start_time",
+                f.window_materialization.get("backfill_start_time", None),
+            )
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
+        continuous_resolver=(
+            _try_parse_resolver_fqn(
+                "continuous_resolver",
+                f.window_materialization.get("continuous_resolver", None),
+            )
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
+        continuous_buffer_duration_seconds=(
+            _try_parse_duration(
+                "continuous_buffer_duration",
+                f.window_materialization.get("continuous_buffer_duration", None),
+            )
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
     )
 
 

{chalkpy-2.94.0 → chalkpy-2.94.1}/chalk/parsed/to_proto.py

@@ -900,7 +900,7 @@ class ToProtoConverter:
                     else None,
                     backfill_schedule=mat.backfill_schedule,
                     approx_top_k_arg_k=aggregation_kwargs.get("k")
-                    if mat.aggregation in ("approx_top_k", "min_by_n", "max_by_n")
+                    if mat.aggregation in ("approx_top_k", "approx_percentile", "min_by_n", "max_by_n")
                     else None,
                 ),
                 tags=f.tags,
@@ -996,7 +996,7 @@ class ToProtoConverter:
                             else None,
                             continuous_resolver=wmp.continuous_resolver,
                             approx_top_k_arg_k=aggregation_kwargs.get("k")
-                            if wmp.aggregation in ("approx_top_k", "min_by_n", "max_by_n")
+                            if wmp.aggregation in ("approx_top_k", "approx_percentile", "min_by_n", "max_by_n")
                             else None,
                         )
                         if wmp is not None

chalkpy-2.94.0/chalk/_version.py

@@ -1 +0,0 @@
-__version__ = "2.94.0"