wandb 0.21.1__py3-none-win32.whl → 0.21.2__py3-none-win32.whl

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/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/_graphql_fragments.py

@@ -1,10 +1,5 @@
 from __future__ import annotations
 
-from textwrap import dedent
-
-from wandb_graphql.language.printer import print_ast
-
-from wandb.apis.public.utils import gql_compat
 from wandb.sdk.internal.internal_api import Api as InternalApi
 
 OMITTABLE_ARTIFACT_FIELDS = frozenset(
@@ -18,85 +13,7 @@ OMITTABLE_ARTIFACT_FIELDS = frozenset(
 )
 
 
-def omit_artifact_fields(api: InternalApi) -> set[str]:
+def omit_artifact_fields(api: InternalApi | None = None) -> set[str]:
     """Return names of Artifact fields to remove from GraphQL requests (for server compatibility)."""
-    allowed_fields = set(api.server_artifact_introspection())
-    return set(OMITTABLE_ARTIFACT_FIELDS - allowed_fields)
-
-
-def _gql_artifact_fragment(include_aliases: bool = True) -> str:
-    """Return a GraphQL query fragment with all parseable Artifact attributes."""
-    omit_fields = omit_artifact_fields(api=InternalApi())
-
-    # Respect the `include_aliases` flag
-    if not include_aliases:
-        omit_fields.add("aliases")
-
-    artifact_fragment_str = dedent(
-        """\
-        fragment ArtifactFragment on Artifact {
-            id
-            artifactSequence {
-                project {
-                    entityName
-                    name
-                }
-                name
-            }
-            versionIndex
-            artifactType {
-                name
-            }
-            description
-            metadata
-            ttlDurationSeconds
-            ttlIsInherited
-            aliases {
-                artifactCollection {
-                    project {
-                        entityName
-                        name
-                    }
-                    name
-                }
-                alias
-            }
-            tags {
-                name
-            }
-            historyStep
-            state
-            currentManifest {
-                file {
-                    directUrl
-                }
-            }
-            commitHash
-            fileCount
-            createdAt
-            updatedAt
-        }"""
-    )
-    compat_doc = gql_compat(artifact_fragment_str, omit_fields=omit_fields)
-    return print_ast(compat_doc)
-
-
-def _gql_registry_fragment() -> str:
-    return """
-        fragment RegistryFragment on Project {
-           id
-            allowAllArtifactTypesInRegistry
-            artifactTypes(includeAll: true) {
-                edges {
-                    node {
-                        name
-                    }
-                }
-            }
-            name
-            description
-            createdAt
-            updatedAt
-            access
-        }
-    """
+    allowed_fields = (api or InternalApi()).server_artifact_introspection()
+    return set(OMITTABLE_ARTIFACT_FIELDS) - set(allowed_fields)

wandb/sdk/artifacts/_validators.py

@@ -12,7 +12,7 @@ from pydantic.dataclasses import dataclass as pydantic_dataclass
 from typing_extensions import Self
 
 from wandb._iterutils import always_list
-from wandb._pydantic import gql_typename
+from wandb._pydantic import from_json, gql_typename
 from wandb.util import json_friendly_val
 
 from ._generated import ArtifactPortfolioTypeFields, ArtifactSequenceTypeFields
@@ -207,9 +207,11 @@ def validate_metadata(metadata: dict[str, Any] | None) -> dict[str, Any]:
     """Validate the artifact metadata and return it as a dict."""
     if metadata is None:
         return {}
-    if not isinstance(metadata, dict):
-        raise TypeError(f"metadata must be dict, not {type(metadata)}")
-    return cast(Dict[str, Any], json.loads(json.dumps(json_friendly_val(metadata))))
+    if isinstance(metadata, str):
+        return from_json(metadata) if metadata else {}
+    if isinstance(metadata, dict):
+        return cast(Dict[str, Any], json.loads(json.dumps(json_friendly_val(metadata))))
+    raise TypeError(f"metadata must be dict, not {type(metadata)}")
 
 
 def validate_ttl_duration_seconds(gql_ttl_duration_seconds: int | None) -> int | None: