chalkpy 2.96.6__tar.gz → 2.96.7__tar.gz

{chalkpy-2.96.6 → chalkpy-2.96.7}/PKG-INFO

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

chalkpy-2.96.7/chalk/_version.py

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

{chalkpy-2.96.6 → chalkpy-2.96.7}/chalk/client/client.py

@@ -526,6 +526,7 @@ class ChalkClient:
         explain: bool = False,
         num_input_rows: Optional[int] = None,
         headers: Mapping[str, str] | None = None,
+        planner_options: Mapping[str, str | int | bool] | None = None,
     ) -> PlanQueryResponse:
         """Plan a query without executing it.
 
@@ -575,6 +576,10 @@ class ChalkClient:
             The number of input rows that this plan will be run with. If unknown, specify `None`.
         headers
             Additional headers to provide with the request
+        planner_options
+            Dictionary of additional options to pass to the Chalk query engine.
+            Values may be provided as part of conversations with Chalk support
+            to enable or disable specific functionality.
 
         Returns
         -------

{chalkpy-2.96.6 → chalkpy-2.96.7}/chalk/client/client_async.py

@@ -486,6 +486,7 @@ class AsyncChalkClient:
         explain: bool = False,
         num_input_rows: Optional[int] = None,
         headers: Mapping[str, str] | None = None,
+        planner_options: Mapping[str, str | int | bool] | None = None,
     ) -> PlanQueryResponse:
         """Plan a query without executing it.
 
@@ -535,6 +536,10 @@ class AsyncChalkClient:
             The number of input rows that this plan will be run with. If unknown, specify `None`.
         headers
             Additional headers to provide with the request
+        planner_options
+            Dictionary of additional options to pass to the Chalk query engine.
+            Values may be provided as part of conversations with Chalk support
+            to enable or disable specific functionality.
 
         Returns
         -------

{chalkpy-2.96.6 → chalkpy-2.96.7}/chalk/client/client_impl.py

@@ -4438,6 +4438,7 @@ https://docs.chalk.ai/cli/apply
         explain: bool = False,
         num_input_rows: Optional[int] = None,
         headers: Mapping[str, str] | None = None,
+        planner_options: Mapping[str, str | int | bool] | None = None,
     ) -> PlanQueryResponse:
         encoded_inputs = encode_outputs(input).string_outputs
         outputs = encode_outputs(output).string_outputs
@@ -4472,6 +4473,7 @@ https://docs.chalk.ai/cli/apply
             store_plan_stages=store_plan_stages,
             explain=explain,
             num_input_rows=num_input_rows,
+            planner_options=planner_options,
         )
 
         extra_headers: dict[str, str] = {}

{chalkpy-2.96.6 → chalkpy-2.96.7}/chalk/client/models.py

@@ -915,7 +915,7 @@ class CreateOfflineQueryJobRequest(BaseModel):
     @root_validator
     def _validate_multiple_computers(cls, values: Dict[str, Any]):
         if values["input"] is None or isinstance(
-            values["input"], (UploadedParquetShardedOfflineQueryInput, OfflineQueryInputUri)
+            values["input"], (UploadedParquetShardedOfflineQueryInput, OfflineQueryInputUri, OfflineQueryInputSql)
         ):
             return values
         expected_use_multiple_computers = isinstance(values["input"], tuple)
@@ -1653,6 +1653,7 @@ class PlanQueryRequest(BaseModel):
     explain: bool = False
     store_plan_stages: bool = False
     encoding_options: FeatureEncodingOptions = FeatureEncodingOptions()
+    planner_options: Mapping[str, str | int | bool | float] | None = None
 
 
 class FeatureSchema(BaseModel):

{chalkpy-2.96.6 → chalkpy-2.96.7}/chalk/df/LazyFramePlaceholder.py

@@ -101,6 +101,11 @@ class LazyFramePlaceholder:
 
     __str__ = __repr__
 
+    def _is_equal(self, other: LazyFramePlaceholder) -> bool:
+        # proto equality is janky but it's hard to write a good eq method here given
+        # we have dicts and the proto round trip is slightly lossy on tuples vs lists
+        return self._to_proto() == other._to_proto()
+
     def _to_proto(self) -> dataframe_pb2.DataFramePlan:
         """
         Convert this proto plan to a dataframe.

{chalkpy-2.96.6 → chalkpy-2.96.7}/chalk/features/_encoding/converter.py

@@ -940,10 +940,9 @@ class PrimitiveFeatureConverter(Generic[_TPrim]):
 
     @staticmethod
     def convert_arrow_table_from_proto(proto: pb.TableParquetBytes) -> pa.Table:
-        import pyarrow.parquet
+        import pyarrow.parquet as pq
 
-        pf = pyarrow.parquet.ParquetFile(io.BytesIO(proto.encoded_parquet_bytes))
-        return pyarrow.parquet.read_table(pf)
+        return pq.read_table(io.BytesIO(proto.encoded_parquet_bytes))
 
     @staticmethod
     def _serialize_pa_decimal_to_pb(value: Union[pa.Decimal128Scalar, pa.Decimal256Scalar]) -> pb.ScalarValue:

chalkpy-2.96.6/chalk/_version.py

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