wandb 0.20.2rc20250616__py3-none-macosx_11_0_arm64.whl → 0.21.1__py3-none-macosx_11_0_arm64.whl

wandb/agents/pyagent.py

@@ -9,6 +9,7 @@ import logging
 import os
 import queue
 import socket
+import sys
 import threading
 import time
 import traceback
@@ -222,17 +223,10 @@ class Agent:
                         self._run_status[run_id] = RunStatus.DONE
                     elif self._run_status[run_id] == RunStatus.ERRORED:
                         exc = self._exceptions[run_id]
-                        exc_type, exc_value, exc_traceback = (
-                            exc.__class__,
-                            exc,
-                            exc.__traceback__,
-                        )
-                        exc_traceback_formatted = traceback.format_exception(
-                            exc_type, exc_value, exc_traceback
-                        )
-                        exc_repr = "".join(exc_traceback_formatted)
-                        logger.error(f"Run {run_id} errored:\n{exc_repr}")
-                        wandb.termerror(f"Run {run_id} errored:\n{exc_repr}")
+                        # Extract to reduce a decision point to avoid ruff c901
+                        log_str, term_str = _get_exception_logger_and_term_strs(exc)
+                        logger.error(f"Run {run_id} errored:\n{log_str}")
+                        wandb.termerror(f"Run {run_id} errored:{term_str}")
                         if os.getenv(wandb.env.AGENT_DISABLE_FLAPPING) == "true":
                             self._exit_flag = True
                             return
@@ -299,7 +293,18 @@ class Agent:
             for k, v in job.config.items():
                 wandb.termlog("\t{}: {}".format(k, v["value"]))
 
-            self._function()
+            try:
+                self._function()
+            except KeyboardInterrupt:
+                raise
+            except Exception as e:
+                # Log the run's exceptions directly to stderr to match CLI case, and wrap so we
+                # can identify it as coming from the job later later. This will get automatically
+                # logged by console_capture.py. Exception handler below will also handle exceptions
+                # in setup code.
+                exc_repr = _format_exception_traceback(e)
+                print(exc_repr, file=sys.stderr)  # noqa: T201
+                raise _JobError(f"Run threw exception: {str(e)}") from e
             wandb.finish()
         except KeyboardInterrupt:
             raise
@@ -350,6 +355,30 @@ def pyagent(sweep_id, function, entity=None, project=None, count=None):
     agent.run()
 
 
+def _format_exception_traceback(exc):
+    return "".join(traceback.format_exception(type(exc), exc, exc.__traceback__))
+
+
+class _JobError(Exception):
+    """Exception raised when a job fails during execution."""
+
+    pass
+
+
+def _get_exception_logger_and_term_strs(exc):
+    if isinstance(exc, _JobError) and exc.__cause__:
+        # If it's a JobException, get the original exception for display
+        job_exc = exc.__cause__
+        log_str = _format_exception_traceback(job_exc)
+        # Don't long full stacktrace to terminal again because we already
+        # printed it to stderr.
+        term_str = " " + str(job_exc)
+    else:
+        log_str = _format_exception_traceback(exc)
+        term_str = "\n" + log_str
+    return log_str, term_str
+
+
 _INSTANCES = 0
 
 

wandb/analytics/sentry.py

@@ -91,13 +91,18 @@ class Sentry:
         self.scope.set_client(client)
 
     @_safe_noop
-    def message(self, message: str, repeat: bool = True) -> str | None:
+    def message(
+        self,
+        message: str,
+        repeat: bool = True,
+        level: str = "info",
+    ) -> str | None:
         """Send a message to Sentry."""
         if not repeat and message in self._sent_messages:
             return None
         self._sent_messages.add(message)
         with sentry_sdk.scope.use_isolation_scope(self.scope):  # type: ignore
-            return sentry_sdk.capture_message(message)  # type: ignore
+            return sentry_sdk.capture_message(message, level=level)  # type: ignore
 
     @_safe_noop
     def exception(

wandb/apis/importers/mlflow.py

@@ -215,7 +215,7 @@ class MlflowImporter:
         settings_override = {
             "api_key": self.dst_api_key,
             "base_url": self.dst_base_url,
-            "resume": "true",
+            "resume": "allow",
             "resumed": True,
         }
 

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

@@ -20,7 +20,7 @@ from wandb.apis.public.jobs import (
     RunQueuePrioritizationMode,
     RunQueueResourceType,
 )
-from wandb.apis.public.projects import PROJECT_FRAGMENT, Project, Projects
+from wandb.apis.public.projects import PROJECT_FRAGMENT, Project, Projects, Sweeps
 from wandb.apis.public.query_generator import QueryGenerator
 from wandb.apis.public.registries.registry import Registry
 from wandb.apis.public.reports import (