wandb 0.22.1__py3-none-macosx_12_0_arm64.whl → 0.22.3__py3-none-macosx_12_0_arm64.whl

wandb/sdk/artifacts/_generated/project_artifact_collection.py

@@ -3,99 +3,31 @@
 
 from __future__ import annotations
 
-from typing import List, Literal, Optional
+from typing import Optional
 
 from pydantic import Field
 
-from wandb._pydantic import GQLBase, GQLId, Typename
+from wandb._pydantic import GQLResult
 
+from .fragments import ArtifactCollectionFragment
 
-class ProjectArtifactCollection(GQLBase):
+
+class ProjectArtifactCollection(GQLResult):
     project: Optional[ProjectArtifactCollectionProject]
 
 
-class ProjectArtifactCollectionProject(GQLBase):
+class ProjectArtifactCollectionProject(GQLResult):
     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 ProjectArtifactCollectionProjectArtifactType(GQLResult):
+    artifact_collection: Optional[ArtifactCollectionFragment] = Field(
+        alias="artifactCollection"
     )
 
 
-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

@@ -3,31 +3,43 @@
 
 from __future__ import annotations
 
-from typing import Optional
+from typing import List, Optional
 
 from pydantic import Field
 
-from wandb._pydantic import GQLBase
+from wandb._pydantic import GQLResult
 
-from .fragments import ArtifactCollectionsFragment
+from .fragments import ArtifactCollectionFragment, PageInfoFragment
 
 
-class ProjectArtifactCollections(GQLBase):
+class ProjectArtifactCollections(GQLResult):
     project: Optional[ProjectArtifactCollectionsProject]
 
 
-class ProjectArtifactCollectionsProject(GQLBase):
+class ProjectArtifactCollectionsProject(GQLResult):
     artifact_type: Optional[ProjectArtifactCollectionsProjectArtifactType] = Field(
         alias="artifactType"
     )
 
 
-class ProjectArtifactCollectionsProjectArtifactType(GQLBase):
-    artifact_collections: Optional[ArtifactCollectionsFragment] = Field(
-        alias="artifactCollections"
-    )
+class ProjectArtifactCollectionsProjectArtifactType(GQLResult):
+    artifact_collections: Optional[
+        ProjectArtifactCollectionsProjectArtifactTypeArtifactCollections
+    ] = Field(alias="artifactCollections")
+
+
+class ProjectArtifactCollectionsProjectArtifactTypeArtifactCollections(GQLResult):
+    total_count: int = Field(alias="totalCount")
+    page_info: PageInfoFragment = Field(alias="pageInfo")
+    edges: List[ProjectArtifactCollectionsProjectArtifactTypeArtifactCollectionsEdges]
+
+
+class ProjectArtifactCollectionsProjectArtifactTypeArtifactCollectionsEdges(GQLResult):
+    node: Optional[ArtifactCollectionFragment]
 
 
 ProjectArtifactCollections.model_rebuild()
 ProjectArtifactCollectionsProject.model_rebuild()
 ProjectArtifactCollectionsProjectArtifactType.model_rebuild()
+ProjectArtifactCollectionsProjectArtifactTypeArtifactCollections.model_rebuild()
+ProjectArtifactCollectionsProjectArtifactTypeArtifactCollectionsEdges.model_rebuild()

wandb/sdk/artifacts/_generated/project_artifact_type.py

@@ -7,16 +7,16 @@ from typing import Optional
 
 from pydantic import Field
 
-from wandb._pydantic import GQLBase
+from wandb._pydantic import GQLResult
 
 from .fragments import ArtifactTypeFragment
 
 
-class ProjectArtifactType(GQLBase):
+class ProjectArtifactType(GQLResult):
     project: Optional[ProjectArtifactTypeProject]
 
 
-class ProjectArtifactTypeProject(GQLBase):
+class ProjectArtifactTypeProject(GQLResult):
     artifact_type: Optional[ArtifactTypeFragment] = Field(alias="artifactType")
 
 

wandb/sdk/artifacts/_generated/project_artifact_types.py

@@ -3,22 +3,35 @@
 
 from __future__ import annotations
 
-from typing import Optional
+from typing import List, Optional
 
 from pydantic import Field
 
-from wandb._pydantic import GQLBase
+from wandb._pydantic import GQLResult
 
-from .fragments import ArtifactTypesFragment
+from .fragments import ArtifactTypeFragment, PageInfoFragment
 
 
-class ProjectArtifactTypes(GQLBase):
+class ProjectArtifactTypes(GQLResult):
     project: Optional[ProjectArtifactTypesProject]
 
 
-class ProjectArtifactTypesProject(GQLBase):
-    artifact_types: ArtifactTypesFragment = Field(alias="artifactTypes")
+class ProjectArtifactTypesProject(GQLResult):
+    artifact_types: ProjectArtifactTypesProjectArtifactTypes = Field(
+        alias="artifactTypes"
+    )
+
+
+class ProjectArtifactTypesProjectArtifactTypes(GQLResult):
+    edges: List[ProjectArtifactTypesProjectArtifactTypesEdges]
+    page_info: PageInfoFragment = Field(alias="pageInfo")
+
+
+class ProjectArtifactTypesProjectArtifactTypesEdges(GQLResult):
+    node: Optional[ArtifactTypeFragment]
 
 
 ProjectArtifactTypes.model_rebuild()
 ProjectArtifactTypesProject.model_rebuild()
+ProjectArtifactTypesProjectArtifactTypes.model_rebuild()
+ProjectArtifactTypesProjectArtifactTypesEdges.model_rebuild()

wandb/sdk/artifacts/_generated/project_artifacts.py

@@ -7,33 +7,32 @@ from typing import Literal, Optional
 
 from pydantic import Field
 
-from wandb._pydantic import GQLBase, Typename
+from wandb._pydantic import GQLResult, Typename
 
-from .fragments import ArtifactsFragment
+from .fragments import VersionedArtifactConnectionFragment
 
 
-class ProjectArtifacts(GQLBase):
+class ProjectArtifacts(GQLResult):
     project: Optional[ProjectArtifactsProject]
 
 
-class ProjectArtifactsProject(GQLBase):
+class ProjectArtifactsProject(GQLResult):
     artifact_type: Optional[ProjectArtifactsProjectArtifactType] = Field(
         alias="artifactType"
     )
 
 
-class ProjectArtifactsProjectArtifactType(GQLBase):
+class ProjectArtifactsProjectArtifactType(GQLResult):
     artifact_collection: Optional[
         ProjectArtifactsProjectArtifactTypeArtifactCollection
     ] = Field(alias="artifactCollection")
 
 
-class ProjectArtifactsProjectArtifactTypeArtifactCollection(GQLBase):
+class ProjectArtifactsProjectArtifactTypeArtifactCollection(GQLResult):
     typename__: Typename[
         Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
     ]
-    name: str
-    artifacts: Optional[ArtifactsFragment]
+    artifacts: Optional[VersionedArtifactConnectionFragment]
 
 
 ProjectArtifacts.model_rebuild()

wandb/sdk/artifacts/_generated/registry_collections.py

@@ -3,32 +3,44 @@
 
 from __future__ import annotations
 
-from typing import Optional
+from typing import List, Optional
 
 from pydantic import Field
 
-from wandb._pydantic import GQLBase
+from wandb._pydantic import GQLResult
 
-from .fragments import RegistryCollectionsPage
+from .fragments import PageInfoFragment, RegistryCollectionFragment
 
 
-class RegistryCollections(GQLBase):
+class RegistryCollections(GQLResult):
     organization: Optional[RegistryCollectionsOrganization]
 
 
-class RegistryCollectionsOrganization(GQLBase):
+class RegistryCollectionsOrganization(GQLResult):
     org_entity: Optional[RegistryCollectionsOrganizationOrgEntity] = Field(
         alias="orgEntity"
     )
 
 
-class RegistryCollectionsOrganizationOrgEntity(GQLBase):
+class RegistryCollectionsOrganizationOrgEntity(GQLResult):
     name: str
-    artifact_collections: Optional[RegistryCollectionsPage] = Field(
-        alias="artifactCollections"
-    )
+    artifact_collections: Optional[
+        RegistryCollectionsOrganizationOrgEntityArtifactCollections
+    ] = Field(alias="artifactCollections")
+
+
+class RegistryCollectionsOrganizationOrgEntityArtifactCollections(GQLResult):
+    total_count: int = Field(alias="totalCount")
+    page_info: PageInfoFragment = Field(alias="pageInfo")
+    edges: List[RegistryCollectionsOrganizationOrgEntityArtifactCollectionsEdges]
+
+
+class RegistryCollectionsOrganizationOrgEntityArtifactCollectionsEdges(GQLResult):
+    node: Optional[RegistryCollectionFragment]
 
 
 RegistryCollections.model_rebuild()
 RegistryCollectionsOrganization.model_rebuild()
 RegistryCollectionsOrganizationOrgEntity.model_rebuild()
+RegistryCollectionsOrganizationOrgEntityArtifactCollections.model_rebuild()
+RegistryCollectionsOrganizationOrgEntityArtifactCollectionsEdges.model_rebuild()

wandb/sdk/artifacts/_generated/registry_versions.py

@@ -3,32 +3,43 @@
 
 from __future__ import annotations
 
-from typing import Optional
+from typing import List, Optional
 
 from pydantic import Field
 
-from wandb._pydantic import GQLBase
+from wandb._pydantic import GQLResult
 
-from .fragments import RegistryVersionsPage
+from .fragments import ArtifactMembershipFragment, PageInfoFragment
 
 
-class RegistryVersions(GQLBase):
+class RegistryVersions(GQLResult):
     organization: Optional[RegistryVersionsOrganization]
 
 
-class RegistryVersionsOrganization(GQLBase):
+class RegistryVersionsOrganization(GQLResult):
     org_entity: Optional[RegistryVersionsOrganizationOrgEntity] = Field(
         alias="orgEntity"
     )
 
 
-class RegistryVersionsOrganizationOrgEntity(GQLBase):
+class RegistryVersionsOrganizationOrgEntity(GQLResult):
     name: str
-    artifact_memberships: Optional[RegistryVersionsPage] = Field(
-        alias="artifactMemberships"
-    )
+    artifact_memberships: Optional[
+        RegistryVersionsOrganizationOrgEntityArtifactMemberships
+    ] = Field(alias="artifactMemberships")
+
+
+class RegistryVersionsOrganizationOrgEntityArtifactMemberships(GQLResult):
+    page_info: PageInfoFragment = Field(alias="pageInfo")
+    edges: List[RegistryVersionsOrganizationOrgEntityArtifactMembershipsEdges]
+
+
+class RegistryVersionsOrganizationOrgEntityArtifactMembershipsEdges(GQLResult):
+    node: Optional[ArtifactMembershipFragment]
 
 
 RegistryVersions.model_rebuild()
 RegistryVersionsOrganization.model_rebuild()
 RegistryVersionsOrganizationOrgEntity.model_rebuild()
+RegistryVersionsOrganizationOrgEntityArtifactMemberships.model_rebuild()
+RegistryVersionsOrganizationOrgEntityArtifactMembershipsEdges.model_rebuild()

wandb/sdk/artifacts/_generated/rename_registry.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 GQLResult
+
+from .fragments import RegistryFragment
+
+
+class RenameRegistry(GQLResult):
+    rename_project: Optional[RenameRegistryRenameProject] = Field(alias="renameProject")
+
+
+class RenameRegistryRenameProject(GQLResult):
+    inserted: Optional[bool]
+    project: Optional[RegistryFragment]
+
+
+RenameRegistry.model_rebuild()
+RenameRegistryRenameProject.model_rebuild()

wandb/sdk/artifacts/_generated/run_input_artifacts.py

@@ -5,25 +5,21 @@ from __future__ import annotations
 
 from typing import Optional
 
-from pydantic import Field
-
-from wandb._pydantic import GQLBase
+from wandb._pydantic import GQLResult
 
 from .fragments import RunInputArtifactConnectionFragment
 
 
-class RunInputArtifacts(GQLBase):
+class RunInputArtifacts(GQLResult):
     project: Optional[RunInputArtifactsProject]
 
 
-class RunInputArtifactsProject(GQLBase):
+class RunInputArtifactsProject(GQLResult):
     run: Optional[RunInputArtifactsProjectRun]
 
 
-class RunInputArtifactsProjectRun(GQLBase):
-    input_artifacts: Optional[RunInputArtifactConnectionFragment] = Field(
-        alias="inputArtifacts"
-    )
+class RunInputArtifactsProjectRun(GQLResult):
+    artifacts: Optional[RunInputArtifactConnectionFragment]
 
 
 RunInputArtifacts.model_rebuild()

wandb/sdk/artifacts/_generated/run_output_artifacts.py

@@ -5,25 +5,21 @@ from __future__ import annotations
 
 from typing import Optional
 
-from pydantic import Field
-
-from wandb._pydantic import GQLBase
+from wandb._pydantic import GQLResult
 
 from .fragments import RunOutputArtifactConnectionFragment
 
 
-class RunOutputArtifacts(GQLBase):
+class RunOutputArtifacts(GQLResult):
     project: Optional[RunOutputArtifactsProject]
 
 
-class RunOutputArtifactsProject(GQLBase):
+class RunOutputArtifactsProject(GQLResult):
     run: Optional[RunOutputArtifactsProjectRun]
 
 
-class RunOutputArtifactsProjectRun(GQLBase):
-    output_artifacts: Optional[RunOutputArtifactConnectionFragment] = Field(
-        alias="outputArtifacts"
-    )
+class RunOutputArtifactsProjectRun(GQLResult):
+    artifacts: Optional[RunOutputArtifactConnectionFragment]
 
 
 RunOutputArtifacts.model_rebuild()

wandb/sdk/artifacts/_generated/type_info.py

@@ -7,12 +7,12 @@ from typing import Optional
 
 from pydantic import Field
 
-from wandb._pydantic import GQLBase
+from wandb._pydantic import GQLResult
 
 from .fragments import TypeInfoFragment
 
 
-class TypeInfo(GQLBase):
+class TypeInfo(GQLResult):
     type: Optional[TypeInfoFragment] = Field(alias="__type")
 
 

wandb/sdk/artifacts/_generated/unlink_artifact.py

@@ -7,19 +7,17 @@ from typing import Optional
 
 from pydantic import Field
 
-from wandb._pydantic import GQLBase, GQLId
+from wandb._pydantic import GQLResult
 
 
-class UnlinkArtifact(GQLBase):
+class UnlinkArtifact(GQLResult):
     unlink_artifact: Optional[UnlinkArtifactUnlinkArtifact] = Field(
         alias="unlinkArtifact"
     )
 
 
-class UnlinkArtifactUnlinkArtifact(GQLBase):
-    artifact_id: GQLId = Field(alias="artifactID")
+class UnlinkArtifactUnlinkArtifact(GQLResult):
     success: bool
-    client_mutation_id: Optional[str] = Field(alias="clientMutationId")
 
 
 UnlinkArtifact.model_rebuild()

wandb/sdk/artifacts/_generated/update_artifact.py

@@ -7,18 +7,18 @@ from typing import Optional
 
 from pydantic import Field
 
-from wandb._pydantic import GQLBase
+from wandb._pydantic import GQLResult
 
 from .fragments import ArtifactFragment
 
 
-class UpdateArtifact(GQLBase):
+class UpdateArtifact(GQLResult):
     update_artifact: Optional[UpdateArtifactUpdateArtifact] = Field(
         alias="updateArtifact"
     )
 
 
-class UpdateArtifactUpdateArtifact(GQLBase):
+class UpdateArtifactUpdateArtifact(GQLResult):
     artifact: ArtifactFragment
 
 

wandb/sdk/artifacts/_generated/update_artifact_collection_type.py

@@ -0,0 +1,28 @@
+# 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 GQLResult
+
+from .fragments import ArtifactCollectionSummary
+
+
+class UpdateArtifactCollectionType(GQLResult):
+    move_artifact_sequence: Optional[
+        UpdateArtifactCollectionTypeMoveArtifactSequence
+    ] = Field(alias="moveArtifactSequence")
+
+
+class UpdateArtifactCollectionTypeMoveArtifactSequence(GQLResult):
+    artifact_collection: Optional[ArtifactCollectionSummary] = Field(
+        alias="artifactCollection"
+    )
+
+
+UpdateArtifactCollectionType.model_rebuild()
+UpdateArtifactCollectionTypeMoveArtifactSequence.model_rebuild()

wandb/sdk/artifacts/_generated/update_artifact_portfolio.py

@@ -3,32 +3,23 @@
 
 from __future__ import annotations
 
-from typing import Literal, Optional
+from typing import Optional
 
 from pydantic import Field
 
-from wandb._pydantic import GQLBase, GQLId, Typename
+from wandb._pydantic import GQLResult
 
+from .fragments import ArtifactCollectionSummary
 
-class UpdateArtifactPortfolio(GQLBase):
+
+class UpdateArtifactPortfolio(GQLResult):
     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]
+class UpdateArtifactPortfolioUpdateArtifactPortfolio(GQLResult):
+    artifact_collection: ArtifactCollectionSummary = Field(alias="artifactCollection")
 
 
 UpdateArtifactPortfolio.model_rebuild()

wandb/sdk/artifacts/_generated/update_artifact_sequence.py

@@ -3,32 +3,23 @@
 
 from __future__ import annotations
 
-from typing import Literal, Optional
+from typing import Optional
 
 from pydantic import Field
 
-from wandb._pydantic import GQLBase, GQLId, Typename
+from wandb._pydantic import GQLResult
 
+from .fragments import ArtifactCollectionSummary
 
-class UpdateArtifactSequence(GQLBase):
+
+class UpdateArtifactSequence(GQLResult):
     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]
+class UpdateArtifactSequenceUpdateArtifactSequence(GQLResult):
+    artifact_collection: ArtifactCollectionSummary = Field(alias="artifactCollection")
 
 
 UpdateArtifactSequence.model_rebuild()

wandb/sdk/artifacts/_generated/upsert_registry.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 GQLResult
+
+from .fragments import RegistryFragment
+
+
+class UpsertRegistry(GQLResult):
+    upsert_model: Optional[UpsertRegistryUpsertModel] = Field(alias="upsertModel")
+
+
+class UpsertRegistryUpsertModel(GQLResult):
+    inserted: Optional[bool]
+    project: Optional[RegistryFragment]
+
+
+UpsertRegistry.model_rebuild()
+UpsertRegistryUpsertModel.model_rebuild()