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.12and 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 <nameoftest>
+ $ nox -r -s system-3.14 -- -k <nameoftest>
 
 - 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
 
 
-defget_default_credentials_with_project() ->tuple[
-google.auth.credentials.Credentials, Optional[str]
-]:
+defget_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_methodisnotNone) and (sampling_methodnotin_SAMPLING_METHODS):
 raiseNotImplementedError(
 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= (
  (notallow_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):
 elsetimestamp(x)
  )
 elifx.type() ==ibis_dtypes.Timestamp(None): # type: ignore
-
 returntimestamp(x)
 else:
 # Numerical inputs.
@@ -1908,7 +1907,6 @@ def struct_op_impl(
 defai_generate(
 *values: ibis_types.Value, op: ops.AIGenerate
 ) ->ibis_types.StructValue:
-
 returnai_ops.AIGenerate(
 _construct_prompt(values, op.prompt_context), # type: ignore
 op.connection_id, # type: ignore
@@ -1923,7 +1921,6 @@ def ai_generate(
 defai_generate_bool(
 *values: ibis_types.Value, op: ops.AIGenerateBool
 ) ->ibis_types.StructValue:
-
 returnai_ops.AIGenerateBool(
 _construct_prompt(values, op.prompt_context), # type: ignore
 op.connection_id, # type: ignore
@@ -1937,7 +1934,6 @@ def ai_generate_bool(
 defai_generate_int(
 *values: ibis_types.Value, op: ops.AIGenerateInt
 ) ->ibis_types.StructValue:
-
 returnai_ops.AIGenerateInt(
 _construct_prompt(values, op.prompt_context), # type: ignore
 op.connection_id, # type: ignore
@@ -1951,7 +1947,6 @@ def ai_generate_int(
 defai_generate_double(
 *values: ibis_types.Value, op: ops.AIGenerateDouble
 ) ->ibis_types.StructValue:
-
 returnai_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)
 defai_if(*values: ibis_types.Value, op: ops.AIIf) ->ibis_types.StructValue:
-
 returnai_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:
 defai_classify(
 *values: ibis_types.Value, op: ops.AIClassify
 ) ->ibis_types.StructValue:
-
 returnai_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)
 defai_score(*values: ibis_types.Value, op: ops.AIScore) ->ibis_types.StructValue:
-
 returnai_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,
 returntuple(nodes.ColumnDef(expr, id) forid, exprinresults.items())
 
-defgraph_extract_window_expr() ->Optional[
-Tuple[Sequence[nodes.ColumnDef], window_spec.WindowSpec]
-]:
+defgraph_extract_window_expr() ->(
+Optional[Tuple[Sequence[nodes.ColumnDef], window_spec.WindowSpec]]
+):
 foridingraph.sinks:
 next_def=by_id[id]
 ifisinstance(next_def.expression, agg_expressions.WindowExpression):