chalkpy 2.94.9__py3-none-any.whl → 2.95.1__py3-none-any.whl

chalk/_version.py

@@ -1 +1 @@
-__version__ = "2.94.9"
+__version__ = "2.95.1"

chalk/client/client.py

@@ -2157,21 +2157,20 @@ class ChalkClient:
         name: str,
         version: Optional[int] = None,
     ) -> Union[GetRegisteredModelResponse, GetRegisteredModelVersionResponse]:
-        """
-        Retrieve a registered model from the Chalk model registry.
+        """Retrieve a registered model from the Chalk model registry.
 
         Parameters
         ----------
-        name : str
-           Name of the model to retrieve
-        version : int, optional
-           Specific version number to retrieve. If not provided, returns
-           information about all versions of the model
+        name
+            Name of the model to retrieve.
+        version
+            Specific version number to retrieve. If not provided, returns
+            information about all versions of the model.
 
         Returns
         -------
-        GetRegisteredModelResponse or GetRegisteredModelVersionResponse
-           Model information including metadata, versions, and configuration details
+        Union[GetRegisteredModelResponse, GetRegisteredModelVersionResponse]
+            Model information including metadata, versions, and configuration details.
 
         Examples
         --------
@@ -2248,49 +2247,49 @@ class ChalkClient:
         source_config: Optional[SourceConfig] = None,
         dependencies: Optional[List[str]] = None,
     ) -> RegisterModelVersionResponse:
-        """
-        Register a model in the Chalk model registry.
+        """Register a model in the Chalk model registry.
 
         Parameters
         ----------
-        name : str
-           Unique name for the model
-        aliases : list of str, optional
-           List of version aliases (e.g., ["v1.0", "latest"])
-        model : object, optional
-           Python model object (for object-based registration)
-        model_paths : list of str, optional
-           Paths to model files (for file-based registration)
-        additional_files : List[str], optional
-           Additional files needed for inference (tokenizers, configs, etc.)
-        model_type : ModelType, optional
-           Type of model framework
-        model_encoding : ModelEncoding, optional
-           Serialization format
-        input_schema : dict, list, or Any
-           Definition of the input schema. Can be:
-           - dict: Dictionary mapping column names to dtypes for tabular data
-           - list: List of (shape, dtype) tuples for tensor data
-        output_schema : dict, list, or Any
-           Definition of the output schema. Can be:
-           - dict: Dictionary mapping column names to dtypes for tabular data
-           - list: List of (shape, dtype) tuples for tensor data
-        metadata : dict, optional
-           Additional metadata dictionary containing framework info,
-           training details, performance metrics, etc.
-        input_features : FeatureReference, str, optional
+        name
+            Unique name for the model.
+        aliases
+            List of version aliases (e.g., `["v1.0", "latest"]`).
+        model
+            Python model object (for object-based registration).
+        model_paths
+            Paths to model files (for file-based registration).
+        additional_files
+            Additional files needed for inference (tokenizers, configs, etc.)
+        model_type
+            Type of model framework
+        model_encoding
+            Serialization format
+        input_schema
+            Definition of the input schema. Can be:
+            - `dict`: Dictionary mapping column names to dtypes for tabular data
+            - `list`: List of `(shape, dtype)` tuples for tensor data
+        output_schema
+            Definition of the output schema. Can be:
+            - `dict`: Dictionary mapping column names to dtypes for tabular data
+            - `list`: List of `(shape, dtype)` tuples for tensor data
+        metadata
+            Additional metadata dictionary containing framework info,
+            training details, performance metrics, etc.
+        input_features
             The features to be used as inputs to the model.
             For example, `[User.message]`. Features can also be expressed as snakecased strings,
-            e.g. `["user.message"]`
-        output_features : FeatureReference, str, optional
+            e.g. `["user.message"]`.
+        output_features
             The features to be used as outputs to the model.
             For example, `[User.is_spam]`. Features can also be expressed as snakecased strings,
-            e.g. `["user.is_spam"]`
-        source_config : LocalSourceConfig, S3SourceConfig, HFSourceConfig, optional
+            e.g. `["user.is_spam"]`.
+        source_config
             Config to pass credentials to access files from a remote source.
-        dependencies : List[str], optional
+        dependencies
             List of package dependencies needed to run this model.
-            e.g. ["torch==2.7.1", "numpy==1.26.4"]
+            e.g. `["torch==2.7.1", "numpy==1.26.4"]`.
+
         Returns
         -------
         ModelVersion

chalk/client/client_grpc.py

@@ -1647,53 +1647,53 @@ class ChalkGRPCClient:
         source_config: Optional[SourceConfig] = None,
         dependencies: Optional[List[str]] = None,
     ) -> RegisterModelVersionResponse:
-        """
-        Register a model in the Chalk model registry.
+        """Register a model in the Chalk model registry.
 
         Parameters
         ----------
-        name : str
-           Unique name for the model
-        aliases : list of str, optional
-           List of version aliases (e.g., ["v1.0", "latest"])
-        model : object, optional
-           Python model object (for object-based registration)
-        model_paths : list of str, optional
-           Paths to model files (for file-based registration)
-        additional_files : List[str], optional
-           Additional files needed for inference (tokenizers, configs, etc.)
-        model_type : ModelType, optional
-           Type of model framework
-        model_encoding : ModelEncoding, optional
-           Serialization format
-        input_schema : dict, list, or Any
-           Definition of the input schema. Can be:
-           - dict: Dictionary mapping column names to dtypes for tabular data
-           - list: List of (shape, dtype) tuples for tensor data
-        output_schema : dict, list, or Any
-           Definition of the output schema. Can be:
-           - dict: Dictionary mapping column names to dtypes for tabular data
-           - list: List of (shape, dtype) tuples for tensor data
-        metadata : dict, optional
-           Additional metadata dictionary containing framework info,
-           training details, performance metrics, etc.
-        input_features : FeatureReference, str, optional
+        name
+            Unique name for the model.
+        aliases
+            List of version aliases (e.g., `["v1.0", "latest"]`).
+        model
+            Python model object (for object-based registration).
+        model_paths
+            Paths to model files (for file-based registration).
+        additional_files
+            Additional files needed for inference (tokenizers, configs, etc.)
+        model_type
+            Type of model framework.
+        model_encoding
+            Serialization format.
+        input_schema
+            Definition of the input schema. Can be:
+            - `dict`: Dictionary mapping column names to dtypes for tabular data
+            - `list`: List of `(shape, dtype)` tuples for tensor data
+        output_schema
+            Definition of the output schema. Can be:
+            - `dict`: Dictionary mapping column names to dtypes for tabular data
+            - `list`: List of `(shape, dtype)` tuples for tensor data
+        metadata
+            Additional metadata dictionary containing framework info,
+            training details, performance metrics, etc.
+        input_features
             The features to be used as inputs to the model.
             For example, `[User.message]`. Features can also be expressed as snakecased strings,
-            e.g. `["user.message"]`
-        output_features : FeatureReference, str, optional
+            e.g. `["user.message"]`.
+        output_features
             The features to be used as outputs to the model.
             For example, `[User.is_spam]`. Features can also be expressed as snakecased strings,
-            e.g. `["user.is_spam"]`
-        source_config : LocalSourceConfig, S3SourceConfig, HFSourceConfig, optional
+            e.g. `["user.is_spam"]`.
+        source_config
             Config to pass credentials to access files from a remote source.
-        dependencies : List[str], optional
+        dependencies
             List of package dependencies needed to run this model.
-            e.g. ["torch==2.7.1", "numpy==1.26.4"]
+            e.g. `["torch==2.7.1", "numpy==1.26.4"]`.
+
         Returns
         -------
         ModelVersion
-           The registered model version object
+            The registered model version object.
 
         Examples
         --------

chalk/df/LazyFramePlaceholder.py

@@ -292,8 +292,6 @@ class LazyFramePlaceholder:
             Region of the Glue catalog.
         aws_role_arn
             IAM role to assume for access.
-        filter_predicate
-            Optional filter applied during scan.
         parquet_scan_range_column
             Column used for range-based reads.
         custom_partitions

chalk/features/feature_field.py

@@ -2199,6 +2199,7 @@ def has_many(
         The maximum number of items to cache for the joined feature. The
         items in the joined feature aggregate, storing the latest values
         of the joined feature for each primary key in the joined feature.
+
     Examples
     --------
     >>> from chalk.features import DataFrame, features, has_many

chalk/features/feature_set_decorator.py

@@ -141,6 +141,7 @@ def features(
         The `cache_nulls` and `cache_defaults` options can be used together on the same feature with the
         following exceptions: if `cache_nulls=False`, then `cache_defaults` cannot be `"evict_defaults"`, and if
         `cache_nulls="evict_defaults"`, then `cache_defaults` cannot be `False`.
+
     Other Parameters
     ----------------
     cls

chalk/functions/__init__.py

@@ -3947,7 +3947,6 @@ def array_filter(
     ...    id: str
     ...    recent_activities: list[float]
     ...    average_activity: float
-    ...
     ...    recent_high_value_activities: list[float] = F.array_filter(
     ...        _.recent_activities,
     ...        lambda amount: amount > _.average_activity,
@@ -4053,7 +4052,6 @@ def array_reduce(
     ... class User:
     ...    id: str
     ...    scores: list[int]
-    ...
     ...    total_score: int = F.array_reduce(
     ...        arr=_.scores,
     ...        initial_value=0,

{chalkpy-2.94.9.dist-info → chalkpy-2.95.1.dist-info}/METADATA

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

{chalkpy-2.94.9.dist-info → chalkpy-2.95.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
 chalk/__init__.py,sha256=vKsx9-cl5kImlVWGHVRYO6bweBm79NAzGs3l36u71wM,2657
-chalk/_version.py,sha256=GtEwETWuEgyr8yjecJU8KLv60qE8ba2NlN15NXFJ54Q,23
+chalk/_version.py,sha256=rk-4TApc5a3eBFZLyztEsm1bUXcfcBUXksYD8HUtU0k,23
 chalk/cli.py,sha256=ckqqfOI-A2mT23-rnZzDMmblYj-2x1VBX8ebHlIEn9A,5873
 chalk/importer.py,sha256=m4lMn1lSYj_euDq8CS7LYTBnek9JOcjGJf9-82dJHbA,64441
 chalk/prompts.py,sha256=2H9UomLAamdfRTNUdKs9i3VTpiossuyRhntqsAXUhhg,16117
@@ -603,10 +603,10 @@ 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=gdGHrC73xyC4boE8z54BDD3-7FMJ4WCv6o2I_ii8obI,103680
+chalk/client/client.py,sha256=fqw75x4yiAk3CXryGbuF_qg09WDIlluhYeFdQ-BfTG8,103337
 chalk/client/client_async.py,sha256=nFFTWJbdBlb7zksyjOMBY566tZTAyNXQhCnq06LHWl0,50803
 chalk/client/client_async_impl.py,sha256=ZphhgTB49JBWHCGXe-dI0wWWKc9zPcOczy02q_gFy50,6925
-chalk/client/client_grpc.py,sha256=7DzSUdYf7P2Y2lAo6AdLALEwwQ5YUNSeJU2rfk3KOec,95763
+chalk/client/client_grpc.py,sha256=zRqk6qV6UO-kHnn3Ann_ckva1MRf5YC1eDWxFO3x1Pc,95443
 chalk/client/client_impl.py,sha256=R2ZUiEQf662q3O1eo_NjzYqLGB71B5mWzcx8opC3Os0,211639
 chalk/client/dataset.py,sha256=LneWwaAOHCjtj7gaJjsSeVNruj-QJ51hjRi62zrFNVE,77561
 chalk/client/exc.py,sha256=kZJ80YbSeSRDmTLTh240j_eRdJFZBa7IaDsNSRoDroU,4145
@@ -627,7 +627,7 @@ chalk/config/_validator.py,sha256=QC1y52m704_bV_TYjq0sdZJ-km8iSkDX1V4sHgw4RJk,13
 chalk/config/auth_config.py,sha256=HAALkQrvDD0i7gaZK5iufS8vDDVbzLIpHLOpcJO1kmw,4498
 chalk/config/project_config.py,sha256=YHB3upvtBJu-tWfNOchBRSc9xGebDbrIpCVmKbBzJ8Q,7217
 chalk/df/ChalkDataFrameImpl.py,sha256=BRwnjQcie3gxaKhKau8HG17NWzS1zdr8SnNVurxF8QY,133
-chalk/df/LazyFramePlaceholder.py,sha256=Y9MQtBRMs-n2_N8UIs6LL-1gUWz2jsccDhDaWp2Ej7M,39185
+chalk/df/LazyFramePlaceholder.py,sha256=uTinB2buEFBT8djBCqiPN2hKmr_nU7n4i_DcCLNLuWs,39111
 chalk/df/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chalk/df/ast_parser.py,sha256=t-DwUxd2hN8LECRSBx85DIv9FtILgMiHyGyCTINfgQw,11199
 chalk/features/__init__.py,sha256=5doD7bFwRthzwdmizbRaPVHiCABiNpiOiAJVFlwqNnA,6943
@@ -639,9 +639,9 @@ chalk/features/_last.py,sha256=IyYe0PVAKBu8FRhCZiK6Dy5afKbwDHEUTES5qVXbaZU,1163
 chalk/features/_tensor.py,sha256=2SQz9cnIywGjHL6LoWj4lrVBa854McFUttJSNV1hKp8,7111
 chalk/features/_vector.py,sha256=jtJmUTtCaUFru4Fw17PYWozl3pPEdIOCYAuZOqIaN3Y,7205
 chalk/features/feature_cache_strategy.py,sha256=bXV-tjHfPzUotMpZ3h_D9Xxq-V1CLj_UcVtGIGMpMkI,4942
-chalk/features/feature_field.py,sha256=tLP9iybcYa7eM2eJJ0iSRks-ptyabA0l_SAJ83t6nEw,93758
+chalk/features/feature_field.py,sha256=x-B8CIt086xMOjtfpRMGcmfysS6G2vA_whsLZgGQeqI,93759
 chalk/features/feature_set.py,sha256=yNi0_J4CylAVkVp1Y67qV6i8vHMdE0p99DnyLPABPHI,12406
-chalk/features/feature_set_decorator.py,sha256=HMGnQdXQevpsTCNGIG2njViHDnm1ltFoZDhABFkp3Dw,65059
+chalk/features/feature_set_decorator.py,sha256=e5yp1FHObicrXPlNUxXQf4iwq4RdWuP-qe0cboDT2QI,65060
 chalk/features/feature_time.py,sha256=iUk8NDelig81jP7QT3tguyzx5eOZ-YC84OVgJRRKVwo,1639
 chalk/features/feature_wrapper.py,sha256=OolNWGGX67IAEMHCObFvOCpH5EmwjbMvMygRSBJJtu0,19259
 chalk/features/filter.py,sha256=2ldMbqvXC-nJ0jc-OZ36qHtrej-Jkx4TNQ1W_NZodAs,11177
@@ -680,7 +680,7 @@ chalk/features/dataframe/__init__.py,sha256=h88XglttpuUBt0TDoltaxp-90jNtIVxZhKAp
 chalk/features/dataframe/_filters.py,sha256=F9c5wExtmO21zhn4HEWcM9t4qSpfIO9Z7ysvoeZRV0I,20034
 chalk/features/dataframe/_impl.py,sha256=6zrwip7jL4DjNDEzmXw3PdKzZ-RJZTFwafqBUdK_Toc,88131
 chalk/features/dataframe/_validation.py,sha256=PsseXqoTgmgqaYtr-sTJP0CzlTo2v2PezkF0A8Za-oo,11037
-chalk/functions/__init__.py,sha256=xffsut7EzAfpAcdKvVtgnAoYlZBNalbme-jYOl_dKJ0,167805
+chalk/functions/__init__.py,sha256=k3-saPn3SFVND55UabjGt2sYtJ32ZY9LQx5lfNIkTcM,167789
 chalk/functions/holidays.py,sha256=7yhLjH4E1nDNGA0cXnvRQl5_h_W6rFZfIGYEBa-WiM8,3444
 chalk/functions/http.py,sha256=f3UpokWnjLyo0A_zZoSDgfGgRWmLu639FkREi6p82b4,6653
 chalk/functions/proto.py,sha256=RpM4GIj0-hVTA8RcK_k_fQOz9hvidgFtAbtcmD5Y4K0,5749
@@ -818,8 +818,8 @@ chalk/utils/tracing.py,sha256=ye5z6UCEsrxXC3ofXUNCDdUCf8ydPahEO92qQTd0AIA,11383
 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.94.9.dist-info/METADATA,sha256=lDQpMIhwUoLi4JAQhvGJgztvytpeHyhoSbz_DLeRrD4,27494
-chalkpy-2.94.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-chalkpy-2.94.9.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
-chalkpy-2.94.9.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
-chalkpy-2.94.9.dist-info/RECORD,,
+chalkpy-2.95.1.dist-info/METADATA,sha256=nig-03oMLofMUVjgNSNjPMJA5q05ufTcKApOi-6eQwo,27494
+chalkpy-2.95.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+chalkpy-2.95.1.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
+chalkpy-2.95.1.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
+chalkpy-2.95.1.dist-info/RECORD,,