wandb 0.19.11__py3-none-win32.whl → 0.20.0__py3-none-win32.whl

wandb/__init__.py

@@ -10,7 +10,7 @@ For reference documentation, see https://docs.wandb.com/ref/python.
 """
 from __future__ import annotations
 
-__version__ = "0.19.11"
+__version__ = "0.20.0"
 
 
 from wandb.errors import Error
@@ -169,7 +169,6 @@ gym = _lazyloader.LazyLoader("wandb.gym", globals(), "wandb.integration.gym")
 lightgbm = _lazyloader.LazyLoader(
     "wandb.lightgbm", globals(), "wandb.integration.lightgbm"
 )
-docker = _lazyloader.LazyLoader("wandb.docker", globals(), "wandb.docker")
 jupyter = _lazyloader.LazyLoader("wandb.jupyter", globals(), "wandb.jupyter")
 sacred = _lazyloader.LazyLoader("wandb.sacred", globals(), "wandb.integration.sacred")
 

wandb/__init__.pyi

@@ -106,7 +106,7 @@ if TYPE_CHECKING:
     import wandb
     from wandb.plot import CustomChart
 
-__version__: str = "0.19.11"
+__version__: str = "0.20.0"
 
 run: Run | None
 config: wandb_config.Config
@@ -512,7 +512,6 @@ def log(
     data: dict[str, Any],
     step: int | None = None,
     commit: bool | None = None,
-    sync: bool | None = None,
 ) -> None:
     """Upload run data.
 
@@ -617,7 +616,6 @@ def log(
             accumulate data for the step. See the notes in the description.
             If `step` is `None`, then the default is `commit=True`;
             otherwise, the default is `commit=False`.
-        sync: This argument is deprecated and does nothing.
 
     Examples:
         For more and more detailed examples, see
@@ -748,7 +746,7 @@ def log(
     ...
 
 def save(
-    glob_str: str | os.PathLike | None = None,
+    glob_str: str | os.PathLike,
     base_path: str | os.PathLike | None = None,
     policy: PolicyName = "live",
 ) -> bool | list[str]:
@@ -978,8 +976,7 @@ def use_artifact(
             You can also pass an Artifact object created by calling `wandb.Artifact`
         type: (str, optional) The type of artifact to use.
         aliases: (list, optional) Aliases to apply to this artifact
-        use_as: (string, optional) Optional string indicating what purpose the artifact was used with.
-                                   Will be shown in UI.
+        use_as: This argument is deprecated and does nothing.
 
     Returns:
         An `Artifact` object.

wandb/_iterutils.py

@@ -1,14 +1,31 @@
 from __future__ import annotations
 
-from typing import Iterable, TypeVar
+from typing import TYPE_CHECKING, Any, Iterable, TypeVar, Union, overload
 
-T = TypeVar("T")
+if TYPE_CHECKING:
+    T = TypeVar("T")
+    ClassInfo = Union[type[T], tuple[type[T], ...]]
+
+
+@overload
+def always_list(obj: Iterable[T], base_type: ClassInfo = ...) -> list[T]: ...
+@overload
+def always_list(obj: T, base_type: ClassInfo = ...) -> list[T]: ...
+def always_list(obj: Any, base_type: Any = (str, bytes)) -> list[T]:
+    """Return a guaranteed list of objects from a single instance OR iterable of such objects.
+
+    By default, assume the returned list should have string-like elements (i.e. `str`/`bytes`).
+
+    Adapted from `more_itertools.always_iterable`, but simplified for internal use.  See:
+    https://more-itertools.readthedocs.io/en/stable/api.html#more_itertools.always_iterable
+    """
+    return [obj] if isinstance(obj, base_type) else list(obj)
 
 
 def one(
     iterable: Iterable[T],
-    too_short: Exception | None = None,
-    too_long: Exception | None = None,
+    too_short: type[Exception] | Exception | None = None,
+    too_long: type[Exception] | Exception | None = None,
 ) -> T:
     """Return the only item in the iterable.
 
@@ -35,12 +52,14 @@ def one(
     it = iter(iterable)
     try:
         obj = next(it)
-    except StopIteration as e:
-        raise too_short or ValueError("Expected 1 item in iterable, got 0") from e
+    except StopIteration:
+        raise (too_short or ValueError("Expected 1 item in iterable, got 0")) from None
 
     # ...the second item doesn't
     try:
         _ = next(it)
     except StopIteration:
         return obj
-    raise too_long or ValueError("Expected 1 item in iterable, got multiple")
+    raise (
+        too_long or ValueError("Expected 1 item in iterable, got multiple")
+    ) from None

wandb/_pydantic/__init__.py

@@ -8,7 +8,7 @@ from .base import (
     Typename,
     ensure_json,
 )
-from .utils import IS_PYDANTIC_V2, from_json, pydantic_isinstance, to_json
+from .utils import IS_PYDANTIC_V2, from_json, gql_typename, pydantic_isinstance, to_json
 from .v1_compat import AliasChoices, computed_field, field_validator, model_validator
 
 __all__ = [
@@ -26,4 +26,5 @@ __all__ = [
     "to_json",
     "from_json",
     "ensure_json",
+    "gql_typename",
 ]

wandb/_pydantic/utils.py

@@ -20,6 +20,13 @@ IS_PYDANTIC_V2: bool = int(pydantic_major) >= 2
 BaseModelType: TypeAlias = Type[BaseModel]
 
 
+def gql_typename(cls: type[BaseModel]) -> str:
+    """Get the GraphQL typename for a Pydantic model."""
+    if (field := cls.model_fields.get("typename__")) and (typename := field.default):
+        return typename
+    raise TypeError(f"Cannot extract GraphQL typename from: {cls.__qualname__!r}.")
+
+
 if IS_PYDANTIC_V2:
     import pydantic_core  # pydantic_core is only installed by pydantic v2
 

wandb/agents/pyagent.py

@@ -176,7 +176,7 @@ class Agent:
                     return
             time.sleep(5)
 
-    def _run_jobs_from_queue(self):  # noqa:C901
+    def _run_jobs_from_queue(self):
         global _INSTANCES
         _INSTANCES += 1
         try:
@@ -239,9 +239,7 @@ class Agent:
                         elif (
                             time.time() - self._start_time < self.FLAPPING_MAX_SECONDS
                         ) and (len(self._exceptions) >= self.FLAPPING_MAX_FAILURES):
-                            msg = "Detected {} failed runs in the first {} seconds, killing sweep.".format(
-                                self.FLAPPING_MAX_FAILURES, self.FLAPPING_MAX_SECONDS
-                            )
+                            msg = f"Detected {self.FLAPPING_MAX_FAILURES} failed runs in the first {self.FLAPPING_MAX_SECONDS} seconds, killing sweep."
                             logger.error(msg)
                             wandb.termerror(msg)
                             wandb.termlog(
@@ -253,9 +251,7 @@ class Agent:
                             self._max_initial_failures < len(self._exceptions)
                             and len(self._exceptions) >= count
                         ):
-                            msg = "Detected {} failed runs in a row at start, killing sweep.".format(
-                                self._max_initial_failures
-                            )
+                            msg = f"Detected {self._max_initial_failures} failed runs in a row at start, killing sweep."
                             logger.error(msg)
                             wandb.termerror(msg)
                             wandb.termlog(
@@ -272,13 +268,13 @@ class Agent:
                     wandb.termlog("Ctrl + C detected. Stopping sweep.")
                     self._exit()
                     return
-                except Exception as e:
+                except Exception:
                     if self._exit_flag:
                         logger.debug("Exiting main loop due to exit flag.")
                         wandb.termlog("Sweep Agent: Killed.")
                         return
                     else:
-                        raise e
+                        raise
         finally:
             _INSTANCES -= 1
 
@@ -305,8 +301,8 @@ class Agent:
 
             self._function()
             wandb.finish()
-        except KeyboardInterrupt as ki:
-            raise ki
+        except KeyboardInterrupt:
+            raise
         except Exception as e:
             wandb.finish(exit_code=1)
             if self._run_status[run_id] == RunStatus.RUNNING:
@@ -320,9 +316,7 @@ class Agent:
 
     def run(self):
         logger.info(
-            "Starting sweep agent: entity={}, project={}, count={}".format(
-                self._entity, self._project, self._count
-            )
+            f"Starting sweep agent: entity={self._entity}, project={self._project}, count={self._count}"
         )
         self._setup()
         # self._main_thread = threading.Thread(target=self._run_jobs_from_queue)
@@ -345,7 +339,7 @@ def pyagent(sweep_id, function, entity=None, project=None, count=None):
         count (int, optional): the number of trials to run.
     """
     if not callable(function):
-        raise Exception("function parameter must be callable!")
+        raise TypeError("function parameter must be callable!")
     agent = Agent(
         sweep_id,
         function=function,

wandb/analytics/sentry.py

@@ -209,7 +209,6 @@ class Sentry:
             "sweep_url",
             "sweep_id",
             "deployment",
-            "x_disable_service",
             "x_require_legacy_service",
             "launch",
             "_platform",
@@ -256,7 +255,7 @@ class Sentry:
 
         email = tags.get("email")
         if email:
-            self.scope.user = {"email": email}  # noqa
+            self.scope.user = {"email": email}
 
         # todo: add back the option to pass general tags see: c645f625d1c1a3db4a6b0e2aa8e924fee101904c (wandb/util.py)
 

wandb/apis/attrs.py

@@ -28,14 +28,13 @@ class Attrs:
 
         try:
             from IPython import display
-
-            display.display(display.HTML(html))
-            return True
-
         except ImportError:
             wandb.termwarn(".display() only works in jupyter environments")
             return False
 
+        display.display(display.HTML(html))
+        return True
+
     def to_html(self, *args, **kwargs):
         return None
 

wandb/apis/importers/internals/util.py

@@ -55,7 +55,7 @@ def parallelize(
                 f"Exception: {func=} {args=} {kwargs=} {e=} {filename=} {lineno=}. {traceback_details=}"
             )
             if raise_on_error:
-                raise e
+                raise
 
     results = []
     with ThreadPoolExecutor(max_workers) as exc:

wandb/apis/importers/validation.py

@@ -92,8 +92,8 @@ def _check_entry_is_downloable(entry):
 
     try:
         resp = requests.head(url, allow_redirects=True)
-    except Exception as e:
-        logger.error(f"Problem validating {entry=}, {e=}")
+    except Exception:
+        logger.exception(f"Problem validating {entry=}")
         return False
 
     if resp.status_code != 200:

wandb/apis/importers/wandb.py

@@ -151,7 +151,7 @@ class WandbRun:
         if self._parquet_history_paths:
             rows = self._get_rows_from_parquet_history_paths()
         else:
-            logger.warn(
+            logger.warning(
                 "No parquet files detected; using scan history (this may not be reliable)"
             )
             rows = self.run.scan_history()
@@ -453,13 +453,15 @@ class WandbImporter:
         try:
             dst_collection = self.dst_api.artifact_collection(art_type, art_name)
         except (wandb.CommError, ValueError):
-            logger.warn(f"Collection doesn't exist {art_type=}, {art_name=}")
+            logger.warning(f"Collection doesn't exist {art_type=}, {art_name=}")
             return
 
         try:
             dst_collection.delete()
         except (wandb.CommError, ValueError) as e:
-            logger.warn(f"Collection can't be deleted, {art_type=}, {art_name=}, {e=}")
+            logger.warning(
+                f"Collection can't be deleted, {art_type=}, {art_name=}, {e=}"
+            )
             return
 
     def _import_artifact_sequence(
@@ -475,7 +477,7 @@ class WandbImporter:
         if not seq.artifacts:
             # The artifact sequence has no versions.  This usually means all artifacts versions were deleted intentionally,
             # but it can also happen if the sequence represents run history and that run was deleted.
-            logger.warn(f"Artifact {seq=} has no artifacts, skipping.")
+            logger.warning(f"Artifact {seq=} has no artifacts, skipping.")
             return
 
         if namespace is None:
@@ -517,7 +519,7 @@ class WandbImporter:
 
                 # Could be logged by None (rare) or ValueError
                 if wandb_run is None:
-                    logger.warn(
+                    logger.warning(
                         f"Run for {art.name=} does not exist (deleted?), using {run_or_dummy=}"
                     )
                     wandb_run = run_or_dummy
@@ -546,10 +548,12 @@ class WandbImporter:
             retry_arts_func = internal.exp_retry(self._dst_api.artifacts)
             dst_arts = list(retry_arts_func(seq.type_, seq.name))
         except wandb.CommError:
-            logger.warn(f"{seq=} does not exist in dst.  Has it already been deleted?")
+            logger.warning(
+                f"{seq=} does not exist in dst.  Has it already been deleted?"
+            )
             return
-        except TypeError as e:
-            logger.error(f"Problem getting dst versions (try again later) {e=}")
+        except TypeError:
+            logger.exception("Problem getting dst versions (try again later).")
             return
 
         for art in dst_arts:
@@ -562,9 +566,9 @@ class WandbImporter:
                 art.delete(delete_aliases=True)
             except wandb.CommError as e:
                 if "cannot delete system managed artifact" in str(e):
-                    logger.warn("Cannot delete system managed artifact")
+                    logger.warning("Cannot delete system managed artifact")
                 else:
-                    raise e
+                    raise
 
     def _get_dst_art(
         self, src_art: Run, entity: Optional[str] = None, project: Optional[str] = None
@@ -701,7 +705,7 @@ class WandbImporter:
                 )
             except ValueError as e:
                 if "Could not find project" in str(e):
-                    logger.error("Could not find project, does it exist?")
+                    logger.exception("Could not find project, does it exist?")
                     continue
 
             for run in runs:
@@ -732,7 +736,7 @@ class WandbImporter:
             api.create_project(project, entity)
         except requests.exceptions.HTTPError as e:
             if e.response.status_code != 409:
-                logger.warn(f"Issue upserting {entity=}/{project=}, {e=}")
+                logger.warning(f"Issue upserting {entity=}/{project=}, {e=}")
 
         logger.debug(f"Upserting report {entity=}, {project=}, {name=}, {title=}")
         api.client.execute(
@@ -1169,8 +1173,8 @@ class WandbImporter:
             for art in seq:
                 try:
                     logged_by = _get_run_or_dummy_from_art(art, self.src_api)
-                except requests.HTTPError as e:
-                    logger.error(f"Failed to get run, skipping: {art=}, {e=}")
+                except requests.HTTPError:
+                    logger.exception(f"Failed to get run, skipping: {art=}")
                     continue
 
                 if art.type == "wandb-history" and isinstance(logged_by, _DummyRun):
@@ -1215,9 +1219,10 @@ class WandbImporter:
                     art = seq.artifacts[0]
                     try:
                         logged_by = _get_run_or_dummy_from_art(art, self.src_api)
-                    except requests.HTTPError as e:
-                        logger.error(
-                            f"Validate Artifact http error: {art.entity=}, {art.project=}, {art.name=}, {e=}"
+                    except requests.HTTPError:
+                        logger.exception(
+                            f"Validate Artifact http error: {art.entity=},"
+                            f" {art.project=}, {art.name=}"
                         )
                         continue
 
@@ -1347,8 +1352,8 @@ class WandbImporter:
                 types = []
                 try:
                     types = [t for t in api.artifact_types(ns.path)]
-                except Exception as e:
-                    logger.error(f"Failed to get artifact types {e=}")
+                except Exception:
+                    logger.exception("Failed to get artifact types.")
 
                 for t in types:
                     collections = []
@@ -1359,8 +1364,8 @@ class WandbImporter:
 
                     try:
                         collections = t.collections()
-                    except Exception as e:
-                        logger.error(f"Failed to get artifact collections {e=}")
+                    except Exception:
+                        logger.exception("Failed to get artifact collections.")
 
                     for c in collections:
                         if c.is_sequence():
@@ -1471,7 +1476,7 @@ def _read_ndjson(fname: str) -> Optional[pl.DataFrame]:
             return None
         if "error parsing ndjson" in str(e):
             return None
-        raise e
+        raise
 
     return df
 
@@ -1482,7 +1487,7 @@ def _get_run_or_dummy_from_art(art: Artifact, api=None):
     try:
         run = art.logged_by()
     except ValueError as e:
-        logger.warn(
+        logger.warning(
             f"Can't log artifact because run doesn't exist, {art=}, {run=}, {e=}"
         )
 
@@ -1538,8 +1543,8 @@ def _download_art(art: Artifact, root: str) -> Optional[str]:
     try:
         with patch("click.echo"):
             return art.download(root=root, skip_cache=True)
-    except Exception as e:
-        logger.error(f"Error downloading artifact {art=}, {e=}")
+    except Exception:
+        logger.exception(f"Error downloading artifact {art=}")
 
 
 def _clone_art(art: Artifact, root: Optional[str] = None):

wandb/apis/normalize.py

@@ -61,8 +61,8 @@ def normalize_exceptions(func: _F) -> _F:
                 raise CommError(message, err.last_exception).with_traceback(
                     sys.exc_info()[2]
                 )
-        except Error as err:
-            raise err
+        except Error:
+            raise
         except Exception as err:
             # gql raises server errors with dict's as strings...
             if len(err.args) > 0:

wandb/apis/public/__init__.py

@@ -22,6 +22,7 @@ from wandb.apis.public.jobs import (
 )
 from wandb.apis.public.projects import PROJECT_FRAGMENT, Project, Projects
 from wandb.apis.public.query_generator import QueryGenerator
+from wandb.apis.public.registries.registry import Registry
 from wandb.apis.public.reports import (
     BetaReport,
     PanelMetricsHelper,

wandb/apis/public/api.py

@@ -103,7 +103,9 @@ class RetryingClient:
         check_retry_fn=util.no_retry_auth,
         retryable_exceptions=(RetryError, requests.RequestException),
     )
-    def execute(self, *args, **kwargs):  # noqa: D102  # User not encouraged to use this class directly
+    def execute(
+        self, *args, **kwargs
+    ):  # User not encouraged to use this class directly
         try:
             return self._client.execute(*args, **kwargs)
         except requests.exceptions.ReadTimeout:
@@ -122,10 +124,12 @@ class RetryingClient:
             self._server_info = self.execute(self.INFO_QUERY).get("serverInfo")
         return self._server_info
 
-    def version_supported(self, min_version: str) -> bool:  # noqa: D102  # User not encouraged to use this class directly
-        from wandb.util import parse_version
+    def version_supported(
+        self, min_version: str
+    ) -> bool:  # User not encouraged to use this class directly
+        from packaging.version import parse
 
-        return parse_version(min_version) <= parse_version(
+        return parse(min_version) <= parse(
             self.server_info["cliVersionInfo"]["max_cli_version"]
         )
 
@@ -272,20 +276,21 @@ class Api:
         api_key: Optional[str] = None,
     ) -> None:
         self.settings = InternalApi().settings()
+
         _overrides = overrides or {}
-        self._api_key = api_key
-        if self.api_key is None and _thread_local_api_settings.cookies is None:
-            wandb.login(host=_overrides.get("base_url"))
         self.settings.update(_overrides)
+        self.settings["base_url"] = self.settings["base_url"].rstrip("/")
+        if "organization" in _overrides:
+            self.settings["organization"] = _overrides["organization"]
         if "username" in _overrides and "entity" not in _overrides:
             wandb.termwarn(
                 'Passing "username" to Api is deprecated. please use "entity" instead.'
             )
             self.settings["entity"] = _overrides["username"]
-        self.settings["base_url"] = self.settings["base_url"].rstrip("/")
 
-        if "organization" in _overrides:
-            self.settings["organization"] = _overrides["organization"]
+        self._api_key = api_key
+        if self.api_key is None and _thread_local_api_settings.cookies is None:
+            wandb.login(host=_overrides.get("base_url"))
 
         self._viewer = None
         self._projects = {}
@@ -318,7 +323,6 @@ class Api:
             )
         )
         self._client = RetryingClient(self._base_client)
-        self._server_features_cache: Optional[dict[str, bool]] = None
 
     def create_project(self, name: str, entity: str) -> None:
         """Create a new project.
@@ -1464,10 +1468,11 @@ class Api:
         """
         try:
             self._artifact(name, type)
-            return True
         except wandb.errors.CommError:
             return False
 
+        return True
+
     @normalize_exceptions
     def artifact_collection_exists(self, name: str, type: str) -> bool:
         """Return whether an artifact collection exists within a specified project and entity.
@@ -1483,10 +1488,11 @@ class Api:
         """
         try:
             self.artifact_collection(type, name)
-            return True
         except wandb.errors.CommError:
             return False
 
+        return True
+
     def registries(
         self,
         organization: Optional[str] = None,
@@ -1539,9 +1545,7 @@ class Api:
         Returns:
             A registry iterator.
         """
-        if not InternalApi()._check_server_feature_with_fallback(
-            ServerFeature.ARTIFACT_REGISTRY_SEARCH
-        ):
+        if not InternalApi()._server_supports(ServerFeature.ARTIFACT_REGISTRY_SEARCH):
             raise RuntimeError(
                 "Registry search API is not enabled on this wandb server version. "
                 "Please upgrade your server version or contact support at support@wandb.com."
@@ -1577,9 +1581,7 @@ class Api:
             registry.save()
             ```
         """
-        if not InternalApi()._check_server_feature_with_fallback(
-            ServerFeature.ARTIFACT_REGISTRY_SEARCH
-        ):
+        if not InternalApi()._server_supports(ServerFeature.ARTIFACT_REGISTRY_SEARCH):
             raise RuntimeError(
                 "api.registry() is not enabled on this wandb server version. "
                 "Please upgrade your server version or contact support at support@wandb.com."
@@ -1635,7 +1637,7 @@ class Api:
             )
             ```
         """
-        if not InternalApi()._check_server_feature_with_fallback(
+        if not InternalApi()._server_supports(
             ServerFeature.INCLUDE_ARTIFACT_TYPES_IN_REGISTRY_CREATION
         ):
             raise RuntimeError(
@@ -1783,19 +1785,18 @@ class Api:
             ALWAYS_SUPPORTED_EVENTS,
         )
 
-        server_features = InternalApi()._server_features()
-        return bool(
-            (
-                (event is None)
-                or (event in ALWAYS_SUPPORTED_EVENTS)
-                or server_features.get(f"AUTOMATION_EVENT_{event.value}")
-            )
-            and (
-                (action is None)
-                or (action in ALWAYS_SUPPORTED_ACTIONS)
-                or server_features.get(f"AUTOMATION_ACTION_{action.value}")
-            )
+        api = InternalApi()
+        supports_event = (
+            (event is None)
+            or (event in ALWAYS_SUPPORTED_EVENTS)
+            or api._server_supports(f"AUTOMATION_EVENT_{event.value}")
+        )
+        supports_action = (
+            (action is None)
+            or (action in ALWAYS_SUPPORTED_ACTIONS)
+            or api._server_supports(f"AUTOMATION_ACTION_{action.value}")
         )
+        return supports_event and supports_action
 
     def _omitted_automation_fragments(self) -> Set[str]:
         """Returns the names of unsupported automation-related fragments.
@@ -1938,6 +1939,7 @@ class Api:
             iterator = filter(lambda x: x.name == name, iterator)
         yield from iterator
 
+    @normalize_exceptions
     def create_automation(
         self,
         obj: "NewAutomation",
@@ -2044,6 +2046,7 @@ class Api:
 
         return Automation.model_validate(result.trigger)
 
+    @normalize_exceptions
     def update_automation(
         self,
         obj: "Automation",
@@ -2151,7 +2154,7 @@ class Api:
 
             # Not a (known) recoverable HTTP error
             wandb.termerror(f"Got response status {status!r}: {e.response.text!r}")
-            raise e
+            raise
 
         try:
             result = UpdateAutomation.model_validate(data).result
@@ -2165,6 +2168,7 @@ class Api:
 
         return Automation.model_validate(result.trigger)
 
+    @normalize_exceptions
     def delete_automation(self, obj: Union["Automation", str]) -> Literal[True]:
         """Delete an automation.