chalkpy 2.98.4__py3-none-any.whl → 2.98.6__py3-none-any.whl

chalk/_version.py

@@ -1 +1 @@
-__version__ = "2.98.4"
+__version__ = "2.98.6"

chalk/features/_encoding/protobuf.py

@@ -37,11 +37,13 @@ def _get_pyarrow_type_for_proto_field(field_descriptor: FieldDescriptor) -> pa.D
         FieldDescriptor.TYPE_UINT64: pa.uint64(),
         FieldDescriptor.TYPE_INT32: pa.int32(),
         FieldDescriptor.TYPE_FIXED64: pa.uint64(),
-        FieldDescriptor.TYPE_FIXED32: pa.uint32(),
+        # Velox doesn't support unsigned ints, so we fit the uint32's into int64's
+        FieldDescriptor.TYPE_FIXED32: pa.int64(),
         FieldDescriptor.TYPE_BOOL: pa.bool_(),
         FieldDescriptor.TYPE_STRING: pa.large_utf8(),
         FieldDescriptor.TYPE_BYTES: pa.binary(),
-        FieldDescriptor.TYPE_UINT32: pa.uint32(),
+        # Velox doesn't support unsigned ints, so we fit the uint32's into int64's
+        FieldDescriptor.TYPE_UINT32: pa.int64(),
         FieldDescriptor.TYPE_SFIXED32: pa.int32(),
         FieldDescriptor.TYPE_SFIXED64: pa.int64(),
         FieldDescriptor.TYPE_SINT32: pa.int32(),

chalk/features/resolver.py

@@ -4589,7 +4589,7 @@ def make_model_resolver(
     name: str,
     model: "ModelVersion",
     input: Dict[Feature, str] | List[Feature],
-    output: Feature | List[Feature] | Dict[Feature, str],
+    output: Feature | List[Feature] | Dict[Feature, str] | Any,
     feature_class: Optional[type[Features]] = None,
     resource_group: Optional[str] = None,
     resource_hint: Optional[ResourceHint] = None,
@@ -4608,7 +4608,7 @@ def make_model_resolver(
         The name of the resolver
     model
         A ModelVersion reference to a deployed model
-    inputs
+    input
         Either a dict mapping Feature objects to model input names (strings), or a list of
         Feature objects. If a dict, the values represent the model's expected input names
         (for future use). If a list, the features will be passed as a single DataFrame to
@@ -4746,6 +4746,7 @@ def make_model_resolver(
         tags=None,
         owner=None,
         resource_hint=resource_hint or model.resource_hint,
+        resource_group=resource_group or model.resource_group,
         data_sources=None,
         is_sql_file_resolver=False,
         source_line=None,

{chalkpy-2.98.4.dist-info → chalkpy-2.98.6.dist-info}/METADATA

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

{chalkpy-2.98.4.dist-info → chalkpy-2.98.6.dist-info}/RECORD

@@ -1,5 +1,5 @@
 chalk/__init__.py,sha256=vKsx9-cl5kImlVWGHVRYO6bweBm79NAzGs3l36u71wM,2657
-chalk/_version.py,sha256=dEzljEJtCL38cLmlHSmH5pTwojeTArPqqy8VF1ezkbo,23
+chalk/_version.py,sha256=lCU8-hElWBDbqghv9vMcFvFH7YnIyamLkmO8KLLgp4g,23
 chalk/cli.py,sha256=ckqqfOI-A2mT23-rnZzDMmblYj-2x1VBX8ebHlIEn9A,5873
 chalk/importer.py,sha256=m4lMn1lSYj_euDq8CS7LYTBnek9JOcjGJf9-82dJHbA,64441
 chalk/prompts.py,sha256=2H9UomLAamdfRTNUdKs9i3VTpiossuyRhntqsAXUhhg,16117
@@ -660,7 +660,7 @@ chalk/features/live_updates.py,sha256=8ZbiDjcLqfFruSL15_aycwzSqJ0TbKNhas06KfZLyL
 chalk/features/namespace_context.py,sha256=fL-nPohqtNiyPDS1uQTAaHLns4aivuBL2Flf50DajU4,1813
 chalk/features/primary.py,sha256=BZ8mrMmKfRNy_wnKGDJt2cdnejP_CZb6xBsD9Ljgajc,5209
 chalk/features/pseudofeatures.py,sha256=50Pe_Xi8ttYWtgNNRpgkhBxP8xoCZCYwyLb0aWUQ-PI,2147
-chalk/features/resolver.py,sha256=txAhcTv60QHn5Ge0Hr1z2LSWVKNDuZZYSogTnY31YMA,193542
+chalk/features/resolver.py,sha256=Et-cngxAjqbstJC5lBD3AhMwzXvUjQ6Tj8FnnW5bACU,193610
 chalk/features/tag.py,sha256=LRmKRA8ANCOvmaIAk-L5j1QW2U0aah2SeASy8Uydkmk,2675
 chalk/features/underscore.py,sha256=4xnfQV3bfvVn0PNEtkT4J-k7hW4ebtH9KBe4_BvGjY4,26763
 chalk/features/underscore_features.py,sha256=PlVCoaDDffOgtiSMaxPSWCoj8IjscbkOzDLA471HsJ4,13005
@@ -680,7 +680,7 @@ chalk/features/_encoding/json.py,sha256=lHSBWIjNKMDmt_AmiFtf4ZsW5QPIEC6r7NvDW0Id
 chalk/features/_encoding/missing_value.py,sha256=EacZ9T4lhlNbKJe7n8fQn_37TxWglqcBHx427ek80CY,508
 chalk/features/_encoding/outputs.py,sha256=W9XQCs-YzWqZKs1rXjMmXtN00kFoMjsITPWQ0pTV32E,5119
 chalk/features/_encoding/primitive.py,sha256=8ECCEOwPRGUpFEUBK0nt_QWVDu3gB5NvPnBMkHyrnjo,1226
-chalk/features/_encoding/protobuf.py,sha256=zwIu7x0UrLPvSyG2YJW5UVKL67rpp69h0kZgpVj1pWA,5971
+chalk/features/_encoding/protobuf.py,sha256=_GCDL9hwREZgOSNrsTwT3OkC5foJYVxv_G5_wN6h0bk,6135
 chalk/features/_encoding/pyarrow.py,sha256=Psj-KrNc0pkSjSxiFKiQcBpwAo5U7uJAc6o-ZNDgc2c,19807
 chalk/features/_encoding/rich.py,sha256=Tln2iwnhJirsPaVFx0Kbyc89CWcCccWiNMgV1m-8Yr4,16177
 chalk/features/_encoding/serialized_dtype.py,sha256=HttVeCAqR-tU4jlq_AA_G61Z67t86gecIMMMnpg95IY,11124
@@ -827,8 +827,8 @@ chalk/utils/tracing.py,sha256=NiiM-9dbuJhSCv6R1npR1uYNKWlkqTR6Ygm0Voi2NrY,13078
 chalk/utils/weak_set_by_identity.py,sha256=VmikA_laYwFeOphCwXJIuyOIkrdlQe0bSzaXq7onoQw,953
 chalk/utils/pydanticutil/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chalk/utils/pydanticutil/pydantic_compat.py,sha256=O575lLYJ5GvZC4HMzR9yATxf9XwjC6NrDUXbNwZidlE,3031
-chalkpy-2.98.4.dist-info/METADATA,sha256=A7vXH5sot4cUwTZRmRIJoNCUz6UciNohn3Xvitpuqgk,27754
-chalkpy-2.98.4.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-chalkpy-2.98.4.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
-chalkpy-2.98.4.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
-chalkpy-2.98.4.dist-info/RECORD,,
+chalkpy-2.98.6.dist-info/METADATA,sha256=_FtE0_mtvo_reEoEYa0_LgLU4Dn5-3vO0Y20jvu-4sU,27754
+chalkpy-2.98.6.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+chalkpy-2.98.6.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
+chalkpy-2.98.6.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
+chalkpy-2.98.6.dist-info/RECORD,,