Browse the repository at this point in the history

Browse the repository at this point in the history

Browse the repository at this point in the history

Browse the repository at this point in the history

* feat: Add dtype parameters to to_geodataframe

This change adds support for `bool_dtype`, `int_dtype`, `float_dtype`, and `string_dtype` parameters to the `to_geodataframe` method in `RowIterator` and `QueryJob`.

These parameters allow you to specify the desired pandas dtypes for boolean, integer, float, and string columns when converting BigQuery results to GeoDataFrames.

The changes include:
- Updating `RowIterator.to_geodataframe` to accept and pass these dtype parameters to the underlying `to_dataframe` method.
- Updating `QueryJob.to_geodataframe` to accept and pass these dtype parameters to the underlying `RowIterator.to_geodataframe` method.
- Adding unit tests to verify the correct handling of these parameters.

* updates to several tests re geopandas as well as imports

* updates to enum import

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* Update pyproject.toml

Co-authored-by: Tim Sweña (Swast) <swast@google.com>

* Update testing/constraints-3.9.txt

Co-authored-by: Tim Sweña (Swast) <swast@google.com>

---------

Co-authored-by: google-labs-jules[bot] <161369871+google-labs-jules[bot]@users.noreply.github.com>
Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
Co-authored-by: Tim Sweña (Swast) <swast@google.com>

4 people authored

Browse the repository at this point in the history

* Fix two types of warnings in unit tests

This commit addresses two warnings that appear when running unit tests:

1.  `PytestRemovedIn9Warning` in `tests/unit/test_opentelemetry_tracing.py`:
    Removed a `@pytest.mark.skipif` decorator from a fixture. The skip condition
    is already present on the test methods using the fixture.

2.  `FutureWarning` in `tests/unit/test_client.py`:
    Updated calls to `client.query()` to include `job_retry=None` when `job_id`
    is also specified. This is to avoid ambiguity as BigQuery cannot retry a
    failed job with the exact same ID.

* Update tests/unit/test_client.py

* Update tests/unit/test_client.py

* Update linting

* adds more examples of functions where job_retry is needed

---------

Co-authored-by: google-labs-jules[bot] <161369871+google-labs-jules[bot]@users.noreply.github.com>

Browse the repository at this point in the history

Browse the repository at this point in the history

…2185)

* Fix DeprecationWarnings for datetime methods in job tests

Replaced calls to deprecated `datetime.datetime.utcnow()` with
`datetime.datetime.now(datetime.UTC)` in `tests/unit/job/test_base.py`.

Replaced calls to deprecated `datetime.datetime.utcfromtimestamp()` with
`datetime.datetime.fromtimestamp(timestamp, datetime.UTC)` in
`tests/unit/job/helpers.py`.

These changes address the specific warnings identified in the issue
for these two files.

* Update tests/unit/job/test_base.py

* Update tests/unit/job/test_base.py

* Updates datetime code related to UTC

---------

Co-authored-by: google-labs-jules[bot] <161369871+google-labs-jules[bot]@users.noreply.github.com>

Browse the repository at this point in the history

Browse the repository at this point in the history

* feat: support job reservation

* Update google/cloud/bigquery/job/base.py

Co-authored-by: Chalmer Lowe <chalmerlowe@google.com>

---------

Co-authored-by: Chalmer Lowe <chalmerlowe@google.com>

Browse the repository at this point in the history

…-around time (#2189)

Update to the CI/CD pipeline via github workflow to help cut turn-around time.

* added github workflow
* changed the number of pytest-xdist workers from "auto" to "8" (based on local tests and discussion with Tim, choosing auto sometimes takes longer to run than choosing a smaller number. I suspect this is partly because for small or short tests the overhead needed to setup a worker exceeds the time savings of having extra workers).
* modified numerous tests to explicitly include a project path to avoid an attempt to find the project by making an external call via the pydata-google-auth workflow (which opens an input and waits for response from the user that never comes).

Browse the repository at this point in the history

Browse the repository at this point in the history

Co-authored-by: release-please[bot] <55107282+release-please[bot]@users.noreply.github.com>
Co-authored-by: Lingqing Gan <lingqing.gan@gmail.com>

Browse the repository at this point in the history