Skip to content

Feature/3391 support gcs artifacts#3394

Draft
Pleezon wants to merge 2 commits intoaimhubio:mainfrom
merantix-momentum:feature/3391-support-gcs-artifacts
Draft

Feature/3391 support gcs artifacts#3394
Pleezon wants to merge 2 commits intoaimhubio:mainfrom
merantix-momentum:feature/3391-support-gcs-artifacts

Conversation

@Pleezon
Copy link

@Pleezon Pleezon commented Feb 4, 2026

This PR implements Google Cloud Storage support as described in this issue: #3391.
I've added 'google-cloud-storage' as an extra, installable via aim[gcs]. In the future, boto3 should probably also be handled like this.

How to test these changes locally

Optionally, host a local gcs emulator:

docker run -d --name fake-gcs-server -p 4443:4443 fsouza/fake-gcs-server

Then, create a new project, add path/to/aim[gcs] as a dependency, and test:

import os
import tempfile
import time

import aim

os.environ["GOOGLE_CLOUD_PROJECT"] = "test"
os.environ["STORAGE_EMULATOR_HOST"] = "http://localhost:4443"

aim_run = aim.Run()
aim_run.set_artifacts_uri("gs://test")

with tempfile.TemporaryDirectory() as temp_dir:
    file_path = os.path.join(temp_dir, "test.yaml")
    with open(file_path, "w") as temp_file:
        temp_file.write("test")

    aim_run.log_artifact(file_path, name="test.yaml")

Note: As of right now, there are no tests implemented for (any) artifact storage. These could either be added to this PR or (because adding them would imply testing things unrelated to this pr) be implemented in another PR.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant