Add Backblaze B2 artifact store flavor#4791
Conversation
|
hey @goanpeca, is the PR ready for review? or is it in the draft stage? |
4cd4f23 to
4d92ba5
Compare
4d92ba5 to
95ea42c
Compare
There was a problem hiding this comment.
Pull request overview
Adds a new b2 integration that provides a Backblaze B2 artifact store flavor by subclassing the existing S3 artifact store implementation, plus unit tests and documentation for users to configure B2 via its S3-compatible API.
Changes:
- Introduce
zenml.integrations.b2with ab2artifact store flavor/config that injects a default B2 S3 endpoint and supports B2-specific credential env vars. - Add a thin B2 artifact store subclass to ensure
config_kwargsare applied consistently to boths3fsand theboto3.resourcepath. - Add docs page + TOC entry and unit tests for the new flavor/config behavior.
Reviewed changes
Copilot reviewed 22 out of 23 changed files in this pull request and generated 4 comments.
Show a summary per file
| File | Description |
|---|---|
| tests/unit/integrations/b2/test_b2_artifact_store_flavor.py | Adds unit tests for the B2 flavor/config defaults and env var credential mapping. |
| tests/unit/integrations/b2/init.py | Marks the new unit-test package. |
| src/zenml/integrations/constants.py | Registers the b2 integration constant. |
| src/zenml/integrations/b2/flavors/b2_artifact_store_flavor.py | Implements B2 flavor + config defaults (endpoint, env var creds, config kwargs). |
| src/zenml/integrations/b2/flavors/init.py | Exposes the B2 flavor/config exports. |
| src/zenml/integrations/b2/artifact_stores/b2_artifact_store.py | Adds the B2 artifact store implementation (S3 subclass + boto3 config handling). |
| src/zenml/integrations/b2/artifact_stores/init.py | Exposes the B2 artifact store export. |
| src/zenml/integrations/b2/init.py | Declares the B2Integration and its requirements/flavors. |
| docs/book/component-guide/toc.md | Adds Backblaze B2 to the component guide TOC. |
| docs/book/component-guide/artifact-stores/backblaze.md | Adds user-facing documentation for configuring and using the B2 artifact store flavor. |
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
bd97dcf to
60c2f88
Compare
|
Hi @bcdurak thanks for the ping, I think we are good to review, however I will need to have the remaining CI tests that need approval to run to see if I need to fix something. I could not test all locally. Thanks! |
60c2f88 to
3e1f48d
Compare
3e1f48d to
782ab14
Compare
|
Hi @bcdurak I think none of the previous failing tests were related to this PR, I rebased the branch with latest develop and made a push. Ready for review Thanks! |
|
Thanks @goanpeca, you are right. Our CI had some errors unrelated to this PR. They should already be fixed now. We will try to take a look as soon as possible. 👍 |
|
|
782ab14 to
c7a134a
Compare
c7a134a to
93b06f3
Compare
|
Updated to the latest develop. |
Describe changes
Implemented #4786 Add Backblaze B2 artifact store flavor and documentation.
Pre-requisites
Please ensure you have done the following:
developand the open PR is targetingdevelop. If your branch wasn't based on develop read Contribution guide on rebasing branch to develop.Types of changes