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

package_readme.md

@@ -68,6 +68,14 @@ Use your favorite framework with W&B. W&B integrations make it fast and easy to
 
  
 
+# Python Version Support
+
+We are committed to supporting our minimum required Python version for *at least* six months after its official end-of-life (EOL) date, as defined by the Python Software Foundation. You can find a list of Python EOL dates [here](https://devguide.python.org/versions/).
+
+When we discontinue support for a Python version, we will increment the library’s minor version number to reflect this change.
+
+ 
+
 # Contribution guidelines
 Weights & Biases ❤️ open source, and we welcome contributions from the community! See the [Contribution guide](https://github.com/wandb/wandb/blob/main/CONTRIBUTING.md) for more information on the development workflow and the internals of the wandb library. For wandb bugs and feature requests, visit [GitHub Issues](https://github.com/wandb/wandb/issues) or contact support@wandb.com.
 

wandb/__init__.py

@@ -10,7 +10,7 @@ For reference documentation, see https://docs.wandb.com/ref/python.
 """
 from __future__ import annotations
 
-__version__ = "0.18.6"
+__version__ = "0.19.0"
 
 
 from wandb.errors import Error
@@ -75,7 +75,7 @@ from wandb.data_types import JoinedTable
 
 from wandb.wandb_agent import agent
 
-from wandb.plot import visualize
+from wandb.plot import visualize, plot_table
 from wandb.integration.sagemaker import sagemaker_auth
 from wandb.sdk.internal import profiler
 
@@ -142,9 +142,6 @@ mark_preempting = _preinit.PreInitCallable(
     "wandb.mark_preempting", wandb_sdk.wandb_run.Run.mark_preempting  # type: ignore
 )
 
-plot_table = _preinit.PreInitCallable(
-    "wandb.plot_table", wandb_sdk.wandb_run.Run.plot_table
-)
 alert = _preinit.PreInitCallable("wandb.alert", wandb_sdk.wandb_run.Run.alert)  # type: ignore
 
 # record of patched libraries
@@ -208,8 +205,8 @@ if "dev" in __version__:
     )
 
     # Enable new features in dev versions.
-    os.environ["WANDB__SHOW_OPERATION_STATS"] = os.environ.get(
-        "WANDB__SHOW_OPERATION_STATS",
+    os.environ["WANDB_X_SHOW_OPERATION_STATS"] = os.environ.get(
+        "WANDB_X_SHOW_OPERATION_STATS",
         "true",
     )
 
@@ -251,4 +248,5 @@ __all__ = (
     "define_metric",
     "watch",
     "unwatch",
+    "plot_table",
 )

wandb/__init__.pyi

@@ -54,6 +54,7 @@ __all__ = (
     "watch",
     "unwatch",
     "plot",
+    "plot_table",
 )
 
 import os
@@ -71,7 +72,8 @@ from typing import (
 
 import wandb.plot as plot
 from wandb.analytics import Sentry
-from wandb.apis import InternalApi, PublicApi
+from wandb.apis import InternalApi
+from wandb.apis import PublicApi as Api
 from wandb.data_types import (
     Audio,
     Graph,
@@ -98,14 +100,14 @@ from wandb.wandb_controller import _WandbController
 if TYPE_CHECKING:
     import torch  # type: ignore [import-not-found]
 
+    import wandb
     from wandb.plot import CustomChart
 
-__version__: str = "0.18.6"
+__version__: str = "0.19.0"
 
 run: Run | None
 config: wandb_config.Config
 summary: wandb_summary.Summary
-Api: type[PublicApi]
 
 # private attributes
 _sentry: Sentry
@@ -113,7 +115,7 @@ api: InternalApi
 patched: Dict[str, List[Callable]]
 
 def setup(
-    settings: Optional[Settings] = None,
+    settings: Settings | None = None,
 ) -> Optional[_WandbSetup]:
     """Prepares W&B for use in the current process and its children.
 
@@ -129,8 +131,8 @@ def setup(
     See also `wandb.teardown()`.
 
     Args:
-        settings (Optional[Union[Dict[str, Any], wandb.Settings]]): Configuration settings
-            to apply globally. These can be overridden by subsequent `wandb.init()` calls.
+        settings: Configuration settings to apply globally. These can be
+            overridden by subsequent `wandb.init()` calls.
 
     Example:
         ```python
@@ -172,7 +174,7 @@ def setup(
     """
     ...
 
-def teardown(exit_code: Optional[int] = None) -> None:
+def teardown(exit_code: int | None = None) -> None:
     """Waits for wandb to finish and frees resources.
 
     Completes any runs that were not explicitly finished
@@ -192,11 +194,10 @@ def init(
     project: str | None = None,
     entity: str | None = None,
     reinit: bool | None = None,
-    tags: Sequence | None = None,
+    tags: Sequence[str] | None = None,
     group: str | None = None,
     name: str | None = None,
     notes: str | None = None,
-    magic: dict | str | bool | None = None,
     config_exclude_keys: list[str] | None = None,
     config_include_keys: list[str] | None = None,
     anonymous: str | None = None,
@@ -335,10 +336,6 @@ def init(
             for more.
         reinit: (bool, optional) Allow multiple `wandb.init()` calls in the same
             process. (default: `False`)
-        magic: (bool, dict, or str, optional) The bool controls whether we try to
-            auto-instrument your script, capturing basic details of your run
-            without you having to add more wandb code. (default: `False`)
-            You can also pass a dict, json string, or yaml filename.
         config_exclude_keys: (list, optional) string keys to exclude from
             `wandb.config`.
         config_include_keys: (list, optional) string keys to include in
@@ -425,15 +422,25 @@ def init(
     """
     ...
 
-def finish(exit_code: int | None = None, quiet: bool | None = None) -> None:
-    """Mark a run as finished, and finish uploading all data.
+def finish(
+    exit_code: int | None = None,
+    quiet: bool | None = None,
+) -> None:
+    """Finish a run and upload any remaining data.
+
+    Marks the completion of a W&B run and ensures all data is synced to the server.
+    The run's final state is determined by its exit conditions and sync status.
 
-    This is used when creating multiple runs in the same process.
-    We automatically call this method when your script exits.
+    Run States:
+    - Running: Active run that is logging data and/or sending heartbeats.
+    - Crashed: Run that stopped sending heartbeats unexpectedly.
+    - Finished: Run completed successfully (`exit_code=0`) with all data synced.
+    - Failed: Run completed with errors (`exit_code!=0`).
 
     Args:
-        exit_code: Set to something other than 0 to mark a run as failed
-        quiet: Set to true to minimize log output
+        exit_code: Integer indicating the run's exit status. Use 0 for success,
+            any other value marks the run as failed.
+        quiet: Deprecated. Configure logging verbosity using `wandb.Settings(quiet=...)`.
     """
     ...
 
@@ -1107,23 +1114,37 @@ def link_model(
 
 def plot_table(
     vega_spec_name: str,
-    data_table: Table,
+    data_table: wandb.Table,
     fields: dict[str, Any],
     string_fields: dict[str, Any] | None = None,
     split_table: bool = False,
 ) -> CustomChart:
-    """Create a custom plot on a table.
+    """Creates a custom charts using a Vega-Lite specification and a `wandb.Table`.
+
+    This function creates a custom chart based on a Vega-Lite specification and
+    a data table represented by a `wandb.Table` object. The specification needs
+    to be predefined and stored in the W&B backend. The function returns a custom
+    chart object that can be logged to W&B using `wandb.log()`.
 
     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
-        fields: a dict mapping from table keys to fields that the custom
-            visualization needs
-        string_fields: a dict that provides values for any string constants
-            the custom visualization needs
-        split_table: a boolean that indicates whether the table should be in
-            a separate section in the UI
+        vega_spec_name (str): The name or identifier of the Vega-Lite spec
+            that defines the visualization structure.
+        data_table (wandb.Table): A `wandb.Table` object containing the data to be
+            visualized.
+        fields (dict[str, Any]): A mapping between the fields in the Vega-Lite spec and the
+            corresponding columns in the data table to be visualized.
+        string_fields (dict[str, Any] | None): A dictionary for providing values for any string constants
+            required by the custom visualization.
+        split_table (bool): 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`.
+
+    Returns:
+        CustomChart: A custom chart object that can be logged to W&B. To log the
+            chart, pass it to `wandb.log()`.
+
+    Raises:
+        wandb.Error: If `data_table` is not a `wandb.Table` object.
     """
     ...
 

wandb/analytics/sentry.py

@@ -9,14 +9,9 @@ import os
 import pathlib
 import sys
 from types import TracebackType
-from typing import TYPE_CHECKING, Any, Callable
+from typing import TYPE_CHECKING, Any, Callable, Literal
 from urllib.parse import quote
 
-if sys.version_info >= (3, 8):
-    from typing import Literal
-else:
-    from typing_extensions import Literal
-
 import sentry_sdk  # type: ignore
 import sentry_sdk.scope  # type: ignore
 import sentry_sdk.utils  # type: ignore
@@ -214,13 +209,12 @@ class Sentry:
             "sweep_url",
             "sweep_id",
             "deployment",
-            "_disable_service",
-            "_require_legacy_service",
+            "x_disable_service",
+            "x_require_legacy_service",
             "launch",
+            "_platform",
         )
 
-        self.scope.set_tag("platform", wandb.util.get_platform_name())
-
         # set context
         if process_context:
             self.scope.set_tag("process_context", process_context)

wandb/apis/importers/internals/internal.py

@@ -311,19 +311,19 @@ def _make_settings(
     _settings_override = coalesce(settings_override, {})
 
     default_settings: Dict[str, Any] = {
-        "files_dir": os.path.join(root_dir, "files"),
+        "x_files_dir": os.path.join(root_dir, "files"),
         "root_dir": root_dir,
         "sync_file": os.path.join(root_dir, "txlog.wandb"),
         "resume": "false",
         "program": None,
         "ignore_globs": [],
         "disable_job_creation": True,
-        "_start_time": 0,
+        "x_start_time": 0,
         "_offline": None,
-        "_sync": True,
-        "_live_policy_rate_limit": 15,  # matches dir_watcher
-        "_live_policy_wait_time": 600,  # matches dir_watcher
-        "_file_stream_timeout_seconds": 60,
+        "x_sync": True,
+        "x_live_policy_rate_limit": 15,  # matches dir_watcher
+        "x_live_policy_wait_time": 600,  # matches dir_watcher
+        "x_file_stream_timeout_seconds": 60,
     }
 
     combined_settings = {**default_settings, **_settings_override}

wandb/apis/importers/internals/protocols.py

@@ -1,14 +1,18 @@
 import logging
-import sys
-from typing import Any, Dict, Iterable, List, Literal, Optional, Tuple
+from typing import (
+    Any,
+    Dict,
+    Iterable,
+    List,
+    Literal,
+    Optional,
+    Protocol,
+    Tuple,
+    runtime_checkable,
+)
 
 from wandb.sdk.artifacts.artifact import Artifact
 
-if sys.version_info >= (3, 8):
-    from typing import Protocol, runtime_checkable
-else:
-    from typing_extensions import Protocol, runtime_checkable
-
 logger = logging.getLogger("import_logger")
 
 PathStr = str

wandb/apis/public/api.py

@@ -1208,7 +1208,11 @@ class Api:
                 entity=settings_entity, organization=organization
             )
         artifact = wandb.Artifact._from_name(
-            entity, project, artifact_name, self.client, enable_tracking
+            entity=entity,
+            project=project,
+            name=artifact_name,
+            client=self.client,
+            enable_tracking=enable_tracking,
         )
         if type is not None and artifact.type != type:
             raise ValueError(

wandb/apis/public/jobs.py

@@ -3,14 +3,8 @@
 import json
 import os
 import shutil
-import sys
 import time
-from typing import TYPE_CHECKING, Any, Dict, List, Mapping, Optional
-
-if sys.version_info >= (3, 8):
-    from typing import Literal
-else:
-    from typing_extensions import Literal
+from typing import TYPE_CHECKING, Any, Dict, List, Literal, Mapping, Optional
 
 from wandb_gql import gql
 

wandb/apis/public/reports.py

@@ -2,7 +2,6 @@
 
 import ast
 import json
-import sys
 import urllib
 
 from wandb_gql import gql
@@ -228,22 +227,12 @@ class PythonMongoishQueryGenerator:
         ast.Not: "$not",
     }
 
-    if sys.version_info >= (3, 8):
-        AST_FIELDS = {
-            ast.Constant: "value",
-            ast.Name: "id",
-            ast.List: "elts",
-            ast.Tuple: "elts",
-        }
-    else:
-        AST_FIELDS = {
-            ast.Str: "s",
-            ast.Num: "n",
-            ast.Name: "id",
-            ast.List: "elts",
-            ast.Tuple: "elts",
-            ast.NameConstant: "value",
-        }
+    AST_FIELDS = {
+        ast.Constant: "value",
+        ast.Name: "id",
+        ast.List: "elts",
+        ast.Tuple: "elts",
+    }
 
     def __init__(self, run_set):
         self.run_set = run_set

wandb/apis/public/runs.py

@@ -2,16 +2,19 @@
 
 import json
 import os
-import sys
 import tempfile
 import time
 import urllib
-from typing import TYPE_CHECKING, Any, Collection, Dict, List, Mapping, Optional
-
-if sys.version_info >= (3, 8):
-    from typing import Literal
-else:
-    from typing_extensions import Literal
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    Collection,
+    Dict,
+    List,
+    Literal,
+    Mapping,
+    Optional,
+)
 
 from wandb_gql import gql
 
@@ -233,7 +236,6 @@ class Runs(Paginator):
             if not histories:
                 return pd.DataFrame()
             combined_df = pd.concat(histories)
-            combined_df.sort_values("run_id", inplace=True)
             combined_df.reset_index(drop=True, inplace=True)
             # sort columns for consistency
             combined_df = combined_df[(sorted(combined_df.columns))]
@@ -259,9 +261,9 @@ class Runs(Paginator):
                 histories.append(df)
             if not histories:
                 return pl.DataFrame()
-            combined_df = pl.concat(histories, how="align")
+            combined_df = pl.concat(histories, how="vertical")
             # sort columns for consistency
-            combined_df = combined_df.select(sorted(combined_df.columns)).sort("run_id")
+            combined_df = combined_df.select(sorted(combined_df.columns))
 
             return combined_df
 

wandb/cli/cli.py

@@ -31,7 +31,6 @@ from wandb import Config, Error, env, util, wandb_agent, wandb_sdk
 from wandb.apis import InternalApi, PublicApi
 from wandb.apis.public import RunQueue
 from wandb.errors.links import url_registry
-from wandb.integration.magic import magic_install
 from wandb.sdk.artifacts._validators import is_artifact_registry_project
 from wandb.sdk.artifacts.artifact_file_cache import get_artifact_file_cache
 from wandb.sdk.launch import utils as launch_utils
@@ -129,7 +128,7 @@ def _get_cling_api(reset=None):
     if _api is None:
         # TODO(jhr): make a settings object that is better for non runs.
         # only override the necessary setting
-        wandb.setup(settings=dict(_cli_only_mode=True))
+        wandb.setup(settings=wandb.Settings(x_cli_only_mode=True))
         _api = InternalApi()
     return _api
 
@@ -238,15 +237,18 @@ def login(key, host, cloud, relogin, anonymously, verify, no_offline=False):
         relogin = True
 
     login_settings = dict(
-        _cli_only_mode=True,
-        _disable_viewer=relogin and not verify,
+        x_cli_only_mode=True,
+        x_disable_viewer=relogin and not verify,
         anonymous=anon_mode,
+        base_url=host,
     )
-    if host is not None:
-        login_settings["base_url"] = host
 
     try:
-        wandb.setup(settings=login_settings)
+        wandb.setup(
+            settings=wandb.Settings(
+                **{k: v for k, v in login_settings.items() if v is not None}
+            )
+        )
     except TypeError as e:
         wandb.termerror(str(e))
         sys.exit(1)
@@ -2656,44 +2658,6 @@ Run `git clone {}` and restore from there or pass the --no-git flag.""".format(r
     return commit, json_config, patch_content, repo, metadata
 
 
-@cli.command(context_settings=CONTEXT, help="Run any script with wandb", hidden=True)
-@click.pass_context
-@click.argument("program")
-@click.argument("args", nargs=-1)
-@display_error
-def magic(ctx, program, args):
-    def magic_run(cmd, globals, locals):
-        try:
-            exec(cmd, globals, locals)
-        finally:
-            pass
-
-    sys.argv[:] = args
-    sys.argv.insert(0, program)
-    sys.path.insert(0, os.path.dirname(program))
-    try:
-        with open(program, "rb") as fp:
-            code = compile(fp.read(), program, "exec")
-    except OSError:
-        click.echo(
-            click.style("Could not launch program: {}".format(program), fg="red")
-        )
-        sys.exit(1)
-    globs = {
-        "__file__": program,
-        "__name__": "__main__",
-        "__package__": None,
-        "wandb_magic_install": magic_install,
-    }
-    prep = """
-import __main__
-__main__.__file__ = "{}"
-wandb_magic_install()
-""".format(program)
-    magic_run(prep, globs, None)
-    magic_run(code, globs, None)
-
-
 @cli.command("online", help="Enable W&B sync")
 @display_error
 def online():

wandb/env.py

@@ -82,11 +82,11 @@ SENTRY_DSN = "WANDB_SENTRY_DSN"
 INIT_TIMEOUT = "WANDB_INIT_TIMEOUT"
 GIT_COMMIT = "WANDB_GIT_COMMIT"
 GIT_REMOTE_URL = "WANDB_GIT_REMOTE_URL"
-_EXECUTABLE = "WANDB_EXECUTABLE"
+_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__REQUIRE_LEGACY_SERVICE"
+_REQUIRE_LEGACY_SERVICE = "WANDB_X_REQUIRE_LEGACY_SERVICE"
 
 # For testing, to be removed in future version
 USE_V1_ARTIFACTS = "_WANDB_USE_V1_ARTIFACTS"
@@ -290,9 +290,7 @@ def get_base_url(
     if env is None:
         env = os.environ
 
-    base_url = env.get(BASE_URL, default)
-
-    return base_url.rstrip("/") if base_url is not None else base_url
+    return env.get(BASE_URL, default)
 
 
 def get_app_url(

wandb/errors/links.py

@@ -19,7 +19,7 @@ class WBURL:
 class Registry:
     """A collection of URLs that can be associated with a name."""
 
-    def __init__(self):
+    def __init__(self) -> None:
         self.urls: dict[str, WBURL] = {
             "wandb-launch": WBURL(
                 "https://wandb.me/launch",

wandb/errors/term.py

@@ -7,12 +7,7 @@ import logging
 import os
 import sys
 import threading
-from typing import TYPE_CHECKING, Iterator
-
-if sys.version_info < (3, 8):
-    from typing_extensions import Protocol
-else:
-    from typing import Protocol
+from typing import TYPE_CHECKING, Iterator, Protocol
 
 import click
 

wandb/filesync/dir_watcher.py

@@ -122,9 +122,9 @@ class PolicyLive(FileEventHandler):
         self._last_uploaded_time: Optional[float] = None
         self._last_uploaded_size: int = 0
         if settings is not None:
-            if settings._live_policy_rate_limit is not None:
-                self.RATE_LIMIT_SECONDS = settings._live_policy_rate_limit
-            self._min_wait_time: Optional[float] = settings._live_policy_wait_time
+            if settings.x_live_policy_rate_limit is not None:
+                self.RATE_LIMIT_SECONDS = settings.x_live_policy_rate_limit
+            self._min_wait_time: Optional[float] = settings.x_live_policy_wait_time
         else:
             self._min_wait_time = None
 

wandb/filesync/step_upload.py

@@ -21,15 +21,12 @@ from wandb.filesync import upload_job
 from wandb.sdk.lib.paths import LogicalPath
 
 if TYPE_CHECKING:
+    from typing import TypedDict
+
     from wandb.filesync import stats
     from wandb.sdk.internal import file_stream, internal_api, progress
     from wandb.sdk.internal.settings_static import SettingsStatic
 
-    if sys.version_info >= (3, 8):
-        from typing import TypedDict
-    else:
-        from typing_extensions import TypedDict
-
     class ArtifactStatus(TypedDict):
         finalize: bool
         pending_count: int

wandb/integration/fastai/__init__.py

@@ -39,12 +39,7 @@ Examples:
 import random
 import sys
 from pathlib import Path
-from typing import Any, Optional
-
-if sys.version_info >= (3, 8):
-    from typing import Literal
-else:
-    from typing_extensions import Literal
+from typing import Any, Literal, Optional
 
 import fastai
 from fastai.callbacks import TrackerCallback

wandb/integration/gym/__init__.py

@@ -1,16 +1,10 @@
 import re
 import sys
-from typing import Optional
+from typing import Literal, Optional
 
 import wandb
 import wandb.util
 
-if sys.version_info >= (3, 8):
-    from typing import Literal
-else:
-    from typing_extensions import Literal
-
-
 _gym_version_lt_0_26: Optional[bool] = None
 _gymnasium_version_lt_1_0_0: Optional[bool] = None
 

wandb/integration/keras/callbacks/metrics_logger.py

@@ -1,5 +1,4 @@
-import sys
-from typing import Any, Dict, Optional, Union
+from typing import Any, Dict, Literal, Optional, Union
 
 import tensorflow as tf  # type: ignore
 from tensorflow.keras import callbacks
@@ -8,12 +7,6 @@ import wandb
 from wandb.integration.keras.keras import patch_tf_keras
 from wandb.sdk.lib import telemetry
 
-if sys.version_info >= (3, 8):
-    from typing import Literal
-else:
-    from typing_extensions import Literal
-
-
 LogStrategy = Literal["epoch", "batch"]