wandb 0.19.9__py3-none-any.whl → 0.19.11__py3-none-any.whl

wandb/sdk/artifacts/_generated/delete_aliases.py

@@ -0,0 +1,21 @@
+# 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
+
+
+class DeleteAliases(GQLBase):
+    delete_aliases: Optional[DeleteAliasesDeleteAliases] = Field(alias="deleteAliases")
+
+
+class DeleteAliasesDeleteAliases(GQLBase):
+    success: bool
+
+
+DeleteAliases.model_rebuild()

wandb/sdk/artifacts/_generated/delete_artifact_collection_tag_assignments.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
+
+
+class DeleteArtifactCollectionTagAssignments(GQLBase):
+    delete_artifact_collection_tag_assignments: Optional[
+        DeleteArtifactCollectionTagAssignmentsDeleteArtifactCollectionTagAssignments
+    ] = Field(alias="deleteArtifactCollectionTagAssignments")
+
+
+class DeleteArtifactCollectionTagAssignmentsDeleteArtifactCollectionTagAssignments(
+    GQLBase
+):
+    success: bool
+
+
+DeleteArtifactCollectionTagAssignments.model_rebuild()

wandb/sdk/artifacts/_generated/delete_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, Typename
+
+from .enums import ArtifactCollectionState
+
+
+class DeleteArtifactPortfolio(GQLBase):
+    delete_artifact_portfolio: Optional[
+        DeleteArtifactPortfolioDeleteArtifactPortfolio
+    ] = Field(alias="deleteArtifactPortfolio")
+
+
+class DeleteArtifactPortfolioDeleteArtifactPortfolio(GQLBase):
+    artifact_collection: DeleteArtifactPortfolioDeleteArtifactPortfolioArtifactCollection = Field(
+        alias="artifactCollection"
+    )
+
+
+class DeleteArtifactPortfolioDeleteArtifactPortfolioArtifactCollection(GQLBase):
+    typename__: Typename[
+        Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
+    ]
+    state: ArtifactCollectionState
+
+
+DeleteArtifactPortfolio.model_rebuild()
+DeleteArtifactPortfolioDeleteArtifactPortfolio.model_rebuild()

wandb/sdk/artifacts/_generated/delete_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, Typename
+
+from .enums import ArtifactCollectionState
+
+
+class DeleteArtifactSequence(GQLBase):
+    delete_artifact_sequence: Optional[DeleteArtifactSequenceDeleteArtifactSequence] = (
+        Field(alias="deleteArtifactSequence")
+    )
+
+
+class DeleteArtifactSequenceDeleteArtifactSequence(GQLBase):
+    artifact_collection: DeleteArtifactSequenceDeleteArtifactSequenceArtifactCollection = Field(
+        alias="artifactCollection"
+    )
+
+
+class DeleteArtifactSequenceDeleteArtifactSequenceArtifactCollection(GQLBase):
+    typename__: Typename[
+        Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
+    ]
+    state: ArtifactCollectionState
+
+
+DeleteArtifactSequence.model_rebuild()
+DeleteArtifactSequenceDeleteArtifactSequence.model_rebuild()

wandb/sdk/artifacts/_generated/enums.py

@@ -0,0 +1,17 @@
+# Generated by ariadne-codegen
+# Source: core/api/graphql/schemas/schema-latest.graphql
+
+from __future__ import annotations
+
+from enum import Enum
+
+
+class ArtifactState(str, Enum):
+    PENDING = "PENDING"
+    COMMITTED = "COMMITTED"
+    DELETED = "DELETED"
+
+
+class ArtifactCollectionState(str, Enum):
+    READY = "READY"
+    DELETED = "DELETED"

wandb/sdk/artifacts/_generated/fetch_linked_artifacts.py

@@ -0,0 +1,67 @@
+# 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, Typename
+
+
+class FetchLinkedArtifacts(GQLBase):
+    artifact: Optional[FetchLinkedArtifactsArtifact]
+
+
+class FetchLinkedArtifactsArtifact(GQLBase):
+    artifact_memberships: FetchLinkedArtifactsArtifactArtifactMemberships = Field(
+        alias="artifactMemberships"
+    )
+
+
+class FetchLinkedArtifactsArtifactArtifactMemberships(GQLBase):
+    edges: List[FetchLinkedArtifactsArtifactArtifactMembershipsEdges]
+
+
+class FetchLinkedArtifactsArtifactArtifactMembershipsEdges(GQLBase):
+    node: Optional[FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNode]
+
+
+class FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNode(GQLBase):
+    aliases: List[FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNodeAliases]
+    version_index: Optional[int] = Field(alias="versionIndex")
+    artifact_collection: Optional[
+        FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNodeArtifactCollection
+    ] = Field(alias="artifactCollection")
+
+
+class FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNodeAliases(GQLBase):
+    alias: str
+
+
+class FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNodeArtifactCollection(
+    GQLBase
+):
+    project: Optional[
+        FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNodeArtifactCollectionProject
+    ]
+    name: str
+    typename__: Typename[
+        Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
+    ]
+
+
+class FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNodeArtifactCollectionProject(
+    GQLBase
+):
+    entity_name: str = Field(alias="entityName")
+    name: str
+
+
+FetchLinkedArtifacts.model_rebuild()
+FetchLinkedArtifactsArtifact.model_rebuild()
+FetchLinkedArtifactsArtifactArtifactMemberships.model_rebuild()
+FetchLinkedArtifactsArtifactArtifactMembershipsEdges.model_rebuild()
+FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNode.model_rebuild()
+FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNodeArtifactCollection.model_rebuild()

wandb/sdk/artifacts/_generated/fragments.py

@@ -0,0 +1,221 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Any, List, Literal, Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase, GQLId, Typename
+
+from .enums import ArtifactState
+
+
+class ArtifactCollectionsFragment(GQLBase):
+    page_info: ArtifactCollectionsFragmentPageInfo = Field(alias="pageInfo")
+    total_count: int = Field(alias="totalCount")
+    edges: List[ArtifactCollectionsFragmentEdges]
+
+
+class ArtifactCollectionsFragmentEdges(GQLBase):
+    node: Optional[ArtifactCollectionsFragmentEdgesNode]
+    cursor: str
+
+
+class ArtifactCollectionsFragmentEdgesNode(GQLBase):
+    typename__: Typename[
+        Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
+    ]
+    id: GQLId
+    name: str
+    description: Optional[str]
+    created_at: str = Field(alias="createdAt")
+
+
+class ArtifactCollectionsFragmentPageInfo(GQLBase):
+    end_cursor: Optional[str] = Field(alias="endCursor")
+    has_next_page: bool = Field(alias="hasNextPage")
+
+
+class ArtifactFragment(GQLBase):
+    id: GQLId
+    artifact_sequence: ArtifactFragmentArtifactSequence = Field(
+        alias="artifactSequence"
+    )
+    version_index: Optional[int] = Field(alias="versionIndex")
+    artifact_type: ArtifactFragmentArtifactType = Field(alias="artifactType")
+    description: Optional[str]
+    metadata: Optional[Any]
+    ttl_duration_seconds: Optional[Any] = Field(
+        alias="ttlDurationSeconds", default=None
+    )
+    ttl_is_inherited: Optional[bool] = Field(alias="ttlIsInherited", default=None)
+    aliases: Optional[List[ArtifactFragmentAliases]] = None
+    tags: Optional[List[ArtifactFragmentTags]] = None
+    history_step: Optional[Any] = Field(alias="historyStep", default=None)
+    state: ArtifactState
+    current_manifest: Optional[ArtifactFragmentCurrentManifest] = Field(
+        alias="currentManifest"
+    )
+    commit_hash: Optional[str] = Field(alias="commitHash")
+    file_count: Any = Field(alias="fileCount")
+    created_at: str = Field(alias="createdAt")
+    updated_at: Optional[str] = Field(alias="updatedAt")
+
+
+class ArtifactFragmentAliases(GQLBase):
+    artifact_collection: Optional[ArtifactFragmentAliasesArtifactCollection] = Field(
+        alias="artifactCollection"
+    )
+    alias: str
+
+
+class ArtifactFragmentAliasesArtifactCollection(GQLBase):
+    typename__: Typename[
+        Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
+    ]
+    project: Optional[ArtifactFragmentAliasesArtifactCollectionProject]
+    name: str
+
+
+class ArtifactFragmentAliasesArtifactCollectionProject(GQLBase):
+    entity_name: str = Field(alias="entityName")
+    name: str
+
+
+class ArtifactFragmentArtifactSequence(GQLBase):
+    project: Optional[ArtifactFragmentArtifactSequenceProject]
+    name: str
+
+
+class ArtifactFragmentArtifactSequenceProject(GQLBase):
+    entity_name: str = Field(alias="entityName")
+    name: str
+
+
+class ArtifactFragmentArtifactType(GQLBase):
+    name: str
+
+
+class ArtifactFragmentCurrentManifest(GQLBase):
+    file: ArtifactFragmentCurrentManifestFile
+
+
+class ArtifactFragmentCurrentManifestFile(GQLBase):
+    direct_url: str = Field(alias="directUrl")
+
+
+class ArtifactFragmentTags(GQLBase):
+    name: str
+
+
+class ArtifactPortfolioTypeFields(GQLBase):
+    typename__: Typename[Literal["ArtifactPortfolio"]] = "ArtifactPortfolio"
+    id: GQLId
+    name: str
+
+
+class ArtifactSequenceTypeFields(GQLBase):
+    typename__: Typename[Literal["ArtifactSequence"]] = "ArtifactSequence"
+    id: GQLId
+    name: str
+
+
+class ArtifactTypeFragment(GQLBase):
+    typename__: Typename[Literal["ArtifactType"]] = "ArtifactType"
+    id: GQLId
+    name: str
+    description: Optional[str]
+    created_at: str = Field(alias="createdAt")
+
+
+class ArtifactTypesFragment(GQLBase):
+    edges: List[ArtifactTypesFragmentEdges]
+    page_info: ArtifactTypesFragmentPageInfo = Field(alias="pageInfo")
+
+
+class ArtifactTypesFragmentEdges(GQLBase):
+    node: Optional[ArtifactTypeFragment]
+    cursor: str
+
+
+class ArtifactTypesFragmentPageInfo(GQLBase):
+    end_cursor: Optional[str] = Field(alias="endCursor")
+    has_next_page: bool = Field(alias="hasNextPage")
+
+
+class ArtifactsFragment(GQLBase):
+    total_count: int = Field(alias="totalCount")
+    edges: List[ArtifactsFragmentEdges]
+    page_info: ArtifactsFragmentPageInfo = Field(alias="pageInfo")
+
+
+class ArtifactsFragmentEdges(GQLBase):
+    node: ArtifactFragment
+    version: str
+    cursor: str
+
+
+class ArtifactsFragmentPageInfo(GQLBase):
+    end_cursor: Optional[str] = Field(alias="endCursor")
+    has_next_page: bool = Field(alias="hasNextPage")
+
+
+class FilesFragment(GQLBase):
+    edges: List[FilesFragmentEdges]
+    page_info: FilesFragmentPageInfo = Field(alias="pageInfo")
+
+
+class FilesFragmentEdges(GQLBase):
+    node: Optional[FilesFragmentEdgesNode]
+    cursor: str
+
+
+class FilesFragmentEdgesNode(GQLBase):
+    id: GQLId
+    name: str
+    url: Optional[str]
+    size_bytes: Any = Field(alias="sizeBytes")
+    storage_path: Optional[str] = Field(alias="storagePath", default=None)
+    mimetype: Optional[str]
+    updated_at: Optional[str] = Field(alias="updatedAt")
+    digest: Optional[str]
+    md_5: Optional[str] = Field(alias="md5")
+    direct_url: str = Field(alias="directUrl")
+
+
+class FilesFragmentPageInfo(GQLBase):
+    end_cursor: Optional[str] = Field(alias="endCursor")
+    has_next_page: bool = Field(alias="hasNextPage")
+
+
+ArtifactCollectionsFragment.model_rebuild()
+ArtifactCollectionsFragmentEdges.model_rebuild()
+ArtifactCollectionsFragmentEdgesNode.model_rebuild()
+ArtifactCollectionsFragmentPageInfo.model_rebuild()
+ArtifactFragment.model_rebuild()
+ArtifactFragmentAliases.model_rebuild()
+ArtifactFragmentAliasesArtifactCollection.model_rebuild()
+ArtifactFragmentAliasesArtifactCollectionProject.model_rebuild()
+ArtifactFragmentArtifactSequence.model_rebuild()
+ArtifactFragmentArtifactSequenceProject.model_rebuild()
+ArtifactFragmentArtifactType.model_rebuild()
+ArtifactFragmentCurrentManifest.model_rebuild()
+ArtifactFragmentCurrentManifestFile.model_rebuild()
+ArtifactFragmentTags.model_rebuild()
+ArtifactPortfolioTypeFields.model_rebuild()
+ArtifactSequenceTypeFields.model_rebuild()
+ArtifactTypeFragment.model_rebuild()
+ArtifactTypesFragment.model_rebuild()
+ArtifactTypesFragmentEdges.model_rebuild()
+ArtifactTypesFragmentPageInfo.model_rebuild()
+ArtifactsFragment.model_rebuild()
+ArtifactsFragmentEdges.model_rebuild()
+ArtifactsFragmentPageInfo.model_rebuild()
+FilesFragment.model_rebuild()
+FilesFragmentEdges.model_rebuild()
+FilesFragmentEdgesNode.model_rebuild()
+FilesFragmentPageInfo.model_rebuild()
+ArtifactFragment.model_rebuild()
+ArtifactTypeFragment.model_rebuild()

wandb/sdk/artifacts/_generated/input_types.py

@@ -0,0 +1,28 @@
+# Generated by ariadne-codegen
+# Source: core/api/graphql/schemas/schema-latest.graphql
+
+from __future__ import annotations
+
+from typing import Any, Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+
+class ArtifactAliasInput(GQLBase):
+    artifact_collection_name: str = Field(alias="artifactCollectionName")
+    alias: str
+
+
+class ArtifactCollectionAliasInput(GQLBase):
+    alias: str
+    entity_name: str = Field(alias="entityName")
+    project_name: str = Field(alias="projectName")
+    artifact_collection_name: str = Field(alias="artifactCollectionName")
+
+
+class TagInput(GQLBase):
+    tag_category_name: Optional[str] = Field(alias="tagCategoryName", default=None)
+    tag_name: str = Field(alias="tagName")
+    attributes: Optional[Any] = None

wandb/sdk/artifacts/_generated/move_artifact_collection.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 MoveArtifactCollection(GQLBase):
+    move_artifact_sequence: Optional[MoveArtifactCollectionMoveArtifactSequence] = (
+        Field(alias="moveArtifactSequence")
+    )
+
+
+class MoveArtifactCollectionMoveArtifactSequence(GQLBase):
+    artifact_collection: Optional[
+        MoveArtifactCollectionMoveArtifactSequenceArtifactCollection
+    ] = Field(alias="artifactCollection")
+
+
+class MoveArtifactCollectionMoveArtifactSequenceArtifactCollection(GQLBase):
+    typename__: Typename[
+        Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
+    ]
+    id: GQLId
+    name: str
+    description: Optional[str]
+
+
+MoveArtifactCollection.model_rebuild()
+MoveArtifactCollectionMoveArtifactSequence.model_rebuild()