wandb 0.18.4__py3-none-any.whl → 0.18.6__py3-none-any.whl

wandb/__init__.py

@@ -10,7 +10,7 @@ For reference documentation, see https://docs.wandb.com/ref/python.
 """
 from __future__ import annotations
 
-__version__ = "0.18.4"
+__version__ = "0.18.6"
 
 
 from wandb.errors import Error
@@ -75,7 +75,7 @@ from wandb.data_types import JoinedTable
 
 from wandb.wandb_agent import agent
 
-from wandb.plot.viz import visualize
+from wandb.plot import visualize
 from wandb.integration.sagemaker import sagemaker_auth
 from wandb.sdk.internal import profiler
 

wandb/__init__.pyi

@@ -53,6 +53,7 @@ __all__ = (
     "teardown",
     "watch",
     "unwatch",
+    "plot",
 )
 
 import os
@@ -68,6 +69,7 @@ from typing import (
     Union,
 )
 
+import wandb.plot as plot
 from wandb.analytics import Sentry
 from wandb.apis import InternalApi, PublicApi
 from wandb.data_types import (
@@ -96,9 +98,9 @@ from wandb.wandb_controller import _WandbController
 if TYPE_CHECKING:
     import torch  # type: ignore [import-not-found]
 
-    from wandb.plot.viz import CustomChart
+    from wandb.plot import CustomChart
 
-__version__: str = "0.18.4"
+__version__: str = "0.18.6"
 
 run: Run | None
 config: wandb_config.Config
@@ -249,7 +251,7 @@ def init(
     For more on using `wandb.init()`, including detailed examples, check out our
     [guide and FAQs](https://docs.wandb.ai/guides/track/launch).
 
-    Arguments:
+    Args:
         project: (str, optional) The name of the project where you're sending
             the new run. If the project is not specified, we will try to infer
             the project name from git root or the current program file. If we
@@ -372,11 +374,11 @@ def init(
             for saving hyperparameters to compare across runs. The ID cannot
             contain the following special characters: `/\#?%:`.
             See [our guide to resuming runs](https://docs.wandb.com/guides/runs/resuming).
-        fork_from: (str, optional) A string with the format {run_id}?_step={step} describing
+        fork_from: (str, optional) A string with the format `{run_id}?_step={step}` describing
             a moment in a previous run to fork a new run from. Creates a new run that picks up
             logging history from the specified run at the specified moment. The target run must
             be in the current project. Example: `fork_from="my-run-id?_step=1234"`.
-        resume_from: (str, optional) A string with the format {run_id}?_step={step} describing
+        resume_from: (str, optional) A string with the format `{run_id}?_step={step}` describing
             a moment in a previous run to resume a run from. This allows users to truncate
             the history logged to a run at an intermediate step and resume logging from that step.
             It uses run forking under the hood. The target run must be in the
@@ -429,7 +431,7 @@ def finish(exit_code: int | None = None, quiet: bool | None = None) -> None:
     This is used when creating multiple runs in the same process.
     We automatically call this method when your script exits.
 
-    Arguments:
+    Args:
         exit_code: Set to something other than 0 to mark a run as failed
         quiet: Set to true to minimize log output
     """
@@ -450,7 +452,7 @@ def login(
     verifying them with the W&B server. To verify credentials, pass
     `verify=True`.
 
-    Arguments:
+    Args:
         anonymous: (string, optional) Can be "must", "allow", or "never".
             If set to "must", always log a user in anonymously. If set to
             "allow", only create an anonymous user if the user
@@ -503,7 +505,7 @@ def log(
     [guides to logging](https://docs.wandb.ai/guides/track/log) for examples,
     from 3D molecular structures and segmentation masks to PR curves and histograms.
     You can use `wandb.Table` to log structured data. See our
-    [guide to logging tables](https://docs.wandb.ai/guides/data-vis/log-tables)
+    [guide to logging tables](https://docs.wandb.ai/guides/tables/tables-walkthrough)
     for details.
 
     The W&B UI organizes metrics with a forward slash (`/`) in their name
@@ -566,7 +568,7 @@ def log(
     run.log({"accuracy": 0.9}, step=current_step)
     ```
 
-    Arguments:
+    Args:
         data: A `dict` with `str` keys and values that are serializable
             Python objects including: `int`, `float` and `string`;
             any of the `wandb.data_types`; lists, tuples and NumPy arrays
@@ -745,7 +747,7 @@ def save(
     Note: when given an absolute path or glob and no `base_path`, one
     directory level is preserved as in the example above.
 
-    Arguments:
+    Args:
         glob_str: A relative or absolute path or Unix glob.
         base_path: A path to use to infer a directory structure; see examples.
         policy: One of `live`, `now`, or `end`.
@@ -830,7 +832,7 @@ def agent(
     is a part of, what function to execute, and (optionally) how
     many agents to run.
 
-    Arguments:
+    Args:
         sweep_id: The unique identifier for a sweep. A sweep ID
             is generated by W&B CLI or Python SDK.
         function: A function to call instead of the "program"
@@ -858,7 +860,7 @@ def define_metric(
 ) -> wandb_metric.Metric:
     """Customize metrics logged with `wandb.log()`.
 
-    Arguments:
+    Args:
         name: The name of the metric to customize.
         step_metric: The name of another metric to serve as the X-axis
             for this metric in automatically generated charts.
@@ -893,7 +895,7 @@ def log_artifact(
 ) -> Artifact:
     """Declare an artifact as an output of a run.
 
-    Arguments:
+    Args:
         artifact_or_path: (str or Artifact) A path to the contents of this artifact,
             can be in the following forms:
                 - `/local/directory`
@@ -927,7 +929,7 @@ def use_artifact(
 
     Call `download` or `file` on the returned object to get the contents locally.
 
-    Arguments:
+    Args:
         artifact_or_name: (str or Artifact) An artifact name.
             May be prefixed with project/ or entity/project/.
             If no entity is specified in the name, the Run or API setting's entity is used.
@@ -952,7 +954,7 @@ def log_model(
 ) -> None:
     """Logs a model artifact containing the contents inside the 'path' to a run and marks it as an output to this run.
 
-    Arguments:
+    Args:
         path: (str) A path to the contents of this model,
             can be in the following forms:
                 - `/local/directory`
@@ -994,7 +996,7 @@ def log_model(
 def use_model(name: str) -> FilePathStr:
     """Download the files logged in a model artifact 'name'.
 
-    Arguments:
+    Args:
         name: (str) A model artifact name. 'name' must match the name of an existing logged
             model artifact.
             May be prefixed with entity/project/. Valid names
@@ -1051,7 +1053,7 @@ def link_model(
         - Link version of model artifact 'name' to registered model, 'registered_model_name'.
         - Attach aliases from 'aliases' list to the newly linked model artifact version.
 
-    Arguments:
+    Args:
         path: (str) A path to the contents of this model,
             can be in the following forms:
                 - `/local/directory`
@@ -1108,11 +1110,11 @@ def plot_table(
     data_table: Table,
     fields: dict[str, Any],
     string_fields: dict[str, Any] | None = None,
-    split_table: bool | None = False,
+    split_table: bool = False,
 ) -> CustomChart:
     """Create a custom plot on a table.
 
-    Arguments:
+    Args:
         vega_spec_name: the name of the spec for the plot
         data_table: a wandb.Table object containing the data to
             be used on the visualization

wandb/agents/pyagent.py

@@ -337,7 +337,7 @@ class Agent:
 def pyagent(sweep_id, function, entity=None, project=None, count=None):
     """Generic agent entrypoint, used for CLI or jupyter.
 
-    Arguments:
+    Args:
         sweep_id (dict): Sweep ID generated by CLI or sweep API
         function (func, optional): A function to call instead of the "program"
         entity (str, optional): W&B Entity

wandb/apis/importers/wandb.py

@@ -573,7 +573,7 @@ class WandbImporter:
         project = coalesce(project, src_art.project)
         name = src_art.name
 
-        return self.dst_api.artifact(f"{entity}/{project}/{name}")
+        return self.dst_api._artifact(f"{entity}/{project}/{name}")
 
     def _get_run_problems(
         self, src_run: Run, dst_run: Run, force_retry: bool = False

wandb/apis/normalize.py

@@ -3,34 +3,18 @@
 import ast
 import sys
 from functools import wraps
-from typing import Callable, List, TypeVar
+from typing import Callable, TypeVar
 
 import requests
 from wandb_gql.client import RetryError
 
 from wandb import env
 from wandb.errors import CommError, Error
+from wandb.util import parse_backend_error_messages
 
 _F = TypeVar("_F", bound=Callable)
 
 
-def parse_backend_error_messages(response: requests.Response) -> List[str]:
-    errors = []
-    try:
-        data = response.json()
-    except ValueError:
-        return errors
-
-    if "errors" in data and isinstance(data["errors"], list):
-        for error in data["errors"]:
-            # Our tests and potentially some api endpoints return a string error?
-            if isinstance(error, str):
-                error = {"message": error}
-            if "message" in error:
-                errors.append(error["message"])
-    return errors
-
-
 def normalize_exceptions(func: _F) -> _F:
     """Function decorator for catching common errors and re-raising as wandb.Error."""