wandb 0.22.1__py3-none-win_arm64.whl

wandb/sdk/artifacts/_generated/project_artifact_collection.py

@@ -0,0 +1,101 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import List, Literal, Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase, GQLId, Typename
+
+
+class ProjectArtifactCollection(GQLBase):
+    project: Optional[ProjectArtifactCollectionProject]
+
+
+class ProjectArtifactCollectionProject(GQLBase):
+    artifact_type: Optional[ProjectArtifactCollectionProjectArtifactType] = Field(
+        alias="artifactType"
+    )
+
+
+class ProjectArtifactCollectionProjectArtifactType(GQLBase):
+    artifact_collection: Optional[
+        ProjectArtifactCollectionProjectArtifactTypeArtifactCollection
+    ] = Field(alias="artifactCollection")
+    artifact_sequence: Optional[
+        ProjectArtifactCollectionProjectArtifactTypeArtifactSequence
+    ] = Field(alias="artifactSequence")
+
+
+class ProjectArtifactCollectionProjectArtifactTypeArtifactCollection(GQLBase):
+    typename__: Typename[
+        Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
+    ]
+    id: GQLId
+    name: str
+    description: Optional[str]
+    created_at: str = Field(alias="createdAt")
+    tags: ProjectArtifactCollectionProjectArtifactTypeArtifactCollectionTags
+    aliases: ProjectArtifactCollectionProjectArtifactTypeArtifactCollectionAliases
+
+
+class ProjectArtifactCollectionProjectArtifactTypeArtifactCollectionTags(GQLBase):
+    edges: List[ProjectArtifactCollectionProjectArtifactTypeArtifactCollectionTagsEdges]
+
+
+class ProjectArtifactCollectionProjectArtifactTypeArtifactCollectionTagsEdges(GQLBase):
+    node: ProjectArtifactCollectionProjectArtifactTypeArtifactCollectionTagsEdgesNode
+
+
+class ProjectArtifactCollectionProjectArtifactTypeArtifactCollectionTagsEdgesNode(
+    GQLBase
+):
+    id: GQLId
+    name: str
+
+
+class ProjectArtifactCollectionProjectArtifactTypeArtifactCollectionAliases(GQLBase):
+    edges: List[
+        ProjectArtifactCollectionProjectArtifactTypeArtifactCollectionAliasesEdges
+    ]
+    page_info: ProjectArtifactCollectionProjectArtifactTypeArtifactCollectionAliasesPageInfo = Field(
+        alias="pageInfo"
+    )
+
+
+class ProjectArtifactCollectionProjectArtifactTypeArtifactCollectionAliasesEdges(
+    GQLBase
+):
+    node: Optional[
+        ProjectArtifactCollectionProjectArtifactTypeArtifactCollectionAliasesEdgesNode
+    ]
+    cursor: str
+
+
+class ProjectArtifactCollectionProjectArtifactTypeArtifactCollectionAliasesEdgesNode(
+    GQLBase
+):
+    alias: str
+
+
+class ProjectArtifactCollectionProjectArtifactTypeArtifactCollectionAliasesPageInfo(
+    GQLBase
+):
+    end_cursor: Optional[str] = Field(alias="endCursor")
+    has_next_page: bool = Field(alias="hasNextPage")
+
+
+class ProjectArtifactCollectionProjectArtifactTypeArtifactSequence(GQLBase):
+    typename__: Typename[Literal["ArtifactSequence"]]
+
+
+ProjectArtifactCollection.model_rebuild()
+ProjectArtifactCollectionProject.model_rebuild()
+ProjectArtifactCollectionProjectArtifactType.model_rebuild()
+ProjectArtifactCollectionProjectArtifactTypeArtifactCollection.model_rebuild()
+ProjectArtifactCollectionProjectArtifactTypeArtifactCollectionTags.model_rebuild()
+ProjectArtifactCollectionProjectArtifactTypeArtifactCollectionTagsEdges.model_rebuild()
+ProjectArtifactCollectionProjectArtifactTypeArtifactCollectionAliases.model_rebuild()
+ProjectArtifactCollectionProjectArtifactTypeArtifactCollectionAliasesEdges.model_rebuild()

wandb/sdk/artifacts/_generated/project_artifact_collections.py

@@ -0,0 +1,33 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+from .fragments import ArtifactCollectionsFragment
+
+
+class ProjectArtifactCollections(GQLBase):
+    project: Optional[ProjectArtifactCollectionsProject]
+
+
+class ProjectArtifactCollectionsProject(GQLBase):
+    artifact_type: Optional[ProjectArtifactCollectionsProjectArtifactType] = Field(
+        alias="artifactType"
+    )
+
+
+class ProjectArtifactCollectionsProjectArtifactType(GQLBase):
+    artifact_collections: Optional[ArtifactCollectionsFragment] = Field(
+        alias="artifactCollections"
+    )
+
+
+ProjectArtifactCollections.model_rebuild()
+ProjectArtifactCollectionsProject.model_rebuild()
+ProjectArtifactCollectionsProjectArtifactType.model_rebuild()

wandb/sdk/artifacts/_generated/project_artifact_type.py

@@ -0,0 +1,24 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+from .fragments import ArtifactTypeFragment
+
+
+class ProjectArtifactType(GQLBase):
+    project: Optional[ProjectArtifactTypeProject]
+
+
+class ProjectArtifactTypeProject(GQLBase):
+    artifact_type: Optional[ArtifactTypeFragment] = Field(alias="artifactType")
+
+
+ProjectArtifactType.model_rebuild()
+ProjectArtifactTypeProject.model_rebuild()

wandb/sdk/artifacts/_generated/project_artifact_types.py

@@ -0,0 +1,24 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+from .fragments import ArtifactTypesFragment
+
+
+class ProjectArtifactTypes(GQLBase):
+    project: Optional[ProjectArtifactTypesProject]
+
+
+class ProjectArtifactTypesProject(GQLBase):
+    artifact_types: ArtifactTypesFragment = Field(alias="artifactTypes")
+
+
+ProjectArtifactTypes.model_rebuild()
+ProjectArtifactTypesProject.model_rebuild()

wandb/sdk/artifacts/_generated/project_artifacts.py

@@ -0,0 +1,42 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Literal, Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase, Typename
+
+from .fragments import ArtifactsFragment
+
+
+class ProjectArtifacts(GQLBase):
+    project: Optional[ProjectArtifactsProject]
+
+
+class ProjectArtifactsProject(GQLBase):
+    artifact_type: Optional[ProjectArtifactsProjectArtifactType] = Field(
+        alias="artifactType"
+    )
+
+
+class ProjectArtifactsProjectArtifactType(GQLBase):
+    artifact_collection: Optional[
+        ProjectArtifactsProjectArtifactTypeArtifactCollection
+    ] = Field(alias="artifactCollection")
+
+
+class ProjectArtifactsProjectArtifactTypeArtifactCollection(GQLBase):
+    typename__: Typename[
+        Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
+    ]
+    name: str
+    artifacts: Optional[ArtifactsFragment]
+
+
+ProjectArtifacts.model_rebuild()
+ProjectArtifactsProject.model_rebuild()
+ProjectArtifactsProjectArtifactType.model_rebuild()
+ProjectArtifactsProjectArtifactTypeArtifactCollection.model_rebuild()

wandb/sdk/artifacts/_generated/registry_collections.py

@@ -0,0 +1,34 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+from .fragments import RegistryCollectionsPage
+
+
+class RegistryCollections(GQLBase):
+    organization: Optional[RegistryCollectionsOrganization]
+
+
+class RegistryCollectionsOrganization(GQLBase):
+    org_entity: Optional[RegistryCollectionsOrganizationOrgEntity] = Field(
+        alias="orgEntity"
+    )
+
+
+class RegistryCollectionsOrganizationOrgEntity(GQLBase):
+    name: str
+    artifact_collections: Optional[RegistryCollectionsPage] = Field(
+        alias="artifactCollections"
+    )
+
+
+RegistryCollections.model_rebuild()
+RegistryCollectionsOrganization.model_rebuild()
+RegistryCollectionsOrganizationOrgEntity.model_rebuild()

wandb/sdk/artifacts/_generated/registry_versions.py

@@ -0,0 +1,34 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+from .fragments import RegistryVersionsPage
+
+
+class RegistryVersions(GQLBase):
+    organization: Optional[RegistryVersionsOrganization]
+
+
+class RegistryVersionsOrganization(GQLBase):
+    org_entity: Optional[RegistryVersionsOrganizationOrgEntity] = Field(
+        alias="orgEntity"
+    )
+
+
+class RegistryVersionsOrganizationOrgEntity(GQLBase):
+    name: str
+    artifact_memberships: Optional[RegistryVersionsPage] = Field(
+        alias="artifactMemberships"
+    )
+
+
+RegistryVersions.model_rebuild()
+RegistryVersionsOrganization.model_rebuild()
+RegistryVersionsOrganizationOrgEntity.model_rebuild()

wandb/sdk/artifacts/_generated/run_input_artifacts.py

@@ -0,0 +1,31 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+from .fragments import RunInputArtifactConnectionFragment
+
+
+class RunInputArtifacts(GQLBase):
+    project: Optional[RunInputArtifactsProject]
+
+
+class RunInputArtifactsProject(GQLBase):
+    run: Optional[RunInputArtifactsProjectRun]
+
+
+class RunInputArtifactsProjectRun(GQLBase):
+    input_artifacts: Optional[RunInputArtifactConnectionFragment] = Field(
+        alias="inputArtifacts"
+    )
+
+
+RunInputArtifacts.model_rebuild()
+RunInputArtifactsProject.model_rebuild()
+RunInputArtifactsProjectRun.model_rebuild()

wandb/sdk/artifacts/_generated/run_output_artifacts.py

@@ -0,0 +1,31 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+from .fragments import RunOutputArtifactConnectionFragment
+
+
+class RunOutputArtifacts(GQLBase):
+    project: Optional[RunOutputArtifactsProject]
+
+
+class RunOutputArtifactsProject(GQLBase):
+    run: Optional[RunOutputArtifactsProjectRun]
+
+
+class RunOutputArtifactsProjectRun(GQLBase):
+    output_artifacts: Optional[RunOutputArtifactConnectionFragment] = Field(
+        alias="outputArtifacts"
+    )
+
+
+RunOutputArtifacts.model_rebuild()
+RunOutputArtifactsProject.model_rebuild()
+RunOutputArtifactsProjectRun.model_rebuild()

wandb/sdk/artifacts/_generated/type_info.py

@@ -0,0 +1,19 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+from .fragments import TypeInfoFragment
+
+
+class TypeInfo(GQLBase):
+    type: Optional[TypeInfoFragment] = Field(alias="__type")
+
+
+TypeInfo.model_rebuild()

wandb/sdk/artifacts/_generated/unlink_artifact.py

@@ -0,0 +1,25 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase, GQLId
+
+
+class UnlinkArtifact(GQLBase):
+    unlink_artifact: Optional[UnlinkArtifactUnlinkArtifact] = Field(
+        alias="unlinkArtifact"
+    )
+
+
+class UnlinkArtifactUnlinkArtifact(GQLBase):
+    artifact_id: GQLId = Field(alias="artifactID")
+    success: bool
+    client_mutation_id: Optional[str] = Field(alias="clientMutationId")
+
+
+UnlinkArtifact.model_rebuild()

wandb/sdk/artifacts/_generated/update_artifact.py

@@ -0,0 +1,26 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+from .fragments import ArtifactFragment
+
+
+class UpdateArtifact(GQLBase):
+    update_artifact: Optional[UpdateArtifactUpdateArtifact] = Field(
+        alias="updateArtifact"
+    )
+
+
+class UpdateArtifactUpdateArtifact(GQLBase):
+    artifact: ArtifactFragment
+
+
+UpdateArtifact.model_rebuild()
+UpdateArtifactUpdateArtifact.model_rebuild()

wandb/sdk/artifacts/_generated/update_artifact_portfolio.py

@@ -0,0 +1,35 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Literal, Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase, GQLId, Typename
+
+
+class UpdateArtifactPortfolio(GQLBase):
+    update_artifact_portfolio: Optional[
+        UpdateArtifactPortfolioUpdateArtifactPortfolio
+    ] = Field(alias="updateArtifactPortfolio")
+
+
+class UpdateArtifactPortfolioUpdateArtifactPortfolio(GQLBase):
+    artifact_collection: UpdateArtifactPortfolioUpdateArtifactPortfolioArtifactCollection = Field(
+        alias="artifactCollection"
+    )
+
+
+class UpdateArtifactPortfolioUpdateArtifactPortfolioArtifactCollection(GQLBase):
+    typename__: Typename[
+        Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
+    ]
+    id: GQLId
+    name: str
+    description: Optional[str]
+
+
+UpdateArtifactPortfolio.model_rebuild()
+UpdateArtifactPortfolioUpdateArtifactPortfolio.model_rebuild()

wandb/sdk/artifacts/_generated/update_artifact_sequence.py

@@ -0,0 +1,35 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Literal, Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase, GQLId, Typename
+
+
+class UpdateArtifactSequence(GQLBase):
+    update_artifact_sequence: Optional[UpdateArtifactSequenceUpdateArtifactSequence] = (
+        Field(alias="updateArtifactSequence")
+    )
+
+
+class UpdateArtifactSequenceUpdateArtifactSequence(GQLBase):
+    artifact_collection: UpdateArtifactSequenceUpdateArtifactSequenceArtifactCollection = Field(
+        alias="artifactCollection"
+    )
+
+
+class UpdateArtifactSequenceUpdateArtifactSequenceArtifactCollection(GQLBase):
+    typename__: Typename[
+        Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
+    ]
+    id: GQLId
+    name: str
+    description: Optional[str]
+
+
+UpdateArtifactSequence.model_rebuild()
+UpdateArtifactSequenceUpdateArtifactSequence.model_rebuild()

wandb/sdk/artifacts/_gqlutils.py

@@ -0,0 +1,47 @@
+from __future__ import annotations
+
+from functools import lru_cache
+
+from wandb_gql import Client, gql
+
+from ._generated import TYPE_INFO_GQL, TypeInfo, TypeInfoFragment
+
+OMITTABLE_ARTIFACT_FIELDS = frozenset(
+    {
+        "ttlDurationSeconds",
+        "ttlIsInherited",
+        "aliases",
+        "tags",
+        "historyStep",
+    }
+)
+
+
+@lru_cache(maxsize=16)
+def type_info(client: Client, typename: str) -> TypeInfoFragment | None:
+    """Returns the type info for a given GraphQL type."""
+    data = client.execute(gql(TYPE_INFO_GQL), variable_values={"name": typename})
+    return TypeInfo.model_validate(data).type
+
+
+def supports_enable_tracking_var(client: Client) -> bool:
+    """Returns True if the server supports the `enableTracking` variable for the `Project.artifact(...)` field."""
+    typ = type_info(client, "Project")
+    if (
+        typ
+        and typ.fields
+        and (art_field := next((f for f in typ.fields if f.name == "artifact"), None))
+    ):
+        return any("enableTracking" == arg.name for arg in art_field.args)
+    return False
+
+
+def allowed_fields(client: Client, typename: str) -> set[str]:
+    """Returns the allowed field names for a given GraphQL type."""
+    typ = type_info(client, typename)
+    return {f.name for f in typ.fields} if (typ and typ.fields) else set()
+
+
+def omit_artifact_fields(client: Client) -> set[str]:
+    """Return names of Artifact fields to remove from GraphQL requests (for server compatibility)."""
+    return set(OMITTABLE_ARTIFACT_FIELDS) - allowed_fields(client, "Artifact")

wandb/sdk/artifacts/_internal_artifact.py

@@ -0,0 +1,54 @@
+from __future__ import annotations
+
+import re
+from base64 import urlsafe_b64encode
+from typing import Any, Final
+from zlib import crc32
+
+from wandb.sdk.artifacts.artifact import Artifact
+
+PLACEHOLDER: Final[str] = "PLACEHOLDER"
+
+
+def sanitize_artifact_name(name: str) -> str:
+    """Sanitize the string to satisfy constraints on artifact names."""
+    # If the name is already sanitized, don't change it.
+    if (sanitized := re.sub(r"[^a-zA-Z0-9_\-.]+", "", name)) == name:
+        return name
+
+    # Append a short alphanumeric suffix to maintain uniqueness.
+    # Yes, CRC is meant for checksums and not as a general hash function, but
+    # a 32-bit CRC hash, encoded as (url-safe) base64, is fairly short while
+    # providing 4B+ possible values, which should be good enough for the corner
+    # case names this function is meant to address.
+    #
+    # As implemented, the final suffix should be 6 characters.
+    crc: int = crc32(name.encode("utf-8")) & 0xFFFFFFFF  # Ensure it's unsigned
+    crc_bytes = crc.to_bytes(4, byteorder="big")
+    suffix = urlsafe_b64encode(crc_bytes).rstrip(b"=").decode("ascii")
+
+    return f"{sanitized}-{suffix}"
+
+
+class InternalArtifact(Artifact):
+    """InternalArtifact is used to create artifacts that are intended for internal use.
+
+    This includes artifacts of type: `job`, `code`(with `source-` prefix in the collection name),
+    `run_table` (with `run-` prefix in the collection name), and artifacts that start with `wandb-`.
+    Users should not use this class directly.
+    """
+
+    def __init__(
+        self,
+        name: str,
+        type: str,
+        description: str | None = None,
+        metadata: dict[str, Any] | None = None,
+        incremental: bool = False,
+        use_as: str | None = None,
+    ) -> None:
+        sanitized_name = sanitize_artifact_name(name)
+        super().__init__(
+            sanitized_name, PLACEHOLDER, description, metadata, incremental, use_as
+        )
+        self._type = type

wandb/sdk/artifacts/_models/__init__.py

@@ -0,0 +1,4 @@
+"""Pydantic model classes and related helpers for artifacts code.
+
+Excludes GraphQL-generated classes.
+"""

wandb/sdk/artifacts/_models/base_model.py

@@ -0,0 +1,20 @@
+from __future__ import annotations
+
+from abc import ABC
+
+from pydantic import ConfigDict
+
+from wandb._pydantic import JsonableModel
+
+
+# Abstract base class with a common default configuration that's shared by all
+# pydantic classes in artifacts code (excluding GraphQL-generated types).
+class ArtifactsBase(JsonableModel, ABC):
+    # See: https://docs.pydantic.dev/latest/api/config/#pydantic.config.ConfigDict
+    model_config = ConfigDict(
+        # Most likely, some fields won't be pydantic types
+        arbitrary_types_allowed=True,
+        # Assume instances of the same class have already been validated to save time,
+        # but validate subclasses in case they override the default behavior.
+        revalidate_instances="subclass-instances",
+    )