chalkpy 2.98.3__py3-none-any.whl → 2.98.5__py3-none-any.whl

chalk/_version.py

@@ -1 +1 @@
-__version__ = "2.98.3"
+__version__ = "2.98.5"

chalk/client/client.py

@@ -63,7 +63,7 @@ if TYPE_CHECKING:
 from chalk.features._encoding.json import FeatureEncodingOptions
 from chalk.features.resolver import Resolver
 from chalk.features.tag import BranchId, DeploymentId, EnvironmentId
-from chalk.ml import ModelEncoding, ModelRunCriterion, ModelType
+from chalk.ml import ModelClass, ModelEncoding, ModelRunCriterion, ModelType
 from chalk.parsed.branch_state import BranchGraphSummary
 from chalk.prompts import Prompt
 
@@ -2271,6 +2271,7 @@ class ChalkClient:
         self,
         name: str,
         model_type: Optional[ModelType] = None,
+        model_class: Optional[ModelClass] = None,
         model_encoding: Optional[ModelEncoding] = None,
         aliases: Optional[List[str]] = None,
         model: Optional[Any] = None,

chalk/client/client_impl.py

@@ -162,7 +162,7 @@ from chalk.features.pseudofeatures import CHALK_TS_FEATURE
 from chalk.features.resolver import Resolver, StreamResolver
 from chalk.features.tag import BranchId, DeploymentId, EnvironmentId
 from chalk.importer import CHALK_IMPORT_FLAG
-from chalk.ml import ModelEncoding, ModelRunCriterion, ModelType
+from chalk.ml import ModelClass, ModelEncoding, ModelRunCriterion, ModelType
 from chalk.ml.model_file_transfer import SourceConfig
 from chalk.parsed._proto.utils import encode_proto_to_b64
 from chalk.parsed.branch_state import BranchGraphSummary
@@ -5270,6 +5270,7 @@ https://docs.chalk.ai/cli/apply
         self,
         name: str,
         model_type: Optional[ModelType] = None,
+        model_class: Optional[ModelClass] = None,
         model_encoding: Optional[ModelEncoding] = None,
         aliases: Optional[List[str]] = None,
         model: Optional[Any] = None,
@@ -5296,6 +5297,7 @@ https://docs.chalk.ai/cli/apply
             name=name,
             aliases=aliases,
             model_type=model_type,
+            model_class=model_class,
             model_encoding=model_encoding,
             model=model,
             model_paths=model_paths,

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.3.dist-info → chalkpy-2.98.5.dist-info}/METADATA

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

{chalkpy-2.98.3.dist-info → chalkpy-2.98.5.dist-info}/RECORD

@@ -1,5 +1,5 @@
 chalk/__init__.py,sha256=vKsx9-cl5kImlVWGHVRYO6bweBm79NAzGs3l36u71wM,2657
-chalk/_version.py,sha256=Bz_5zhHPwDjTPM5Kpa6MCRtG34E0bkBIbT_od0j6sq8,23
+chalk/_version.py,sha256=tszjNZxNaUifHB5tE2pbEFfmv3R_0ZShqqHELZSpWIQ,23
 chalk/cli.py,sha256=ckqqfOI-A2mT23-rnZzDMmblYj-2x1VBX8ebHlIEn9A,5873
 chalk/importer.py,sha256=m4lMn1lSYj_euDq8CS7LYTBnek9JOcjGJf9-82dJHbA,64441
 chalk/prompts.py,sha256=2H9UomLAamdfRTNUdKs9i3VTpiossuyRhntqsAXUhhg,16117
@@ -612,11 +612,11 @@ chalk/_validation/validation.py,sha256=9cCMfZa9-1wxkXLme_ylmD5vIA1qExJD6aqbYvbmK
 chalk/byte_transmit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chalk/byte_transmit/model.py,sha256=LFX8pj9X_CWXeap7fDnMl9YmXsYTgq7jBAbEWkxoYoE,13048
 chalk/client/__init__.py,sha256=wu3WQVzE5gRj6noQQDOdYJUgDaz_9QtbjXH4KuoIlXQ,1782
-chalk/client/client.py,sha256=DcRJltVqWy_B3Qvbnjz3qOECT9TZ7KD0eovds2CNRIw,104746
+chalk/client/client.py,sha256=VfaLfbWyk-VaImPLNrNeAP4FvuJwQ5yEtJrkLST_-H4,104808
 chalk/client/client_async.py,sha256=YmuExumlDkenlGc8ROtmPjUE2m6nQUZisUDvJmwTRdk,51546
 chalk/client/client_async_impl.py,sha256=ZphhgTB49JBWHCGXe-dI0wWWKc9zPcOczy02q_gFy50,6925
 chalk/client/client_grpc.py,sha256=z1WeFWjmKKCU8dK7fdvWEOsCUlBi-v_2-HijC2_oIVs,108854
-chalk/client/client_impl.py,sha256=m2-dovRppOchuEwt2LlKJAv0LBW5obs9ztuJeDhuqY0,213660
+chalk/client/client_impl.py,sha256=L0qE7Gr5MTMv71lg7-HkoPS672eERGVsfu8FObM_ZyQ,213759
 chalk/client/dataset.py,sha256=LneWwaAOHCjtj7gaJjsSeVNruj-QJ51hjRi62zrFNVE,77561
 chalk/client/exc.py,sha256=kZJ80YbSeSRDmTLTh240j_eRdJFZBa7IaDsNSRoDroU,4145
 chalk/client/models.py,sha256=TWiRHDMP_9NUqlCAGaw2RI2UlFI5ED1BCBM1rZRK72k,67482
@@ -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
@@ -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.3.dist-info/METADATA,sha256=rhMVAP_p_UvatDRbPlTqI6QNHtIg3fNOioqUZ3HCwJ0,27754
-chalkpy-2.98.3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-chalkpy-2.98.3.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
-chalkpy-2.98.3.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
-chalkpy-2.98.3.dist-info/RECORD,,
+chalkpy-2.98.5.dist-info/METADATA,sha256=JSI5DNc3rEb8L4zKLTTNPL2y6uloR3VWxgg5-R_QWiE,27754
+chalkpy-2.98.5.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+chalkpy-2.98.5.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
+chalkpy-2.98.5.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
+chalkpy-2.98.5.dist-info/RECORD,,