_factories.py 461 B

1234567891011121314
  1. from __future__ import annotations
  2. from typing import TYPE_CHECKING
  3. from ._models.storage import StoragePolicyConfig
  4. from .storage_policies import WandbStoragePolicy
  5. if TYPE_CHECKING:
  6. from .storage_policy import StoragePolicy
  7. def make_storage_policy(region: str | None = None) -> StoragePolicy:
  8. """Returns the default `StoragePolicy` for the current environment."""
  9. return WandbStoragePolicy(config=StoragePolicyConfig.from_env(region=region))