feat: add support for Python 3.14 (#2232)

Co-authored-by: Lingqing Gan <lingqing.gan@gmail.com>
Co-authored-by: Shuowei Li <shuowei@google.com>
Co-authored-by: Tim Swena <swast@google.com>
Co-authored-by: Trevor Bergeron <tbergeron@google.com>

Author: 5 people

.github/workflows/lint.yml

@@ -15,7 +15,7 @@ jobs:
     - name: Setup Python
       uses: actions/setup-python@v5
       with:
-        python-version: "3.10"
+        python-version: "3.14"
     - name: Install nox
       run: |
         python -m pip install --upgrade setuptools pip wheel

.github/workflows/unittest.yml

@@ -14,7 +14,7 @@ jobs:
     runs-on: ubuntu-22.04
     strategy:
       matrix:
-        python: ['3.10', '3.11', '3.12', '3.13']
+        python: ['3.10', '3.11', '3.12', '3.13', '3.14']
     steps:
     - name: Checkout
       uses: actions/checkout@v4
@@ -48,7 +48,7 @@ jobs:
     - name: Setup Python
       uses: actions/setup-python@v5
       with:
-        python-version: "3.10"
+        python-version: "3.14"
     - name: Install coverage
       run: |
         python -m pip install --upgrade setuptools pip wheel

.pre-commit-config.yaml

@@ -32,7 +32,7 @@ repos:
     -   id: isort
         name: isort (python)
 -   repo: https://github.com/psf/black
-    rev: 22.3.0
+    rev: 23.7.0
     hooks:
     - id: black
 -   repo: https://github.com/pycqa/flake8

CONTRIBUTING.rst

@@ -22,7 +22,7 @@ In order to add a feature:
   documentation.
 
 - The feature must work fully on the following CPython versions:
-  3.10, 3.11, 3.12 and 3.13 on both UNIX and Windows.
+  3.10, 3.11, 3.12, 3.13 and 3.14 on both UNIX and Windows.
 
 - The feature must not add unnecessary dependencies (where
   "unnecessary" is of course subjective, but new dependencies should
@@ -72,7 +72,7 @@ We use `nox <https://nox.readthedocs.io/en/latest/>`__ to instrument our tests.
 
 - To run a single unit test::
 
-    $ nox -s unit-3.13 -- -k <name of test>
+    $ nox -s unit-3.14 -- -k <name of test>
 
 
   .. note::
@@ -143,12 +143,12 @@ Running System Tests
    $ nox -s system
 
    # Run a single system test
-   $ nox -s system-3.13 -- -k <name of test>
+   $ nox -s system-3.14 -- -k <name of test>
 
 
   .. note::
 
-      System tests are only configured to run under Python 3.10, 3.11, 3.12 and 3.13.
+      System tests are only configured to run under Python 3.10, 3.12 and 3.14.
       For expediency, we do not run them in older versions of Python 3.
 
   This alone will not run the tests. You'll need to change some local
@@ -262,11 +262,13 @@ We support:
 -  `Python 3.11`_
 -  `Python 3.12`_
 -  `Python 3.13`_
+-  `Python 3.14`_
 
 .. _Python 3.10: https://docs.python.org/3.10/
 .. _Python 3.11: https://docs.python.org/3.11/
 .. _Python 3.12: https://docs.python.org/3.12/
 .. _Python 3.13: https://docs.python.org/3.13/
+.. _Python 3.14: https://docs.python.org/3.14/
 
 
 Supported versions can be found in our ``noxfile.py`` `config`_.

GEMINI.md

@@ -13,7 +13,7 @@ We use `nox` to instrument our tests.
 - To run a single unit test:
 
   ```bash
-  nox -r -s unit-3.13 -- -k <name of test>
+  nox -r -s unit-3.14 -- -k <name of test>
   ```
 
 - Ignore this step if you lack access to Google Cloud resources. To run system
@@ -23,7 +23,7 @@ We use `nox` to instrument our tests.
    $ nox -r -s system
 
    # Run a single system test
-   $ nox -r -s system-3.13 -- -k <name of test>
+   $ nox -r -s system-3.14 -- -k <name of test>
 
 - The codebase must have better coverage than it had previously after each
   change. You can test coverage via `nox -s unit system cover` (takes a long

bigframes/_config/auth.py

@@ -30,9 +30,9 @@
 _cached_project_default: Optional[str] = None
 
 
-def get_default_credentials_with_project() -> tuple[
-    google.auth.credentials.Credentials, Optional[str]
-]:
+def get_default_credentials_with_project() -> (
+    tuple[google.auth.credentials.Credentials, Optional[str]]
+):
     global _AUTH_LOCK, _cached_credentials, _cached_project_default
 
     with _AUTH_LOCK:

bigframes/core/block_transforms.py

@@ -618,7 +618,6 @@ def skew(
     skew_column_ids: typing.Sequence[str],
     grouping_column_ids: typing.Sequence[str] = (),
 ) -> blocks.Block:
-
     original_columns = skew_column_ids
     column_labels = block.select_columns(original_columns).column_labels
 

bigframes/core/blocks.py

@@ -657,7 +657,6 @@ def _get_sampling_option(
         sampling_method: Optional[str] = None,
         random_state: Optional[int] = None,
     ) -> sampling_options.SamplingOptions:
-
         if (sampling_method is not None) and (sampling_method not in _SAMPLING_METHODS):
             raise NotImplementedError(
                 f"The downsampling method {sampling_method} is not implemented, "
@@ -700,7 +699,8 @@ def to_pandas_batches(
         """Download results one message at a time.
 
         page_size and max_results determine the size and number of batches,
-        see https://cloud.google.com/python/docs/reference/bigquery/latest/google.cloud.bigquery.job.QueryJob#google_cloud_bigquery_job_QueryJob_result"""
+        see https://cloud.google.com/python/docs/reference/bigquery/latest/google.cloud.bigquery.job.QueryJob#google_cloud_bigquery_job_QueryJob_result
+        """
 
         under_10gb = (
             (not allow_large_results)

bigframes/core/compile/ibis_compiler/scalar_op_registry.py

@@ -1006,7 +1006,6 @@ def to_timestamp_op_impl(x: ibis_types.Value, op: ops.ToTimestampOp):
             else timestamp(x)
         )
     elif x.type() == ibis_dtypes.Timestamp(None):  # type: ignore
-
         return timestamp(x)
     else:
         # Numerical inputs.
@@ -1908,7 +1907,6 @@ def struct_op_impl(
 def ai_generate(
     *values: ibis_types.Value, op: ops.AIGenerate
 ) -> ibis_types.StructValue:
-
     return ai_ops.AIGenerate(
         _construct_prompt(values, op.prompt_context),  # type: ignore
         op.connection_id,  # type: ignore
@@ -1923,7 +1921,6 @@ def ai_generate(
 def ai_generate_bool(
     *values: ibis_types.Value, op: ops.AIGenerateBool
 ) -> ibis_types.StructValue:
-
     return ai_ops.AIGenerateBool(
         _construct_prompt(values, op.prompt_context),  # type: ignore
         op.connection_id,  # type: ignore
@@ -1937,7 +1934,6 @@ def ai_generate_bool(
 def ai_generate_int(
     *values: ibis_types.Value, op: ops.AIGenerateInt
 ) -> ibis_types.StructValue:
-
     return ai_ops.AIGenerateInt(
         _construct_prompt(values, op.prompt_context),  # type: ignore
         op.connection_id,  # type: ignore
@@ -1951,7 +1947,6 @@ def ai_generate_int(
 def ai_generate_double(
     *values: ibis_types.Value, op: ops.AIGenerateDouble
 ) -> ibis_types.StructValue:
-
     return ai_ops.AIGenerateDouble(
         _construct_prompt(values, op.prompt_context),  # type: ignore
         op.connection_id,  # type: ignore
@@ -1963,7 +1958,6 @@ def ai_generate_double(
 
 @scalar_op_compiler.register_nary_op(ops.AIIf, pass_op=True)
 def ai_if(*values: ibis_types.Value, op: ops.AIIf) -> ibis_types.StructValue:
-
     return ai_ops.AIIf(
         _construct_prompt(values, op.prompt_context),  # type: ignore
         op.connection_id,  # type: ignore
@@ -1974,7 +1968,6 @@ def ai_if(*values: ibis_types.Value, op: ops.AIIf) -> ibis_types.StructValue:
 def ai_classify(
     *values: ibis_types.Value, op: ops.AIClassify
 ) -> ibis_types.StructValue:
-
     return ai_ops.AIClassify(
         _construct_prompt(values, op.prompt_context),  # type: ignore
         op.categories,  # type: ignore
@@ -1984,7 +1977,6 @@ def ai_classify(
 
 @scalar_op_compiler.register_nary_op(ops.AIScore, pass_op=True)
 def ai_score(*values: ibis_types.Value, op: ops.AIScore) -> ibis_types.StructValue:
-
     return ai_ops.AIScore(
         _construct_prompt(values, op.prompt_context),  # type: ignore
         op.connection_id,  # type: ignore

bigframes/core/expression_factoring.py

@@ -381,9 +381,9 @@ def graph_extract_scalar_exprs() -> Sequence[nodes.ColumnDef]:
         # TODO: We can prune expressions that won't be reused here,
         return tuple(nodes.ColumnDef(expr, id) for id, expr in results.items())
 
-    def graph_extract_window_expr() -> Optional[
-        Tuple[Sequence[nodes.ColumnDef], window_spec.WindowSpec]
-    ]:
+    def graph_extract_window_expr() -> (
+        Optional[Tuple[Sequence[nodes.ColumnDef], window_spec.WindowSpec]]
+    ):
         for id in graph.sinks:
             next_def = by_id[id]
             if isinstance(next_def.expression, agg_expressions.WindowExpression):