wandb 0.20.1rc20250604__py3-none-musllinux_1_2_aarch64.whl → 0.21.0__py3-none-musllinux_1_2_aarch64.whl

wandb/apis/public/reports.py

@@ -2,6 +2,7 @@
 
 import ast
 import json
+import re
 import urllib
 
 from wandb_gql import gql
@@ -32,9 +33,11 @@ class Reports(SizedPaginator["BetaReport"]):
                             user {
                                 username
                                 photoUrl
+                                email
                             }
                             spec
                             updatedAt
+                            createdAt
                         }
                         cursor
                     }
@@ -60,7 +63,7 @@ class Reports(SizedPaginator["BetaReport"]):
         super().__init__(client, variables, per_page)
 
     @property
-    def length(self):
+    def _length(self):
         # TODO: Add the count the backend
         if self.last_response:
             return len(self.objects)
@@ -111,11 +114,15 @@ class BetaReport(Attrs):
     WARNING: this API will likely change in a future release
 
     Attributes:
+        id (string): unique identifier of the report
         name (string): report name
-        description (string): report description;
-        user (User): the user that created the report
-        spec (dict): the spec off the report;
+        display_name (string): display name of the report
+        description (string): report description
+        user (User): the user that created the report (contains username and email)
+        spec (dict): the spec of the report
+        url (string): the url of the report
         updated_at (string): timestamp of last update
+        created_at (string): timestamp when the report was created
     """
 
     def __init__(self, client, attrs, entity=None, project=None):
@@ -124,7 +131,16 @@ class BetaReport(Attrs):
         self.entity = entity
         self.query_generator = public.QueryGenerator()
         super().__init__(dict(attrs))
-        self._attrs["spec"] = json.loads(self._attrs["spec"])
+
+        if "spec" in self._attrs:
+            if isinstance(self._attrs["spec"], str):
+                self._attrs["spec"] = json.loads(self._attrs["spec"])
+        else:
+            self._attrs["spec"] = {}
+
+    @property
+    def spec(self):
+        return self._attrs["spec"]
 
     @property
     def sections(self):
@@ -153,12 +169,44 @@ class BetaReport(Attrs):
             per_page=per_page,
         )
 
+    @property
+    def id(self):
+        return self._attrs.get("id")
+
+    @property
+    def name(self):
+        return self._attrs.get("name")
+
+    @property
+    def display_name(self):
+        return self._attrs.get("displayName")
+
+    @property
+    def description(self):
+        return self._attrs.get("description")
+
+    @property
+    def user(self):
+        return self._attrs.get("user")
+
     @property
     def updated_at(self):
-        return self._attrs["updatedAt"]
+        return self._attrs.get("updatedAt")
+
+    @property
+    def created_at(self):
+        return self._attrs.get("createdAt")
 
     @property
     def url(self):
+        if (
+            not self.client
+            or not self.entity
+            or not self.project
+            or not self.display_name
+            or not self.id
+        ):
+            return None
         return self.client.app_url + "/".join(
             [
                 self.entity,
@@ -166,7 +214,12 @@ class BetaReport(Attrs):
                 "reports",
                 "--".join(
                     [
-                        urllib.parse.quote(self.display_name.replace(" ", "-")),
+                        # made this more closely match the url creation in the frontend (https://github.com/wandb/core/blob/76943979c8e967f7a62dae8bef0a001a2672584c/frontends/app/src/util/report/urls.ts#L19)
+                        urllib.parse.quote(
+                            re.sub(
+                                r"-+", "-", re.sub(r"\W", "-", self.display_name)
+                            ).strip("-")
+                        ),
                         self.id.replace("=", ""),
                     ]
                 ),
@@ -175,7 +228,10 @@ class BetaReport(Attrs):
 
     def to_html(self, height=1024, hidden=False):
         """Generate HTML containing an iframe displaying this report."""
-        url = self.url + "?jupyter=true"
+        url = self.url
+        if url is None:
+            return "<div>Report URL not available</div>"
+        url = url + "?jupyter=true"
         style = f"border:none;width:100%;height:{height}px;"
         prefix = ""
         if hidden:

wandb/apis/public/runs.py

@@ -111,7 +111,7 @@ class Runs(SizedPaginator["Run"]):
                     runs(filters: $filters, after: $cursor, first: $perPage, order: $order) {{
                         edges {{
                             node {{
-                                {"" if _server_provides_internal_id_for_project(client) else "internalId"}
+                                {"projectId" if _server_provides_internal_id_for_project(client) else ""}
                                 ...RunFragment
                             }}
                             cursor
@@ -143,11 +143,10 @@ class Runs(SizedPaginator["Run"]):
         super().__init__(client, variables, per_page)
 
     @property
-    def length(self):
-        if self.last_response:
-            return self.last_response["project"]["runCount"]
-        else:
-            return None
+    def _length(self):
+        if not self.last_response:
+            self._load_page()
+        return self.last_response["project"]["runCount"]
 
     @property
     def more(self):
@@ -454,25 +453,19 @@ class Run(Attrs):
         )
 
     def load(self, force=False):
-        query = gql(
-            """
-        query Run($project: String!, $entity: String!, $name: String!) {{
-            project(name: $project, entityName: $entity) {{
-                run(name: $name) {{
-                    {}
-                    ...RunFragment
+        if force or not self._attrs:
+            query = gql(f"""#graphql
+            query Run($project: String!, $entity: String!, $name: String!) {{
+                project(name: $project, entityName: $entity) {{
+                    run(name: $name) {{
+                        {"projectId" if _server_provides_internal_id_for_project(self.client) else ""}
+                        ...RunFragment
+                    }}
                 }}
             }}
-        }}
-        {}
-        """.format(
-                "projectId"
-                if _server_provides_internal_id_for_project(self.client)
-                else "",
-                RUN_FRAGMENT,
-            )
-        )
-        if force or not self._attrs:
+            {RUN_FRAGMENT}
+            """)
+
             response = self._exec(query)
             if (
                 response is None

wandb/automations/__init__.py

@@ -45,21 +45,21 @@ else:
 
 __all__ = [
     # Scopes
-    "ScopeType",
-    "ArtifactCollectionScope",
-    "ProjectScope",
+    "ScopeType",  # doc:exclude
+    "ArtifactCollectionScope",  # doc:exclude
+    "ProjectScope",  # doc:exclude
     # Events
-    "EventType",
+    "EventType",  # doc:exclude
     "OnAddArtifactAlias",
     "OnCreateArtifact",
     "OnLinkArtifact",
     "OnRunMetric",
-    "ArtifactEvent",
-    "RunEvent",
+    "ArtifactEvent",  # doc:exclude
+    "RunEvent",  # doc:exclude
     "MetricThresholdFilter",
     "MetricChangeFilter",
     # Actions
-    "ActionType",
+    "ActionType",  # doc:exclude
     "SendNotification",
     "SendWebhook",
     "DoNothing",
@@ -67,7 +67,7 @@ __all__ = [
     "Automation",
     "NewAutomation",
     # Integrations
-    "Integration",
-    "SlackIntegration",
-    "WebhookIntegration",
+    "Integration",  # doc:exclude
+    "SlackIntegration",  # doc:exclude
+    "WebhookIntegration",  # doc:exclude
 ]

wandb/automations/_filters/run_metrics.py

@@ -1,5 +1,3 @@
-# ruff: noqa: UP007
-
 from __future__ import annotations
 
 from abc import ABC, abstractmethod

wandb/automations/_utils.py

@@ -1,5 +1,3 @@
-# ruff: noqa: UP007  # Avoid using `X | Y` for union fields, as this can cause issues with pydantic < 2.6
-
 from __future__ import annotations
 
 from typing import Any, Collection, Final, Optional, Protocol, TypedDict

wandb/automations/actions.py

@@ -1,7 +1,5 @@
 """Actions that are triggered by W&B Automations."""
 
-# ruff: noqa: UP007  # Avoid using `X | Y` for union fields, as this can cause issues with pydantic < 2.6
-
 from __future__ import annotations
 
 from typing import Any, Literal, Optional, Union

wandb/automations/automations.py

@@ -1,5 +1,3 @@
-# ruff: noqa: UP007  # Avoid using `X | Y` for union fields, as this can cause issues with pydantic < 2.6
-
 from __future__ import annotations
 
 from datetime import datetime

wandb/automations/events.py

@@ -1,7 +1,5 @@
 """Events that trigger W&B Automations."""
 
-# ruff: noqa: UP007  # Avoid using `X | Y` for union fields, as this can cause issues with pydantic < 2.6
-
 from __future__ import annotations
 
 from typing import TYPE_CHECKING, Any, Literal, Optional, Union

wandb/cli/beta.py

@@ -11,7 +11,7 @@ import sys
 import click
 
 import wandb
-from wandb.errors import UsageError, WandbCoreNotAvailableError
+from wandb.errors import WandbCoreNotAvailableError
 from wandb.sdk.wandb_sync import _sync
 from wandb.util import get_core_path
 
@@ -24,12 +24,6 @@ def beta():
 
     wandb._sentry.configure_scope(process_context="wandb_beta")
 
-    if wandb.env.is_require_legacy_service():
-        raise UsageError(
-            "wandb beta commands can only be used with wandb-core. "
-            f"Please make sure that `{wandb.env._REQUIRE_LEGACY_SERVICE}` is not set."
-        )
-
     try:
         get_core_path()
     except WandbCoreNotAvailableError as e:

wandb/cli/cli.py

@@ -252,36 +252,6 @@ def login(key, host, cloud, relogin, anonymously, verify, no_offline=False):
     )
 
 
-@cli.command(
-    context_settings=CONTEXT, help="Run a wandb service", name="service", hidden=True
-)
-@click.option(
-    "--sock-port", default=None, type=int, help="The host port to bind socket service."
-)
-@click.option("--port-filename", default=None, help="Save allocated port to file.")
-@click.option("--address", default=None, help="The address to bind service.")
-@click.option("--pid", default=None, type=int, help="The parent process id to monitor.")
-@click.option("--debug", is_flag=True, help="log debug info")
-@display_error
-def service(
-    sock_port=None,
-    port_filename=None,
-    address=None,
-    pid=None,
-    debug=False,
-):
-    from wandb.sdk.service.server import WandbServer
-
-    server = WandbServer(
-        sock_port=sock_port,
-        port_fname=port_filename,
-        address=address,
-        pid=pid,
-        debug=debug,
-    )
-    server.serve()
-
-
 @cli.command(
     context_settings=CONTEXT, help="Configure a directory with Weights & Biases"
 )

wandb/env.py

@@ -86,7 +86,6 @@ _EXECUTABLE = "WANDB_X_EXECUTABLE"
 LAUNCH_QUEUE_NAME = "WANDB_LAUNCH_QUEUE_NAME"
 LAUNCH_QUEUE_ENTITY = "WANDB_LAUNCH_QUEUE_ENTITY"
 LAUNCH_TRACE_ID = "WANDB_LAUNCH_TRACE_ID"
-_REQUIRE_LEGACY_SERVICE = "WANDB_X_REQUIRE_LEGACY_SERVICE"
 ENABLE_DCGM_PROFILING = "WANDB_ENABLE_DCGM_PROFILING"
 
 # For testing, to be removed in future version
@@ -150,11 +149,6 @@ def _env_as_bool(
         return False
 
 
-def is_require_legacy_service(env: MutableMapping | None = None) -> bool:
-    """Return whether wandb.require("legacy-service") was used."""
-    return _env_as_bool(_REQUIRE_LEGACY_SERVICE, default="False", env=env)
-
-
 def is_debug(default: str | None = None, env: MutableMapping | None = None) -> bool:
     return _env_as_bool(DEBUG, default=default, env=env)
 

wandb/integration/catboost/catboost.py

@@ -22,7 +22,9 @@ class WandbCallback:
 
     Example:
         ```
-        train_pool = Pool(train[features], label=train["label"], cat_features=cat_features)
+        train_pool = Pool(
+            train[features], label=train["label"], cat_features=cat_features
+        )
         test_pool = Pool(test[features], label=test["label"], cat_features=cat_features)
 
         model = CatBoostRegressor(
@@ -130,7 +132,9 @@ def log_summary(
 
     Example:
         ```python
-        train_pool = Pool(train[features], label=train["label"], cat_features=cat_features)
+        train_pool = Pool(
+            train[features], label=train["label"], cat_features=cat_features
+        )
         test_pool = Pool(test[features], label=test["label"], cat_features=cat_features)
 
         model = CatBoostRegressor(

wandb/integration/kfp/kfp_patch.py

@@ -270,7 +270,9 @@ def create_component_from_func(
         Example of a component function declaring file input and output::
 
             def catboost_train_classifier(
-                training_data_path: InputPath("CSV"),  # Path to input data file of type "CSV"
+                training_data_path: InputPath(
+                    "CSV"
+                ),  # Path to input data file of type "CSV"
                 trained_model_path: OutputPath(
                     "CatBoostModel"
                 ),  # Path to output data file of type "CatBoostModel"

wandb/integration/sb3/sb3.py

@@ -106,9 +106,9 @@ class WandbCallback(BaseCallback):
             os.makedirs(self.model_save_path, exist_ok=True)
             self.path = os.path.join(self.model_save_path, "model.zip")
         else:
-            assert (
-                self.model_save_freq == 0
-            ), "to use the `model_save_freq` you have to set the `model_save_path` parameter"
+            assert self.model_save_freq == 0, (
+                "to use the `model_save_freq` you have to set the `model_save_path` parameter"
+            )
 
     def _init_callback(self) -> None:
         d = {}

wandb/integration/ultralytics/callback.py

@@ -102,7 +102,9 @@ class WandBUltralyticsCallback:
         model = YOLO("yolov8n.pt")
 
         # add wandb callback
-        add_wandb_callback(model, max_validation_batches=2, enable_model_checkpointing=True)
+        add_wandb_callback(
+            model, max_validation_batches=2, enable_model_checkpointing=True
+        )
 
         # train
         model.train(data="coco128.yaml", epochs=5, imgsz=640)
@@ -454,7 +456,9 @@ def add_wandb_callback(
         model = YOLO("yolov8n.pt")
 
         # add wandb callback
-        add_wandb_callback(model, max_validation_batches=2, enable_model_checkpointing=True)
+        add_wandb_callback(
+            model, max_validation_batches=2, enable_model_checkpointing=True
+        )
 
         # train
         model.train(data="coco128.yaml", epochs=5, imgsz=640)

wandb/plot/__init__.py

@@ -14,6 +14,8 @@ __all__ = [
     "pr_curve",
     "confusion_matrix",
     "line_series",
+    "plot_table",
+    "visualize",  # doc:exclude
 ]
 
 from wandb.plot.bar import bar

wandb/plot/bar.py

@@ -19,11 +19,11 @@ def bar(
     """Constructs a bar chart from a wandb.Table of data.
 
     Args:
-        table (wandb.Table): A table containing the data for the bar chart.
-        label (str): The name of the column to use for the labels of each bar.
-        value (str): The name of the column to use for the values of each bar.
-        title (str): The title of the bar chart.
-        split_table (bool): Whether the table should be split into a separate section
+        table: A table containing the data for the bar chart.
+        label: The name of the column to use for the labels of each bar.
+        value: The name of the column to use for the values of each bar.
+        title: The title of the bar chart.
+        split_table: Whether the table should be split into a separate section
             in the W&B UI. If `True`, the table will be displayed in a section named
             "Custom Chart Tables". Default is `False`.
 
@@ -32,34 +32,35 @@ def bar(
             chart, pass it to `wandb.log()`.
 
     Example:
-        ```
-        import random
-        import wandb
 
-        # Generate random data for the table
-        data = [
-            ["car", random.uniform(0, 1)],
-            ["bus", random.uniform(0, 1)],
-            ["road", random.uniform(0, 1)],
-            ["person", random.uniform(0, 1)],
-        ]
+    ```python
+    import random
+    import wandb
+
+    # Generate random data for the table
+    data = [
+        ["car", random.uniform(0, 1)],
+        ["bus", random.uniform(0, 1)],
+        ["road", random.uniform(0, 1)],
+        ["person", random.uniform(0, 1)],
+    ]
 
-        # Create a table with the data
-        table = wandb.Table(data=data, columns=["class", "accuracy"])
+    # Create a table with the data
+    table = wandb.Table(data=data, columns=["class", "accuracy"])
 
-        # Initialize a W&B run and log the bar plot
-        with wandb.init(project="bar_chart") as run:
-            # Create a bar plot from the table
-            bar_plot = wandb.plot.bar(
-                table=table,
-                label="class",
-                value="accuracy",
-                title="Object Classification Accuracy",
-            )
+    # Initialize a W&B run and log the bar plot
+    with wandb.init(project="bar_chart") as run:
+        # Create a bar plot from the table
+        bar_plot = wandb.plot.bar(
+            table=table,
+            label="class",
+            value="accuracy",
+            title="Object Classification Accuracy",
+        )
 
-            # Log the bar chart to W&B
-            run.log({"bar_plot": bar_plot})
-        ```
+        # Log the bar chart to W&B
+        run.log({"bar_plot": bar_plot})
+    ```
     """
     return plot_table(
         data_table=table,