wandb 0.20.1rc20250604__py3-none-manylinux_2_17_aarch64.manylinux2014_aarch64.whl → 0.21.0__py3-none-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

wandb/__init__.py

@@ -10,7 +10,7 @@ For reference documentation, see https://docs.wandb.com/ref/python.
 """
 from __future__ import annotations
 
-__version__ = "0.20.1rc20250604"
+__version__ = "0.21.0"
 
 
 from wandb.errors import Error
@@ -30,9 +30,9 @@ wandb.wandb_lib = wandb_sdk.lib  # type: ignore
 
 init = wandb_sdk.init
 setup = wandb_sdk.setup
-_attach = wandb_sdk._attach
+attach = _attach = wandb_sdk._attach
 _sync = wandb_sdk._sync
-_teardown = wandb_sdk.teardown
+teardown = _teardown = wandb_sdk.teardown
 finish = wandb_sdk.finish
 join = finish
 login = wandb_sdk.login
@@ -51,9 +51,6 @@ from wandb.errors import CommError, UsageError
 _preinit = wandb.wandb_lib.preinit  # type: ignore
 _lazyloader = wandb.wandb_lib.lazyloader  # type: ignore
 
-# Call import module hook to set up any needed require hooks
-wandb.sdk.wandb_require._import_module_hook()
-
 from wandb.integration.torch import wandb_torch
 
 # Move this (keras.__init__ expects it at top level)

wandb/__init__.pyi

@@ -12,20 +12,20 @@ For reference documentation, see https://docs.wandb.com/ref/python.
 from __future__ import annotations
 
 __all__ = (
-    "__version__",
+    "__version__",  # doc:exclude
     "init",
     "finish",
     "setup",
     "login",
-    "save",
+    "save",  # doc:exclude
     "sweep",
     "controller",
     "agent",
-    "config",
-    "log",
-    "summary",
+    "config",  # doc:exclude
+    "log",  # doc:exclude
+    "summary",  # doc:exclude
     "Api",
-    "Graph",
+    "Graph",  # doc:exclude
     "Image",
     "Plotly",
     "Video",
@@ -36,26 +36,27 @@ __all__ = (
     "Object3D",
     "Molecule",
     "Histogram",
-    "ArtifactTTL",
-    "log_artifact",
-    "use_artifact",
-    "log_model",
-    "use_model",
-    "link_model",
-    "define_metric",
-    "Error",
-    "termsetup",
-    "termlog",
-    "termerror",
-    "termwarn",
+    "ArtifactTTL",  # doc:exclude
+    "log_artifact",  # doc:exclude
+    "use_artifact",  # doc:exclude
+    "log_model",  # doc:exclude
+    "use_model",  # doc:exclude
+    "link_model",  # doc:exclude
+    "define_metric",  # doc:exclude
+    "Error",  # doc:exclude
+    "termsetup",  # doc:exclude
+    "termlog",  # doc:exclude
+    "termerror",  # doc:exclude
+    "termwarn",  # doc:exclude
     "Artifact",
     "Settings",
     "teardown",
-    "watch",
-    "unwatch",
-    "plot",
+    "watch",  # doc:exclude
+    "unwatch",  # doc:exclude
+    "plot",  # doc:exclude
     "plot_table",
     "restore",
+    "Run",
 )
 
 import os
@@ -106,7 +107,7 @@ if TYPE_CHECKING:
     import wandb
     from wandb.plot import CustomChart
 
-__version__: str = "0.20.1rc20250604"
+__version__: str = "0.21.0"
 
 run: Run | None
 config: wandb_config.Config
@@ -325,7 +326,7 @@ def init(
             the UI.
             If resuming a run, the tags provided here will replace any existing
             tags. To add tags to a resumed run without overwriting the current
-            tags, use `run.tags += ["new_tag"]` after calling `run = wandb.init()`.
+            tags, use `run.tags += ("new_tag",)` after calling `run = wandb.init()`.
         config: Sets `wandb.config`, a dictionary-like object for storing input
             parameters to your run, such as model hyperparameters or data
             preprocessing settings.

wandb/analytics/sentry.py

@@ -15,6 +15,7 @@ from urllib.parse import quote
 import sentry_sdk  # type: ignore
 import sentry_sdk.scope  # type: ignore
 import sentry_sdk.utils  # type: ignore
+from typing_extensions import Never
 
 import wandb
 import wandb.env
@@ -143,7 +144,7 @@ class Sentry:
 
         return event_id
 
-    def reraise(self, exc: Any) -> None:
+    def reraise(self, exc: Any) -> Never:
         """Re-raise an exception after logging it to Sentry.
 
         Use this for top-level exceptions when you want the user to see the traceback.
@@ -209,7 +210,6 @@ class Sentry:
             "sweep_url",
             "sweep_id",
             "deployment",
-            "x_require_legacy_service",
             "launch",
             "_platform",
         )

wandb/apis/importers/internals/internal.py

@@ -364,9 +364,6 @@ def send_run(
     sm = AlternateSendManager(
         settings, sm_record_q, result_q, interface, context_keeper
     )
-    # wm = WriteManager(
-    #     settings, wm_record_q, result_q, sm_record_q, interface, context_keeper
-    # )
 
     if extra_arts or extra_used_arts:
         records = rm.make_artifacts_only_records(extra_arts, extra_used_arts)

wandb/apis/internal.py

@@ -211,6 +211,9 @@ class Api:
     def upsert_run_queue(self, *args, **kwargs):
         return self.api.upsert_run_queue(*args, **kwargs)
 
+    def create_custom_chart(self, *args, **kwargs):
+        return self.api.create_custom_chart(*args, **kwargs)
+
     def update_launch_agent_status(self, *args, **kwargs):
         return self.api.update_launch_agent_status(*args, **kwargs)
 

wandb/apis/paginator.py

@@ -13,6 +13,8 @@ from typing import (
     overload,
 )
 
+import wandb
+
 if TYPE_CHECKING:
     from wandb_graphql.language.ast import Document
 
@@ -112,14 +114,25 @@ class Paginator(Iterator[T]):
 class SizedPaginator(Paginator[T], Sized):
     """A Paginator for objects with a known total count."""
 
+    @property
+    def length(self) -> int | None:
+        wandb.termwarn(
+            (
+                "`.length` is deprecated and will be removed in a future version. "
+                "Use `len(...)` instead."
+            ),
+            repeat=False,
+        )
+        return len(self)
+
     def __len__(self) -> int:
-        if self.length is None:
+        if self._length is None:
             self._load_page()
-        if self.length is None:
+        if self._length is None:
             raise ValueError("Object doesn't provide length")
-        return self.length
+        return self._length
 
     @property
     @abstractmethod
-    def length(self) -> int | None:
+    def _length(self) -> int | None:
         raise NotImplementedError

wandb/apis/public/api.py

@@ -40,9 +40,9 @@ from wandb._iterutils import one
 from wandb.apis import public
 from wandb.apis.normalize import normalize_exceptions
 from wandb.apis.public.const import RETRY_TIMEDELTA
+from wandb.apis.public.registries._utils import fetch_org_entity_from_organization
 from wandb.apis.public.registries.registries_search import Registries
 from wandb.apis.public.registries.registry import Registry
-from wandb.apis.public.registries.utils import _fetch_org_entity_from_organization
 from wandb.apis.public.utils import (
     PathType,
     fetch_org_from_settings_or_entity,
@@ -467,6 +467,85 @@ class Api:
             _default_resource_config=config,
         )
 
+    def create_custom_chart(
+        self,
+        entity: str,
+        name: str,
+        display_name: str,
+        spec_type: Literal["vega2"],
+        access: Literal["private", "public"],
+        spec: Union[str, dict],
+    ) -> str:
+        """Create a custom chart preset and return its id.
+
+        Args:
+            entity: The entity (user or team) that owns the chart
+            name: Unique identifier for the chart preset
+            display_name: Human-readable name shown in the UI
+            spec_type: Type of specification. Must be "vega2" for Vega-Lite v2 specifications.
+            access: Access level for the chart:
+                - "private": Chart is only accessible to the entity that created it
+                - "public": Chart is publicly accessible
+            spec: The Vega/Vega-Lite specification as a dictionary or JSON string
+
+        Returns:
+            The ID of the created chart preset in the format "entity/name"
+
+        Raises:
+            wandb.Error: If chart creation fails
+            UnsupportedError: If the server doesn't support custom charts
+
+        Example:
+            ```python
+            import wandb
+
+            api = wandb.Api()
+
+            # Define a simple bar chart specification
+            vega_spec = {
+                "$schema": "https://vega.github.io/schema/vega-lite/v6.json",
+                "mark": "bar",
+                "data": {"name": "wandb"},
+                "encoding": {
+                    "x": {"field": "${field:x}", "type": "ordinal"},
+                    "y": {"field": "${field:y}", "type": "quantitative"},
+                },
+            }
+
+            # Create the custom chart
+            chart_id = api.create_custom_chart(
+                entity="my-team",
+                name="my-bar-chart",
+                display_name="My Custom Bar Chart",
+                spec_type="vega2",
+                access="private",
+                spec=vega_spec,
+            )
+
+            # Use with wandb.plot_table()
+            chart = wandb.plot_table(
+                vega_spec_name=chart_id,
+                data_table=my_table,
+                fields={"x": "category", "y": "value"},
+            )
+            ```
+        """
+        # Convert user-facing lowercase access to backend uppercase
+        backend_access = access.upper()
+
+        api = InternalApi(retry_timedelta=RETRY_TIMEDELTA)
+        result = api.create_custom_chart(
+            entity=entity,
+            name=name,
+            display_name=display_name,
+            spec_type=spec_type,
+            access=backend_access,
+            spec=spec,
+        )
+        if result is None or result.get("chart") is None:
+            raise wandb.Error("failed to create custom chart")
+        return result["chart"]["id"]
+
     def upsert_run_queue(
         self,
         name: str,
@@ -713,7 +792,7 @@ class Api:
                 return public.BetaReport(
                     self.client,
                     {
-                        "display_name": urllib.parse.unquote(name.replace("-", " ")),
+                        "displayName": urllib.parse.unquote(name.replace("-", " ")),
                         "id": id,
                         "spec": "{}",
                     },
@@ -1514,7 +1593,9 @@ class Api:
 
             Find all collections in the registries with the name "my_collection" and the tag "my_tag"
             ```python
-            api.registries().collections(filter={"name": "my_collection", "tag": "my_tag"})
+            api.registries().collections(
+                filter={"name": "my_collection", "tag": "my_tag"}
+            )
             ```
 
             Find all artifact versions in the registries with a collection name that contains "my_collection" and a version that has the alias "best"
@@ -1589,7 +1670,7 @@ class Api:
         organization = organization or fetch_org_from_settings_or_entity(
             self.settings, self.default_entity
         )
-        org_entity = _fetch_org_entity_from_organization(self.client, organization)
+        org_entity = fetch_org_entity_from_organization(self.client, organization)
         registry = Registry(self.client, organization, org_entity, name)
         registry.load()
         return registry

wandb/apis/public/artifacts.py

@@ -100,7 +100,7 @@ class ArtifactTypes(Paginator["ArtifactType"]):
         self.last_response = ArtifactTypesFragment.model_validate(conn)
 
     @property
-    def length(self) -> None:
+    def _length(self) -> None:
         # TODO
         return None
 
@@ -240,9 +240,9 @@ class ArtifactCollections(SizedPaginator["ArtifactCollection"]):
         self.last_response = ArtifactCollectionsFragment.model_validate(conn)
 
     @property
-    def length(self):
+    def _length(self) -> int:
         if self.last_response is None:
-            return None
+            self._load_page()
         return self.last_response.total_count
 
     @property
@@ -608,9 +608,9 @@ class Artifacts(SizedPaginator["Artifact"]):
         self.last_response = ArtifactsFragment.model_validate(conn)
 
     @property
-    def length(self) -> int | None:
+    def _length(self) -> int:
         if self.last_response is None:
-            return None
+            self._load_page()
         return self.last_response.total_count
 
     @property
@@ -698,9 +698,9 @@ class RunArtifacts(SizedPaginator["Artifact"]):
         self.last_response = self._response_cls.model_validate(inner_data)
 
     @property
-    def length(self) -> int | None:
+    def _length(self) -> int:
         if self.last_response is None:
-            return None
+            self._load_page()
         return self.last_response.total_count
 
     @property
@@ -799,7 +799,9 @@ class ArtifactFiles(SizedPaginator["public.File"]):
         return [self.artifact.entity, self.artifact.project, self.artifact.name]
 
     @property
-    def length(self) -> int:
+    def _length(self) -> int:
+        if self.last_response is None:
+            self._load_page()
         return self.artifact.file_count
 
     @property

wandb/apis/public/files.py

@@ -72,11 +72,11 @@ class Files(SizedPaginator["File"]):
         super().__init__(client, variables, per_page)
 
     @property
-    def length(self):
-        if self.last_response:
-            return self.last_response["project"]["run"]["fileCount"]
-        else:
-            return None
+    def _length(self):
+        if not self.last_response:
+            self._load_page()
+
+        return self.last_response["project"]["run"]["fileCount"]
 
     @property
     def more(self):

wandb/apis/public/projects.py

@@ -9,6 +9,7 @@ from wandb.apis import public
 from wandb.apis.attrs import Attrs
 from wandb.apis.normalize import normalize_exceptions
 from wandb.apis.paginator import Paginator
+from wandb.apis.public.api import RetryingClient
 from wandb.sdk.lib import ipython
 
 PROJECT_FRAGMENT = """fragment ProjectFragment on Project {
@@ -43,7 +44,19 @@ class Projects(Paginator["Project"]):
         """.format(PROJECT_FRAGMENT)
     )
 
-    def __init__(self, client, entity, per_page=50):
+    def __init__(
+        self,
+        client: RetryingClient,
+        entity: str,
+        per_page: int = 50,
+    ) -> "Projects":
+        """An iterable collection of `Project` objects.
+
+        Args:
+            client: The API client used to query W&B.
+            entity: The entity which owns the projects.
+            per_page: The number of projects to fetch per request to the API.
+        """
         self.client = client
         self.entity = entity
         variables = {
@@ -83,7 +96,31 @@ class Projects(Paginator["Project"]):
 class Project(Attrs):
     """A project is a namespace for runs."""
 
-    def __init__(self, client, entity, project, attrs):
+    QUERY = gql(
+        """
+        query Project($project: String!, $entity: String!) {
+            project(name: $project, entityName: $entity) {
+                id
+            }
+        }
+        """
+    )
+
+    def __init__(
+        self,
+        client: RetryingClient,
+        entity: str,
+        project: str,
+        attrs: dict,
+    ) -> "Project":
+        """A single project associated with an entity.
+
+        Args:
+            client: The API client used to query W&B.
+            entity: The entity which owns the project.
+            project: The name of the project to query.
+            attrs: The attributes of the project.
+        """
         super().__init__(dict(attrs))
         self.client = client
         self.name = project
@@ -143,7 +180,7 @@ class Project(Attrs):
         )
         variable_values = {"project": self.name, "entity": self.entity}
         ret = self.client.execute(query, variable_values)
-        if ret["project"]["totalSweeps"] < 1:
+        if not ret.get("project") or ret["project"]["totalSweeps"] < 1:
             return []
 
         return [
@@ -178,6 +215,10 @@ class Project(Attrs):
         variable_values = {"projectName": self.name, "entityName": self.entity}
         try:
             data = self.client.execute(self._PROJECT_ID, variable_values)
+
+            if not data.get("project") or not data["project"].get("id"):
+                raise ValueError(f"Project {self.name} not found")
+
             self._attrs["id"] = data["project"]["id"]
             return self._attrs["id"]
         except (HTTPError, LookupError, TypeError) as e:

wandb/apis/public/registries/{utils.py → _utils.py}

@@ -13,7 +13,7 @@ if TYPE_CHECKING:
 from wandb_gql import gql
 
 
-class _Visibility(str, Enum):
+class Visibility(str, Enum):
     # names are what users see/pass into Python methods
     # values are what's expected by backend API
     organization = "PRIVATE"
@@ -27,7 +27,7 @@ class _Visibility(str, Enum):
         )
 
 
-def _format_gql_artifact_types_input(
+def format_gql_artifact_types_input(
     artifact_types: Optional[List[str]] = None,
 ):
     """Format the artifact types for the GQL input.
@@ -44,7 +44,7 @@ def _format_gql_artifact_types_input(
     return [{"name": type} for type in new_types]
 
 
-def _gql_to_registry_visibility(
+def gql_to_registry_visibility(
     visibility: str,
 ) -> Literal["organization", "restricted"]:
     """Convert the GQL visibility to the registry visibility.
@@ -56,25 +56,25 @@ def _gql_to_registry_visibility(
         The registry visibility.
     """
     try:
-        return _Visibility(visibility).name
+        return Visibility(visibility).name
     except ValueError:
         raise ValueError(f"Invalid visibility: {visibility!r} from backend")
 
 
-def _registry_visibility_to_gql(
+def registry_visibility_to_gql(
     visibility: Literal["organization", "restricted"],
 ) -> str:
     """Convert the registry visibility to the GQL visibility."""
     try:
-        return _Visibility[visibility].value
+        return Visibility[visibility].value
     except KeyError:
         raise ValueError(
             f"Invalid visibility: {visibility!r}. "
-            f"Must be one of: {', '.join(map(repr, (e.name for e in _Visibility)))}"
+            f"Must be one of: {', '.join(map(repr, (e.name for e in Visibility)))}"
         )
 
 
-def _ensure_registry_prefix_on_names(query, in_name=False):
+def ensure_registry_prefix_on_names(query, in_name=False):
     """Traverse the filter to prepend the `name` key value with the registry prefix unless the value is a regex.
 
     - in_name: True if we are under a "name" key (or propagating from one).
@@ -89,23 +89,23 @@ def _ensure_registry_prefix_on_names(query, in_name=False):
         new_dict = {}
         for key, obj in dct.items():
             if key == "name":
-                new_dict[key] = _ensure_registry_prefix_on_names(obj, in_name=True)
+                new_dict[key] = ensure_registry_prefix_on_names(obj, in_name=True)
             elif key == "$regex":
                 # For regex operator, we skip transformation of its value.
                 new_dict[key] = obj
             else:
                 # For any other key, propagate the in_name and skip_transform flags as-is.
-                new_dict[key] = _ensure_registry_prefix_on_names(obj, in_name=in_name)
+                new_dict[key] = ensure_registry_prefix_on_names(obj, in_name=in_name)
         return new_dict
     if isinstance((objs := query), Sequence):
         return list(
-            map(lambda x: _ensure_registry_prefix_on_names(x, in_name=in_name), objs)
+            map(lambda x: ensure_registry_prefix_on_names(x, in_name=in_name), objs)
         )
     return query
 
 
 @lru_cache(maxsize=10)
-def _fetch_org_entity_from_organization(client: "Client", organization: str) -> str:
+def fetch_org_entity_from_organization(client: "Client", organization: str) -> str:
     """Fetch the org entity from the organization.
 
     Args:

wandb/apis/public/registries/registries_search.py

@@ -15,7 +15,7 @@ from wandb.sdk.artifacts._graphql_fragments import (
     _gql_registry_fragment,
 )
 
-from .utils import _ensure_registry_prefix_on_names
+from ._utils import ensure_registry_prefix_on_names
 
 
 class Registries(Paginator):
@@ -54,7 +54,7 @@ class Registries(Paginator):
     ):
         self.client = client
         self.organization = organization
-        self.filter = _ensure_registry_prefix_on_names(filter or {})
+        self.filter = ensure_registry_prefix_on_names(filter or {})
         variables = {
             "organization": organization,
             "filters": json.dumps(self.filter),

wandb/apis/public/registries/registry.py

@@ -3,26 +3,27 @@ from typing import TYPE_CHECKING, Any, Dict, List, Literal, Optional
 from wandb_gql import gql
 
 import wandb
-from wandb.apis.public.registries._freezable_list import AddOnlyArtifactTypesList
-from wandb.apis.public.registries.registries_search import Collections, Versions
-from wandb.apis.public.registries.utils import (
-    _fetch_org_entity_from_organization,
-    _format_gql_artifact_types_input,
-    _gql_to_registry_visibility,
-    _registry_visibility_to_gql,
-)
 from wandb.proto.wandb_internal_pb2 import ServerFeature
 from wandb.sdk.artifacts._validators import REGISTRY_PREFIX, validate_project_name
 from wandb.sdk.internal.internal_api import Api as InternalApi
-from wandb.sdk.projects._generated.delete_project import DeleteProject
-from wandb.sdk.projects._generated.operations import (
+from wandb.sdk.projects._generated import (
     DELETE_PROJECT_GQL,
     FETCH_REGISTRY_GQL,
     RENAME_PROJECT_GQL,
     UPSERT_REGISTRY_PROJECT_GQL,
+    DeleteProject,
+    RenameProject,
+    UpsertRegistryProject,
+)
+
+from ._freezable_list import AddOnlyArtifactTypesList
+from ._utils import (
+    fetch_org_entity_from_organization,
+    format_gql_artifact_types_input,
+    gql_to_registry_visibility,
+    registry_visibility_to_gql,
 )
-from wandb.sdk.projects._generated.rename_project import RenameProject
-from wandb.sdk.projects._generated.upsert_registry_project import UpsertRegistryProject
+from .registries_search import Collections, Versions
 
 if TYPE_CHECKING:
     from wandb_gql import Client
@@ -62,7 +63,7 @@ class Registry:
         )
         self._created_at = attrs.get("createdAt", "")
         self._updated_at = attrs.get("updatedAt", "")
-        self._visibility = _gql_to_registry_visibility(attrs.get("access", ""))
+        self._visibility = gql_to_registry_visibility(attrs.get("access", ""))
 
     @property
     def full_name(self) -> str:
@@ -220,13 +221,13 @@ class Registry:
             ValueError: If a registry with the same name already exists in the
                 organization or if the creation fails.
         """
-        org_entity = _fetch_org_entity_from_organization(client, organization)
+        org_entity = fetch_org_entity_from_organization(client, organization)
         full_name = REGISTRY_PREFIX + name
         validate_project_name(full_name)
         accepted_artifact_types = []
         if artifact_types:
-            accepted_artifact_types = _format_gql_artifact_types_input(artifact_types)
-        visibility_value = _registry_visibility_to_gql(visibility)
+            accepted_artifact_types = format_gql_artifact_types_input(artifact_types)
+        visibility_value = registry_visibility_to_gql(visibility)
         registry_creation_error = (
             f"Failed to create registry {name!r} in organization {organization!r}."
         )
@@ -310,8 +311,8 @@ class Registry:
             )
 
         validate_project_name(self.full_name)
-        visibility_value = _registry_visibility_to_gql(self.visibility)
-        newly_added_types = _format_gql_artifact_types_input(self.artifact_types.draft)
+        visibility_value = registry_visibility_to_gql(self.visibility)
+        newly_added_types = format_gql_artifact_types_input(self.artifact_types.draft)
         registry_save_error = f"Failed to save and update registry: {self.name} in organization: {self.organization}"
         full_saved_name = f"{REGISTRY_PREFIX}{self._saved_name}"
         try: