Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions .github/workflows/ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,7 @@ jobs:
- name: install
run: |
pip install -e .
pip install --upgrade --force-reinstall git+https://github.com/googleapis/python-storage.git@main
- name: Run Standard Tests
run: |
export GOOGLE_APPLICATION_CREDENTIALS=$(pwd)/gcsfs/tests/fake-secret.json
Expand Down
3 changes: 2 additions & 1 deletion cloudbuild/e2e-tests-cloudbuild.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -99,6 +99,7 @@ steps:
pip install --upgrade pip > /dev/null
# Install testing libraries explicitly, as they are not in setup.py
pip install pytest pytest-timeout pytest-subtests pytest-asyncio fusepy google-cloud-storage > /dev/null
pip install --upgrade --force-reinstall git+https://github.com/googleapis/python-storage.git@main
pip install -e . > /dev/null
"

Expand Down Expand Up @@ -146,7 +147,7 @@ steps:
export GCSFS_TEST_PROJECT=${PROJECT_ID} && \
export GCSFS_TEST_KMS_KEY=projects/${PROJECT_ID}/locations/${_REGION}/keyRings/${_GCSFS_KEY_RING_NAME}/cryptoKeys/${_GCSFS_KEY_NAME} && \
echo '--- Running Zonal tests on VM ---' && \
ulimit -n 2048 && export GCSFS_EXPERIMENTAL_ZB_HNS_SUPPORT='true' && \
ulimit -n 4096 && export GCSFS_EXPERIMENTAL_ZB_HNS_SUPPORT='true' && \
pip install --upgrade --force-reinstall git+https://github.com/googleapis/python-storage.git@main && \
pytest -vv -s --log-format='%(asctime)s %(levelname)s %(message)s' --log-date-format='%H:%M:%S' --color=no gcsfs/tests/test_extended_gcsfs.py gcsfs/tests/test_zonal_file.py gcsfs/tests/test_async_gcsfs.py
"
Expand Down
2 changes: 1 addition & 1 deletion gcsfs/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -2157,7 +2157,7 @@ async def upload_chunk(fs, location, data, offset, size, content_type):
Uploads a chunk of data. This function has a conditional path to support
experimental features for Zonal buckets to append data using gRPC.
"""
from google.cloud.storage._experimental.asyncio.async_appendable_object_writer import (
from google.cloud.storage.asyncio.async_appendable_object_writer import (
AsyncAppendableObjectWriter,
)

Expand Down
8 changes: 4 additions & 4 deletions gcsfs/extended_gcsfs.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,11 +13,11 @@
from google.api_core.client_info import ClientInfo
from google.auth.credentials import AnonymousCredentials
from google.cloud import storage_control_v2
from google.cloud.storage._experimental.asyncio.async_appendable_object_writer import (
from google.cloud.storage.asyncio.async_appendable_object_writer import (
AsyncAppendableObjectWriter,
)
from google.cloud.storage._experimental.asyncio.async_grpc_client import AsyncGrpcClient
from google.cloud.storage._experimental.asyncio.async_multi_range_downloader import (
from google.cloud.storage.asyncio.async_grpc_client import AsyncGrpcClient
from google.cloud.storage.asyncio.async_multi_range_downloader import (
AsyncMultiRangeDownloader,
)

Expand Down Expand Up @@ -85,7 +85,7 @@ async def _get_grpc_client(self):
self._grpc_client = AsyncGrpcClient(
credentials=self.credential,
client_info=ClientInfo(user_agent=f"{USER_AGENT}/{version}"),
).grpc_client
)
return self._grpc_client

async def _get_control_plane_client(self):
Expand Down
2 changes: 1 addition & 1 deletion gcsfs/tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
import pytest_asyncio
import requests
from google.cloud import storage
from google.cloud.storage._experimental.asyncio.async_appendable_object_writer import (
from google.cloud.storage.asyncio.async_appendable_object_writer import (
AsyncAppendableObjectWriter,
)

Expand Down
4 changes: 2 additions & 2 deletions gcsfs/tests/test_extended_gcsfs.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,10 +11,10 @@

import pytest
from google.api_core.exceptions import NotFound
from google.cloud.storage._experimental.asyncio.async_appendable_object_writer import (
from google.cloud.storage.asyncio.async_appendable_object_writer import (
AsyncAppendableObjectWriter,
)
from google.cloud.storage._experimental.asyncio.async_multi_range_downloader import (
from google.cloud.storage.asyncio.async_multi_range_downloader import (
AsyncMultiRangeDownloader,
)
from google.cloud.storage.exceptions import DataCorruption
Expand Down
2 changes: 1 addition & 1 deletion gcsfs/zb_hns_utils.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
from io import BytesIO

from google.cloud.storage._experimental.asyncio.async_appendable_object_writer import (
from google.cloud.storage.asyncio.async_appendable_object_writer import (
AsyncAppendableObjectWriter,
)

Expand Down
2 changes: 1 addition & 1 deletion gcsfs/zonal_file.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import logging

from fsspec import asyn
from google.cloud.storage._experimental.asyncio.async_multi_range_downloader import (
from google.cloud.storage.asyncio.async_multi_range_downloader import (
AsyncMultiRangeDownloader,
)

Expand Down
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,6 @@ decorator>4.1.2
fsspec==2026.1.0
google-auth>=1.2
google-auth-oauthlib
google-cloud-storage>=3.8.0
google-cloud-storage>=3.9.0
google-cloud-storage-control
requests
Loading