wandb 0.21.2__py3-none-macosx_12_0_arm64.whl

wandb/sdk/artifacts/_generated/fragments.py

@@ -0,0 +1,459 @@
+# 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 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 ArtifactFragmentWithoutAliases(GQLBase):
+    id: GQLId
+    artifact_sequence: ArtifactFragmentWithoutAliasesArtifactSequence = Field(
+        alias="artifactSequence"
+    )
+    version_index: Optional[int] = Field(alias="versionIndex")
+    artifact_type: ArtifactFragmentWithoutAliasesArtifactType = 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)
+    tags: Optional[List[ArtifactFragmentWithoutAliasesTags]] = None
+    history_step: Optional[Any] = Field(alias="historyStep", default=None)
+    state: ArtifactState
+    current_manifest: Optional[ArtifactFragmentWithoutAliasesCurrentManifest] = 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 ArtifactFragmentWithoutAliasesArtifactSequence(GQLBase):
+    project: Optional[ArtifactFragmentWithoutAliasesArtifactSequenceProject]
+    name: str
+
+
+class ArtifactFragmentWithoutAliasesArtifactSequenceProject(GQLBase):
+    entity_name: str = Field(alias="entityName")
+    name: str
+
+
+class ArtifactFragmentWithoutAliasesArtifactType(GQLBase):
+    name: str
+
+
+class ArtifactFragmentWithoutAliasesCurrentManifest(GQLBase):
+    file: ArtifactFragmentWithoutAliasesCurrentManifestFile
+
+
+class ArtifactFragmentWithoutAliasesCurrentManifestFile(GQLBase):
+    direct_url: str = Field(alias="directUrl")
+
+
+class ArtifactFragmentWithoutAliasesTags(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 FileUrlsFragment(GQLBase):
+    page_info: FileUrlsFragmentPageInfo = Field(alias="pageInfo")
+    edges: List[FileUrlsFragmentEdges]
+
+
+class FileUrlsFragmentEdges(GQLBase):
+    node: Optional[FileUrlsFragmentEdgesNode]
+
+
+class FileUrlsFragmentEdgesNode(GQLBase):
+    name: str
+    direct_url: str = Field(alias="directUrl")
+
+
+class FileUrlsFragmentPageInfo(GQLBase):
+    has_next_page: bool = Field(alias="hasNextPage")
+    end_cursor: Optional[str] = Field(alias="endCursor")
+
+
+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")
+
+
+class MembershipWithArtifact(GQLBase):
+    id: GQLId
+    artifact_collection: Optional[MembershipWithArtifactArtifactCollection] = Field(
+        alias="artifactCollection"
+    )
+    artifact: Optional[ArtifactFragment]
+
+
+class MembershipWithArtifactArtifactCollection(GQLBase):
+    typename__: Typename[
+        Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
+    ]
+    id: GQLId
+    name: str
+    project: Optional[MembershipWithArtifactArtifactCollectionProject]
+
+
+class MembershipWithArtifactArtifactCollectionProject(GQLBase):
+    id: GQLId
+    entity_name: str = Field(alias="entityName")
+    name: str
+
+
+class RegistriesPage(GQLBase):
+    page_info: RegistriesPagePageInfo = Field(alias="pageInfo")
+    edges: List[RegistriesPageEdges]
+
+
+class RegistriesPageEdges(GQLBase):
+    node: Optional[RegistryFragment]
+
+
+class RegistriesPagePageInfo(GQLBase):
+    end_cursor: Optional[str] = Field(alias="endCursor")
+    has_next_page: bool = Field(alias="hasNextPage")
+
+
+class RegistryCollectionsPage(GQLBase):
+    total_count: int = Field(alias="totalCount")
+    page_info: RegistryCollectionsPagePageInfo = Field(alias="pageInfo")
+    edges: List[RegistryCollectionsPageEdges]
+
+
+class RegistryCollectionsPageEdges(GQLBase):
+    cursor: str
+    node: Optional[RegistryCollectionsPageEdgesNode]
+
+
+class RegistryCollectionsPageEdgesNode(GQLBase):
+    typename__: Typename[
+        Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
+    ]
+    id: GQLId
+    name: str
+    description: Optional[str]
+    created_at: str = Field(alias="createdAt")
+    tags: RegistryCollectionsPageEdgesNodeTags
+    project: Optional[RegistryCollectionsPageEdgesNodeProject]
+    default_artifact_type: RegistryCollectionsPageEdgesNodeDefaultArtifactType = Field(
+        alias="defaultArtifactType"
+    )
+    aliases: RegistryCollectionsPageEdgesNodeAliases
+
+
+class RegistryCollectionsPageEdgesNodeAliases(GQLBase):
+    edges: List[RegistryCollectionsPageEdgesNodeAliasesEdges]
+
+
+class RegistryCollectionsPageEdgesNodeAliasesEdges(GQLBase):
+    node: Optional[RegistryCollectionsPageEdgesNodeAliasesEdgesNode]
+
+
+class RegistryCollectionsPageEdgesNodeAliasesEdgesNode(GQLBase):
+    alias: str
+
+
+class RegistryCollectionsPageEdgesNodeDefaultArtifactType(GQLBase):
+    name: str
+
+
+class RegistryCollectionsPageEdgesNodeProject(GQLBase):
+    name: str
+    entity: RegistryCollectionsPageEdgesNodeProjectEntity
+
+
+class RegistryCollectionsPageEdgesNodeProjectEntity(GQLBase):
+    name: str
+
+
+class RegistryCollectionsPageEdgesNodeTags(GQLBase):
+    edges: List[RegistryCollectionsPageEdgesNodeTagsEdges]
+
+
+class RegistryCollectionsPageEdgesNodeTagsEdges(GQLBase):
+    node: RegistryCollectionsPageEdgesNodeTagsEdgesNode
+
+
+class RegistryCollectionsPageEdgesNodeTagsEdgesNode(GQLBase):
+    name: str
+
+
+class RegistryCollectionsPagePageInfo(GQLBase):
+    end_cursor: Optional[str] = Field(alias="endCursor")
+    has_next_page: bool = Field(alias="hasNextPage")
+
+
+class RegistryFragment(GQLBase):
+    id: GQLId
+    allow_all_artifact_types_in_registry: bool = Field(
+        alias="allowAllArtifactTypesInRegistry"
+    )
+    artifact_types: RegistryFragmentArtifactTypes = Field(alias="artifactTypes")
+    name: str
+    description: Optional[str]
+    created_at: str = Field(alias="createdAt")
+    updated_at: Optional[str] = Field(alias="updatedAt")
+    access: Optional[str]
+
+
+class RegistryFragmentArtifactTypes(GQLBase):
+    edges: List[RegistryFragmentArtifactTypesEdges]
+
+
+class RegistryFragmentArtifactTypesEdges(GQLBase):
+    node: Optional[RegistryFragmentArtifactTypesEdgesNode]
+
+
+class RegistryFragmentArtifactTypesEdgesNode(GQLBase):
+    name: str
+
+
+class RegistryVersionsPage(GQLBase):
+    page_info: RegistryVersionsPagePageInfo = Field(alias="pageInfo")
+    edges: List[RegistryVersionsPageEdges]
+
+
+class RegistryVersionsPageEdges(GQLBase):
+    node: Optional[RegistryVersionsPageEdgesNode]
+
+
+class RegistryVersionsPageEdgesNode(GQLBase):
+    artifact_collection: Optional[RegistryVersionsPageEdgesNodeArtifactCollection] = (
+        Field(alias="artifactCollection")
+    )
+    version_index: Optional[int] = Field(alias="versionIndex")
+    artifact: Optional[ArtifactFragmentWithoutAliases]
+    aliases: List[RegistryVersionsPageEdgesNodeAliases]
+
+
+class RegistryVersionsPageEdgesNodeAliases(GQLBase):
+    alias: str
+
+
+class RegistryVersionsPageEdgesNodeArtifactCollection(GQLBase):
+    typename__: Typename[
+        Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
+    ]
+    project: Optional[RegistryVersionsPageEdgesNodeArtifactCollectionProject]
+    name: str
+
+
+class RegistryVersionsPageEdgesNodeArtifactCollectionProject(GQLBase):
+    name: str
+    entity: RegistryVersionsPageEdgesNodeArtifactCollectionProjectEntity
+
+
+class RegistryVersionsPageEdgesNodeArtifactCollectionProjectEntity(GQLBase):
+    name: str
+
+
+class RegistryVersionsPagePageInfo(GQLBase):
+    end_cursor: Optional[str] = Field(alias="endCursor")
+    has_next_page: bool = Field(alias="hasNextPage")
+
+
+class ArtifactFragment(ArtifactFragmentWithoutAliases):
+    aliases: Optional[List[ArtifactFragmentAliases]] = None
+
+
+ArtifactCollectionsFragment.model_rebuild()
+ArtifactCollectionsFragmentEdges.model_rebuild()
+ArtifactCollectionsFragmentEdgesNode.model_rebuild()
+ArtifactCollectionsFragmentPageInfo.model_rebuild()
+ArtifactFragmentAliases.model_rebuild()
+ArtifactFragmentAliasesArtifactCollection.model_rebuild()
+ArtifactFragmentAliasesArtifactCollectionProject.model_rebuild()
+ArtifactFragmentWithoutAliases.model_rebuild()
+ArtifactFragmentWithoutAliasesArtifactSequence.model_rebuild()
+ArtifactFragmentWithoutAliasesArtifactSequenceProject.model_rebuild()
+ArtifactFragmentWithoutAliasesArtifactType.model_rebuild()
+ArtifactFragmentWithoutAliasesCurrentManifest.model_rebuild()
+ArtifactFragmentWithoutAliasesCurrentManifestFile.model_rebuild()
+ArtifactFragmentWithoutAliasesTags.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()
+FileUrlsFragment.model_rebuild()
+FileUrlsFragmentEdges.model_rebuild()
+FileUrlsFragmentEdgesNode.model_rebuild()
+FileUrlsFragmentPageInfo.model_rebuild()
+FilesFragment.model_rebuild()
+FilesFragmentEdges.model_rebuild()
+FilesFragmentEdgesNode.model_rebuild()
+FilesFragmentPageInfo.model_rebuild()
+MembershipWithArtifact.model_rebuild()
+MembershipWithArtifactArtifactCollection.model_rebuild()
+MembershipWithArtifactArtifactCollectionProject.model_rebuild()
+RegistriesPage.model_rebuild()
+RegistriesPageEdges.model_rebuild()
+RegistriesPagePageInfo.model_rebuild()
+RegistryCollectionsPage.model_rebuild()
+RegistryCollectionsPageEdges.model_rebuild()
+RegistryCollectionsPageEdgesNode.model_rebuild()
+RegistryCollectionsPageEdgesNodeAliases.model_rebuild()
+RegistryCollectionsPageEdgesNodeAliasesEdges.model_rebuild()
+RegistryCollectionsPageEdgesNodeAliasesEdgesNode.model_rebuild()
+RegistryCollectionsPageEdgesNodeDefaultArtifactType.model_rebuild()
+RegistryCollectionsPageEdgesNodeProject.model_rebuild()
+RegistryCollectionsPageEdgesNodeProjectEntity.model_rebuild()
+RegistryCollectionsPageEdgesNodeTags.model_rebuild()
+RegistryCollectionsPageEdgesNodeTagsEdges.model_rebuild()
+RegistryCollectionsPageEdgesNodeTagsEdgesNode.model_rebuild()
+RegistryCollectionsPagePageInfo.model_rebuild()
+RegistryFragment.model_rebuild()
+RegistryFragmentArtifactTypes.model_rebuild()
+RegistryFragmentArtifactTypesEdges.model_rebuild()
+RegistryFragmentArtifactTypesEdgesNode.model_rebuild()
+RegistryVersionsPage.model_rebuild()
+RegistryVersionsPageEdges.model_rebuild()
+RegistryVersionsPageEdgesNode.model_rebuild()
+RegistryVersionsPageEdgesNodeAliases.model_rebuild()
+RegistryVersionsPageEdgesNodeArtifactCollection.model_rebuild()
+RegistryVersionsPageEdgesNodeArtifactCollectionProject.model_rebuild()
+RegistryVersionsPageEdgesNodeArtifactCollectionProjectEntity.model_rebuild()
+RegistryVersionsPagePageInfo.model_rebuild()
+ArtifactFragment.model_rebuild()
+ArtifactFragmentWithoutAliases.model_rebuild()
+ArtifactTypeFragment.model_rebuild()
+RegistryFragment.model_rebuild()
+ArtifactFragment.model_rebuild()
+ArtifactFragment.model_rebuild()

wandb/sdk/artifacts/_generated/input_types.py

@@ -0,0 +1,46 @@
+# Generated by ariadne-codegen
+# Source: core/api/graphql/schemas/schema-latest.graphql
+
+from __future__ import annotations
+
+from typing import Any, List, Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase, GQLId
+
+
+class ArtifactAliasInput(GQLBase):
+    artifact_collection_name: str = Field(alias="artifactCollectionName")
+    alias: str
+
+
+class LinkArtifactInput(GQLBase):
+    artifact_id: Optional[GQLId] = Field(alias="artifactID", default=None)
+    artifact_portfolio_id: Optional[GQLId] = Field(
+        alias="artifactPortfolioID", default=None
+    )
+    artifact_portfolio_name: Optional[str] = Field(
+        alias="artifactPortfolioName", default=None
+    )
+    entity_name: Optional[str] = Field(alias="entityName", default=None)
+    project_name: Optional[str] = Field(alias="projectName", default=None)
+    aliases: Optional[List[ArtifactAliasInput]] = None
+    client_id: Optional[GQLId] = Field(alias="clientID", default=None)
+    client_mutation_id: Optional[str] = Field(alias="clientMutationId", default=None)
+
+
+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
+
+
+LinkArtifactInput.model_rebuild()

wandb/sdk/artifacts/_generated/link_artifact.py

@@ -0,0 +1,27 @@
+# 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 MembershipWithArtifact
+
+
+class LinkArtifact(GQLBase):
+    link_artifact: Optional[LinkArtifactLinkArtifact] = Field(alias="linkArtifact")
+
+
+class LinkArtifactLinkArtifact(GQLBase):
+    version_index: Optional[int] = Field(alias="versionIndex")
+    artifact_membership: Optional[MembershipWithArtifact] = Field(
+        alias="artifactMembership", default=None
+    )
+
+
+LinkArtifact.model_rebuild()
+LinkArtifactLinkArtifact.model_rebuild()

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()