chalkpy 2.167.1__tar.gz → 2.167.2__tar.gz

{chalkpy-2.167.1 → chalkpy-2.167.2}/PKG-INFO

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

chalkpy-2.167.2/chalk/_version.py

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

{chalkpy-2.167.1 → chalkpy-2.167.2}/chalk/client/client_impl.py

@@ -496,14 +496,15 @@ def _pandas_to_arrow_table(dataframe: Any) -> pa.Table:
 
 
 def _query_input_to_arrow_table(query_input: QueryInput) -> pa.Table:
+    # polars is optional here: a polars DataFrame input can only exist if polars is installed.
     try:
         import polars as pl
     except ImportError:
-        raise missing_dependency_exception("chalkpy[runtime]")
+        pl = None
 
     if isinstance(query_input, DataFrame):
         return query_input.to_pyarrow()
-    elif isinstance(query_input, pl.DataFrame):
+    elif pl is not None and isinstance(query_input, pl.DataFrame):
         return query_input.to_arrow()
     elif isinstance(query_input, pa.Table):
         return query_input
@@ -545,7 +546,7 @@ def _offline_query_inputs_should_be_uploaded(
     try:
         import polars as pl
     except ImportError:
-        raise missing_dependency_exception("chalkpy[runtime]")
+        pl = None
     if isinstance(inputs, (list, tuple)):
         if len(inputs) > row_limit:
             return True
@@ -558,7 +559,7 @@ def _offline_query_inputs_should_be_uploaded(
             num_rows = max(len(v) if hasattr(v, "__len__") else 1 for v in single_input.values())
         elif isinstance(single_input, DataFrame):
             num_rows = single_input.shape[0]
-        elif isinstance(single_input, pl.DataFrame):
+        elif pl is not None and isinstance(single_input, pl.DataFrame):
             num_rows = single_input.height
         elif is_pandas_dataframe(single_input):
             num_rows = single_input.shape[0]
@@ -2898,6 +2899,17 @@ https://docs.chalk.ai/cli/apply
                 stacklevel=2,
             )
 
+        if store_offline and self._resolve_branch(branch) is not None:
+            warnings.warn(
+                (
+                    "Branch deployments read from the online and offline stores but never write to them, so "
+                    + "`store_offline=True` has no effect on a branch: this query's results won't be stored "
+                    + "offline. Run the query without a branch to store its results."
+                ),
+                UserWarning,
+                stacklevel=2,
+            )
+
         lower_bound = _convert_datetime_or_timedelta_param("lower_bound", lower_bound)
         upper_bound = _convert_datetime_or_timedelta_param("upper_bound", upper_bound)
         lower_bound_inserted_at = _convert_datetime_or_timedelta_param(
@@ -2906,11 +2918,6 @@ https://docs.chalk.ai/cli/apply
         upper_bound_inserted_at = _convert_datetime_or_timedelta_param(
             "upper_bound_inserted_at", upper_bound_inserted_at
         )
-        try:
-            import polars as pl
-        except ImportError:
-            raise missing_dependency_exception("chalkpy[runtime]")
-        del pl  # unused
 
         if len(output) == 0 and len(required_output) == 0 and query_name is None:
             raise ValueError("Either 'output' or 'required_output' must be specified.")

{chalkpy-2.167.1 → chalkpy-2.167.2}/chalk/client/dataset.py

@@ -354,11 +354,13 @@ def _load_dataset_inner(
     columns: Optional[Sequence[ColumnMetadata]],
     return_type: Literal["polars_dataframe", "polars_lazyframe", "pandas", "pyarrow"],
 ) -> Union[pl.DataFrame, pl.LazyFrame, pd.DataFrame, pa.Table]:
-    try:
-        import polars as pl
-    except ImportError:
-        raise missing_dependency_exception("chalkpy[runtime]")
-    del pl  # unused
+    if return_type != "pyarrow":
+        # Only the non-pyarrow return types go through polars; `to_arrow()` is pure pyarrow.
+        try:
+            import polars as pl
+        except ImportError:
+            raise missing_dependency_exception("chalkpy[runtime]")
+        del pl  # unused
     # V2 datasets are in multiple files, and have column names encoded
     # due to DB limitations (e.g. bigquery does not support '.' in column names)
     # In addition, the datasets may contain extra columns (e.g. replaced observed at)
@@ -726,11 +728,13 @@ def load_dataset(
             "No outputs found. Check `dataset.status` to see if query is still running, and "
             + "'dataset.errors' for any query errors that may have occurred."
         )
-    try:
-        import polars as pl
-    except ImportError:
-        raise missing_dependency_exception("chalkpy[runtime]")
-    del pl  # Unused
+    if return_type != "pyarrow":
+        # Only the non-pyarrow return types go through polars; `to_arrow()` is pure pyarrow.
+        try:
+            import polars as pl
+        except ImportError:
+            raise missing_dependency_exception("chalkpy[runtime]")
+        del pl  # Unused
     if not isinstance(version, DatasetVersion):
         try:
             version = DatasetVersion(version)

chalkpy-2.167.1/chalk/_version.py

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