wandb 0.21.1__py3-none-manylinux_2_17_aarch64.manylinux2014_aarch64.whl → 0.21.3__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.21.1"
+__version__ = "0.21.3"
 
 
 from wandb.errors import Error

wandb/__init__.pyi

@@ -107,7 +107,7 @@ if TYPE_CHECKING:
     import wandb
     from wandb.plot import CustomChart
 
-__version__: str = "0.21.1"
+__version__: str = "0.21.3"
 
 run: Run | None
 config: wandb_config.Config

wandb/apis/public/api.py

@@ -989,7 +989,7 @@ class Api:
         future releases.
 
         Args:
-            path: The path to project the report resides in. Specify the
+            path: The path to the project the report resides in. Specify the
                 entity that created the project as a prefix followed by a
                 forward slash.
             name: Name of the report requested.
@@ -1007,7 +1007,6 @@ class Api:
 
         wandb.Api.reports("entity/project")
         ```
-
         """
         entity, project, _ = self._parse_path(path + "/fake_run")
 

wandb/apis/public/artifacts.py

@@ -714,7 +714,7 @@ class Artifacts(SizedPaginator["Artifact"]):
 
         self.QUERY = gql_compat(
             PROJECT_ARTIFACTS_GQL,
-            omit_fields=omit_artifact_fields(api=InternalApi()),
+            omit_fields=omit_artifact_fields(),
             rename_fields=rename_fields,
         )
 
@@ -818,15 +818,13 @@ class RunArtifacts(SizedPaginator["Artifact"]):
         if mode == "logged":
             self.run_key = "outputArtifacts"
             self.QUERY = gql_compat(
-                RUN_OUTPUT_ARTIFACTS_GQL,
-                omit_fields=omit_artifact_fields(api=InternalApi()),
+                RUN_OUTPUT_ARTIFACTS_GQL, omit_fields=omit_artifact_fields()
             )
             self._response_cls = RunOutputArtifactsProjectRunOutputArtifacts
         elif mode == "used":
             self.run_key = "inputArtifacts"
             self.QUERY = gql_compat(
-                RUN_INPUT_ARTIFACTS_GQL,
-                omit_fields=omit_artifact_fields(api=InternalApi()),
+                RUN_INPUT_ARTIFACTS_GQL, omit_fields=omit_artifact_fields()
             )
             self._response_cls = RunInputArtifactsProjectRunInputArtifacts
         else:

wandb/apis/public/registries/_utils.py

@@ -1,6 +1,8 @@
+from __future__ import annotations
+
 from enum import Enum
 from functools import lru_cache
-from typing import TYPE_CHECKING, Any, List, Literal, Mapping, Optional, Sequence
+from typing import TYPE_CHECKING, Any, Literal, Mapping, Sequence
 
 from wandb.sdk.artifacts._validators import (
     REGISTRY_PREFIX,
@@ -21,15 +23,12 @@ class Visibility(str, Enum):
 
     @classmethod
     def _missing_(cls, value: object) -> Any:
-        return next(
-            (e for e in cls if e.name == value),
-            None,
-        )
+        return next((e for e in cls if e.name == value), None)
 
 
 def format_gql_artifact_types_input(
-    artifact_types: Optional[List[str]] = None,
-):
+    artifact_types: list[str] | None,
+) -> list[dict[str, str]]:
     """Format the artifact types for the GQL input.
 
     Args:
@@ -40,8 +39,7 @@ def format_gql_artifact_types_input(
     """
     if artifact_types is None:
         return []
-    new_types = validate_artifact_types_list(artifact_types)
-    return [{"name": type} for type in new_types]
+    return [{"name": typ} for typ in validate_artifact_types_list(artifact_types)]
 
 
 def gql_to_registry_visibility(
@@ -67,14 +65,14 @@ def registry_visibility_to_gql(
     """Convert the registry visibility to the GQL visibility."""
     try:
         return Visibility[visibility].value
-    except KeyError:
+    except LookupError:
+        allowed_str = ", ".join(map(repr, (e.name for e in Visibility)))
         raise ValueError(
-            f"Invalid visibility: {visibility!r}. "
-            f"Must be one of: {', '.join(map(repr, (e.name for e in Visibility)))}"
+            f"Invalid visibility: {visibility!r}. Must be one of: {allowed_str}"
         )
 
 
-def ensure_registry_prefix_on_names(query, in_name=False):
+def ensure_registry_prefix_on_names(query: Any, in_name: bool = False) -> Any:
     """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).
@@ -105,7 +103,7 @@ def ensure_registry_prefix_on_names(query, in_name=False):
 
 
 @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:
@@ -127,7 +125,7 @@ def fetch_org_entity_from_organization(client: "Client", organization: str) -> s
         response = client.execute(query, variable_values={"organization": organization})
     except Exception as e:
         raise ValueError(
-            f"Error fetching org entity for organization: {organization}"
+            f"Error fetching org entity for organization: {organization!r}"
         ) from e
 
     if (
@@ -135,6 +133,6 @@ def fetch_org_entity_from_organization(client: "Client", organization: str) -> s
         or not (org_entity := org["orgEntity"])
         or not (org_name := org_entity["name"])
     ):
-        raise ValueError(f"Organization entity for {organization} not found.")
+        raise ValueError(f"Organization entity for {organization!r} not found.")
 
     return org_name