chalkruntime 3.35.0__tar.gz → 3.35.2__tar.gz

{chalkruntime-3.35.0 → chalkruntime-3.35.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chalkruntime
-Version: 3.35.0
+Version: 3.35.2
 Summary: Runtime support library for Chalk AI
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{chalkruntime-3.35.0 → chalkruntime-3.35.2}/chalkruntime/graph/convert_chalkpy_underscore.py

@@ -2493,12 +2493,15 @@ def _convert_underscore_expression(
             # Create an operation expression for catalog_call
             operands = MaybeNamedCollectionBuilder(positional_items=parsed_args)
 
+            policies = getattr(underscore, "_chalk__policies", None) or []
+
             operation = UnderscoreOperationRegistry.resolve(
                 "catalog_call",
                 frozendict(
                     {
                         "qualified_name": qualified_name,
                         "output_type": output_dtype,
+                        "policies": policies,
                     }
                 ),
             )
@@ -2536,12 +2539,15 @@ def _convert_underscore_expression(
 
             operands = MaybeNamedCollectionBuilder(positional_items=parsed_args)
 
+            policies = getattr(underscore, "_chalk__policies", None) or []
+
             operation = UnderscoreOperationRegistry.resolve(
                 "call_resolver",
                 frozendict(
                     {
                         "resolver_fqn": resolver_fqn_arg,
                         "output_type": output_type_arg,
+                        "policies": policies,
                     }
                 ),
             )

{chalkruntime-3.35.0 → chalkruntime-3.35.2}/chalkruntime/graph/feature.py

@@ -1402,7 +1402,12 @@ class WindowedFeatureType(_CompatibleWithNestedFeatureTypeMixin, NamedFeatureTyp
         pseudos = []
         for w in self.windows:
             name = get_name_with_duration(self.fqn, w)
-            f = self.graph.fqn_to_feature[name]
+            # Some windows only exist for a specific version (e.g. a "90d" bucket added in v2
+            # won't have a bare pseudo-feature for the v1 root). Skip missing entries rather
+            # than raising KeyError.
+            f = self.graph.fqn_to_feature.get(name)
+            if f is None:
+                continue
             assert isinstance(f, ScalarFeatureType)
             pseudos.append(f)
         return pseudos

{chalkruntime-3.35.0 → chalkruntime-3.35.2}/chalkruntime/graph/underscore_operation_registry.py

@@ -1,4 +1,6 @@
-from typing import Any, Sequence, Type
+import json
+from collections.abc import Mapping, Sequence
+from typing import Any, Type, cast
 
 import pyarrow as pa
 import pyarrow.compute as pc
@@ -8,6 +10,40 @@ from chalkruntime.graph.underscore import UnderscoreOperation
 from chalkruntime.graph.underscore_codec_info import FromPythonRegCodecInfo
 
 
+def _blocking_policies_from_params(fn_name: str, params: frozendict[str, object]) -> list[dict[str, Any]]:
+    policies_json = params.get("policies_json")
+    if policies_json is not None:
+        if not isinstance(policies_json, str):
+            raise ValueError(
+                f"UnderscoreOperationRegistry: while resolving '{fn_name}', expect parameter policies_json to be of type {str} but instead got {policies_json} of type {type(policies_json).__name__}"
+            )
+        policies = json.loads(policies_json)
+    else:
+        policies = params.get("policies", [])
+
+    if policies is None:
+        return []
+    if not isinstance(policies, Sequence) or isinstance(policies, (str, bytes)):
+        raise ValueError(
+            f"UnderscoreOperationRegistry: while resolving '{fn_name}', expect parameter policies to be of type {Sequence} but instead got {policies} of type {type(policies).__name__}"
+        )
+
+    normalized: list[dict[str, Any]] = []
+    for i, policy in enumerate(policies):
+        if not isinstance(policy, Mapping):
+            raise ValueError(
+                f"UnderscoreOperationRegistry: while resolving '{fn_name}', expect policies[{i}] to be a mapping but got {policy} of type {type(policy).__name__}"
+            )
+        normalized.append(dict(cast(Mapping[str, Any], policy)))
+    return normalized
+
+
+def _policy_codec_params(policies: list[dict[str, Any]]) -> dict[str, str]:
+    if not policies:
+        return {}
+    return {"policies_json": json.dumps(policies, sort_keys=True, separators=(",", ":"))}
+
+
 class UnderscoreOperationRegistry:
     @staticmethod
     def resolve(function_name: str, params: frozendict[str, object]):
@@ -123,11 +159,13 @@ class UnderscoreOperationRegistry:
             ),
         )
 
+        struct_fields: list[pa.Field] = [
+            pa.field(cast(str, name), cast(pa.DataType, type_val)) for name, type_val in zip(names, types)
+        ]
+
         return UnderscoreOperation(
             operation_name="struct_pack",
-            return_type=lambda _: pa.struct(
-                {name: type for name, type in zip(names, types)}  # ty: ignore[invalid-argument-type]
-            ),
+            return_type=lambda _: pa.struct(struct_fields),
             fn=_struct_pack_python_op,
             codec_info=codec_info,
         )
@@ -212,12 +250,15 @@ class UnderscoreOperationRegistry:
                 )
             )
 
+        policies = _blocking_policies_from_params("catalog_call", params)
+
         codec_info = FromPythonRegCodecInfo(
             function_name="catalog_call",
             params=frozendict(
                 {
                     "qualified_name": qualified_name,
                     "output_type": output_type,
+                    **_policy_codec_params(policies),
                 }
             ),
         )
@@ -263,12 +304,15 @@ class UnderscoreOperationRegistry:
                 )
             )
 
+        policies = _blocking_policies_from_params("call_resolver", params)
+
         codec_info = FromPythonRegCodecInfo(
             function_name="call_resolver",
             params=frozendict(
                 {
                     "resolver_fqn": resolver_fqn,
                     "output_type": output_type,
+                    **_policy_codec_params(policies),
                 }
             ),
         )

{chalkruntime-3.35.0 → chalkruntime-3.35.2}/chalkruntime.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chalkruntime
-Version: 3.35.0
+Version: 3.35.2
 Summary: Runtime support library for Chalk AI
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{chalkruntime-3.35.0 → chalkruntime-3.35.2}/pyproject.toml

@@ -42,7 +42,7 @@ name = "chalkruntime"
 description = "Runtime support library for Chalk AI"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "3.35.0"
+version = "3.35.2"
 
 
 [tool.deptry]