"""Storage layout.""" from __future__ import annotations from enum import Enum class StorageLayout(str, Enum): V1 = "V1" V2 = "V2" @classmethod def from_env(cls) -> StorageLayout: from wandb.env import get_use_v1_artifacts return cls.V1 if get_use_v1_artifacts() else cls.V2
