wandb 0.21.0__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/_internal_artifact.py

@@ -1,5 +1,9 @@
+from __future__ import annotations
+
 import re
-from typing import Any, Dict, Final, Optional
+from base64 import urlsafe_b64encode
+from typing import Any, Final
+from zlib import crc32
 
 from wandb.sdk.artifacts.artifact import Artifact
 
@@ -8,14 +12,21 @@ PLACEHOLDER: Final[str] = "PLACEHOLDER"
 
 def sanitize_artifact_name(name: str) -> str:
     """Sanitize the string to satisfy constraints on artifact names."""
-    from wandb.sdk.lib.hashutil import _md5
-
     # If the name is already sanitized, don't change it.
     if (sanitized := re.sub(r"[^a-zA-Z0-9_\-.]+", "", name)) == name:
         return name
 
-    # Append an alphanumeric suffix to maintain uniqueness of the name.
-    suffix = _md5(name.encode("utf-8")).hexdigest()
+    # 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}"
 
 
@@ -31,10 +42,10 @@ class InternalArtifact(Artifact):
         self,
         name: str,
         type: str,
-        description: Optional[str] = None,
-        metadata: Optional[Dict[str, Any]] = None,
+        description: str | None = None,
+        metadata: dict[str, Any] | None = None,
         incremental: bool = False,
-        use_as: Optional[str] = None,
+        use_as: str | None = None,
     ) -> None:
         sanitized_name = sanitize_artifact_name(name)
         super().__init__(

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:
@@ -264,6 +266,14 @@ def is_artifact_registry_project(project: str) -> bool:
     return project.startswith(REGISTRY_PREFIX)
 
 
+def remove_registry_prefix(project: str) -> str:
+    if is_artifact_registry_project(project):
+        return project[len(REGISTRY_PREFIX) :]
+    raise ValueError(
+        f"Project {project!r} does not have the prefix {REGISTRY_PREFIX}. It is not a registry project"
+    )
+
+
 @pydantic_dataclass
 class ArtifactPath:
     #: The collection name.