wandb 0.16.6__py3-none-any.whl → 0.17.0rc1__py3-none-any.whl

wandb/sdk/wandb_run.py

@@ -327,8 +327,7 @@ class RunStatusChecker:
 class _run_decorator:  # noqa: N801
     _is_attaching: str = ""
 
-    class Dummy:
-        ...
+    class Dummy: ...
 
     @classmethod
     def _attach(cls, func: Callable) -> Callable:
@@ -1866,16 +1865,19 @@ class Run:
         # => Saves files in an "are/myfiles/" folder in the run.
 
         wandb.save("/User/username/Documents/run123/*.txt")
-        # => Saves files in a "run123/" folder in the run.
+        # => Saves files in a "run123/" folder in the run. See note below.
 
         wandb.save("/User/username/Documents/run123/*.txt", base_path="/User")
         # => Saves files in a "username/Documents/run123/" folder in the run.
 
         wandb.save("files/*/saveme.txt")
         # => Saves each "saveme.txt" file in an appropriate subdirectory
-        # of "files/".
+        #    of "files/".
         ```
 
+        Note: when given an absolute path or glob and no `base_path`, one
+        directory level is preserved as in the example above.
+
         Arguments:
             glob_str: A relative or absolute path or Unix glob.
             base_path: A path to use to infer a directory structure; see examples.
@@ -3171,13 +3173,13 @@ class Run:
             )
         if entity and artifact._source_entity and entity != artifact._source_entity:
             raise ValueError(
-                f"Artifact {artifact.name} is owned by entity '{entity}'; it can't be "
-                f"moved to '{artifact._source_entity}'"
+                f"Artifact {artifact.name} is owned by entity "
+                f"'{artifact._source_entity}'; it can't be moved to '{entity}'"
             )
         if project and artifact._source_project and project != artifact._source_project:
             raise ValueError(
-                f"Artifact {artifact.name} exists in project '{project}'; it can't be "
-                f"moved to '{artifact._source_project}'"
+                f"Artifact {artifact.name} exists in project "
+                f"'{artifact._source_project}'; it can't be moved to '{project}'"
             )
 
     def _prepare_artifact(
@@ -3304,8 +3306,8 @@ class Run:
             path: (str) path to downloaded model artifact file(s).
         """
         artifact = self.use_artifact(artifact_or_name=name)
-        assert "model" in str(
-            artifact.type.lower()
+        assert (
+            "model" in str(artifact.type.lower())
         ), "You can only use this method for 'model' artifacts. For an artifact to be a 'model' artifact, its type property must contain the substring 'model'."
         path = artifact.download()
 
@@ -3397,8 +3399,8 @@ class Run:
         public_api = self._public_api()
         try:
             artifact = public_api.artifact(name=f"{name}:latest")
-            assert "model" in str(
-                artifact.type.lower()
+            assert (
+                "model" in str(artifact.type.lower())
             ), "You can only use this method for 'model' artifacts. For an artifact to be a 'model' artifact, its type property must contain the substring 'model'."
             artifact = self._log_artifact(
                 artifact_or_path=path, name=name, type=artifact.type

wandb/sdk/wandb_settings.py

@@ -1675,7 +1675,6 @@ class Settings(SettingsData):
             "WANDB_TRACELOG": "_tracelog",
             "WANDB_DISABLE_SERVICE": "_disable_service",
             "WANDB_SERVICE_TRANSPORT": "_service_transport",
-            "WANDB_REQUIRE_CORE": "_require_core",
             "WANDB_DIR": "root_dir",
             "WANDB_NAME": "run_name",
             "WANDB_NOTES": "run_notes",

wandb/sklearn/__init__.py

@@ -1,4 +1,5 @@
 """Create informative charts for scikit-learn models and log them to W&B."""
+
 from .plot import (
     plot_calibration_curve,
     plot_class_proportions,

wandb/sklearn/plot/__init__.py

@@ -1,4 +1,5 @@
 """Create and logs charts introspecting models built with scikit-learn to W&B."""
+
 from .classifier import calibration_curve as plot_calibration_curve
 from .classifier import class_proportions as plot_class_proportions
 from .classifier import classifier as plot_classifier

wandb/sklearn/plot/classifier.py

@@ -1,4 +1,5 @@
 """Define plots for classification models built with scikit-learn."""
+
 from warnings import simplefilter
 
 import numpy as np

wandb/sklearn/plot/clusterer.py

@@ -1,4 +1,5 @@
 """Define plots for clustering models built with scikit-learn."""
+
 from warnings import simplefilter
 
 import pandas as pd

wandb/sklearn/plot/regressor.py

@@ -1,4 +1,5 @@
 """Define plots for regression models built with scikit-learn."""
+
 from warnings import simplefilter
 
 import numpy as np

wandb/sklearn/plot/shared.py

@@ -1,4 +1,5 @@
 """Define plots used by multiple sklearn model classes."""
+
 from warnings import simplefilter
 
 import numpy as np

wandb/sklearn/utils.py

@@ -1,4 +1,5 @@
 """Shared utilities for the modules in wandb.sklearn."""
+
 from collections.abc import Iterable, Sequence
 
 import numpy as np

wandb/testing/relay.py

@@ -586,11 +586,11 @@ class InjectedResponse:
 
 
 class RelayControlProtocol(Protocol):
-    def process(self, request: "flask.Request") -> None:
-        ...  # pragma: no cover
+    def process(self, request: "flask.Request") -> None: ...  # pragma: no cover
 
-    def control(self, request: "flask.Request") -> Mapping[str, str]:
-        ...  # pragma: no cover
+    def control(
+        self, request: "flask.Request"
+    ) -> Mapping[str, str]: ...  # pragma: no cover
 
 
 class RelayServer:

wandb/trigger.py

@@ -6,6 +6,7 @@ Usage:
     trigger.call('on_something', *args, **kwargs)
     trigger.unregister('on_something', func)
 """
+
 from typing import Any, Callable
 
 _triggers = {}

wandb/util.py

@@ -11,6 +11,7 @@ import logging
 import math
 import numbers
 import os
+import pathlib
 import platform
 import queue
 import random
@@ -55,7 +56,13 @@ import yaml
 
 import wandb
 import wandb.env
-from wandb.errors import AuthenticationError, CommError, UsageError, term
+from wandb.errors import (
+    AuthenticationError,
+    CommError,
+    UsageError,
+    WandbCoreNotAvailableError,
+    term,
+)
 from wandb.sdk.internal.thread_local_settings import _thread_local_api_settings
 from wandb.sdk.lib import filesystem, runid
 from wandb.sdk.lib.json_util import dump, dumps
@@ -1842,15 +1849,6 @@ def sample_with_exponential_decay_weights(
     return sampled_xs, sampled_ys, sampled_keys
 
 
-def get_core_path() -> str:
-    core_path: str = os.environ.get("_WANDB_CORE_PATH", "")
-    wandb_core = get_module("wandb_core")
-    if not core_path and wandb_core:
-        _check_wandb_core_version_compatibility(wandb_core.__version__)
-        core_path = wandb_core.get_core_path()
-    return core_path
-
-
 @dataclasses.dataclass(frozen=True)
 class InstalledDistribution:
     """An installed distribution.
@@ -1888,15 +1886,40 @@ def parse_version(version: str) -> "packaging.version.Version":
     try:
         from packaging.version import parse as parse_version  # type: ignore
     except ImportError:
-        from pkg_resources import parse_version
+        from pkg_resources import parse_version  # type: ignore[assignment]
 
     return parse_version(version)
 
 
-def _check_wandb_core_version_compatibility(core_version: str) -> None:
-    """Checks if the installed wandb-core version is compatible with the wandb version."""
-    if parse_version(core_version) < parse_version(wandb._minimum_core_version):
-        raise ImportError(
-            f"Requires wandb-core version {wandb._minimum_core_version} or later, "
-            f"but you have {core_version}. Run `pip install --upgrade wandb-core` to upgrade."
+def get_core_path() -> str:
+    """Returns the path to the wandb-core binary.
+
+    The path can be set explicitly via the _WANDB_CORE_PATH environment
+    variable. Otherwise, the path to the binary in the current package
+    is returned.
+
+    Returns:
+        str: The path to the wandb-core package.
+
+    Raises:
+        WandbCoreNotAvailableError: If wandb-core was not built for the current system.
+    """
+    # NOTE: Environment variable _WANDB_CORE_PATH is a temporary development feature
+    #       to assist in running the core service from a live development directory.
+    path_from_env: str = os.environ.get("_WANDB_CORE_PATH", "")
+    if path_from_env:
+        wandb.termwarn(
+            f"Using wandb-core from path `_WANDB_CORE_PATH={path_from_env}`. "
+            "This is a development feature and may not work as expected."
+        )
+        return path_from_env
+
+    bin_path = pathlib.Path(__file__).parent / "bin" / "wandb-core"
+    if not bin_path.exists():
+        raise WandbCoreNotAvailableError(
+            f"Looks like wandb-core is not compiled for your system ({platform.platform()}):"
+            " Please contact support at support@wandb.com to request `wandb-core`"
+            " support for your system."
         )
+
+    return str(bin_path)

wandb/wandb_controller.py

@@ -47,7 +47,6 @@ Example:
             tuner.stop_runs(runs)
 """
 
-
 import json
 import os
 import random

wandb/wandb_torch.py

@@ -1,7 +1,6 @@
 #!/usr/bin/env python
 
-"""PyTorch-specific functionality
-"""
+"""PyTorch-specific functionality"""
 
 import itertools
 from functools import reduce

{wandb-0.16.6.dist-info → wandb-0.17.0rc1.dist-info}/METADATA

@@ -1,7 +1,10 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: wandb
-Version: 0.16.6
+Version: 0.17.0rc1
 Summary: A CLI and library for interacting with the Weights & Biases API.
+Project-URL: Source, https://github.com/wandb/wandb
+Project-URL: Bug Reports, https://github.com/wandb/wandb/issues
+Project-URL: Documentation, https://docs.wandb.ai/
 Author-email: Weights & Biases <support@wandb.com>
 License: MIT License
         
@@ -24,104 +27,100 @@ License: MIT License
         LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
         OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
         SOFTWARE.
-        
-Project-URL: Source, https://github.com/wandb/wandb
-Project-URL: Bug Reports, https://github.com/wandb/wandb/issues
-Project-URL: Documentation, https://docs.wandb.ai/
+License-File: LICENSE
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Natural Language :: English
 Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Programming Language :: Python :: 3.7
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Topic :: System :: Logging
 Classifier: Topic :: System :: Monitoring
 Requires-Python: >=3.7
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: Click !=8.0.0,>=7.1
-Requires-Dist: GitPython !=3.1.29,>=1.0.0
-Requires-Dist: requests <3,>=2.0.0
-Requires-Dist: psutil >=5.0.0
-Requires-Dist: sentry-sdk >=1.0.0
-Requires-Dist: docker-pycreds >=0.4.0
-Requires-Dist: PyYAML
+Requires-Dist: appdirs>=1.4.3
+Requires-Dist: click!=8.0.0,>=7.1
+Requires-Dist: docker-pycreds>=0.4.0
+Requires-Dist: gitpython!=3.1.29,>=1.0.0
+Requires-Dist: protobuf!=4.21.0,<5,>=3.12.0; python_version < '3.9' and sys_platform == 'linux'
+Requires-Dist: protobuf!=4.21.0,<5,>=3.15.0; python_version == '3.9' and sys_platform == 'linux'
+Requires-Dist: protobuf!=4.21.0,<5,>=3.19.0; python_version > '3.9' and sys_platform == 'linux'
+Requires-Dist: protobuf!=4.21.0,<5,>=3.19.0; sys_platform != 'linux'
+Requires-Dist: psutil>=5.0.0
+Requires-Dist: pyyaml
+Requires-Dist: requests<3,>=2.0.0
+Requires-Dist: sentry-sdk>=1.0.0
 Requires-Dist: setproctitle
 Requires-Dist: setuptools
-Requires-Dist: appdirs >=1.4.3
-Requires-Dist: typing-extensions ; python_version < "3.10"
-Requires-Dist: protobuf !=4.21.0,<5,>=3.12.0 ; python_version < "3.9" and sys_platform == "linux"
-Requires-Dist: protobuf !=4.21.0,<5,>=3.15.0 ; python_version == "3.9" and sys_platform == "linux"
-Requires-Dist: protobuf !=4.21.0,<5,>=3.19.0 ; python_version > "3.9" and sys_platform == "linux"
-Requires-Dist: protobuf !=4.21.0,<5,>=3.19.0 ; sys_platform != "linux"
+Requires-Dist: typing-extensions; python_version < '3.10'
 Provides-Extra: async
-Requires-Dist: httpx >=0.23.0 ; extra == 'async'
+Requires-Dist: httpx>=0.23.0; extra == 'async'
 Provides-Extra: aws
-Requires-Dist: boto3 ; extra == 'aws'
+Requires-Dist: boto3; extra == 'aws'
 Provides-Extra: azure
-Requires-Dist: azure-identity ; extra == 'azure'
-Requires-Dist: azure-storage-blob ; extra == 'azure'
+Requires-Dist: azure-identity; extra == 'azure'
+Requires-Dist: azure-storage-blob; extra == 'azure'
 Provides-Extra: gcp
-Requires-Dist: google-cloud-storage ; extra == 'gcp'
+Requires-Dist: google-cloud-storage; extra == 'gcp'
 Provides-Extra: importers
-Requires-Dist: polars ; extra == 'importers'
-Requires-Dist: rich ; extra == 'importers'
-Requires-Dist: filelock ; extra == 'importers'
-Requires-Dist: mlflow ; extra == 'importers'
-Requires-Dist: tenacity ; extra == 'importers'
+Requires-Dist: filelock; extra == 'importers'
+Requires-Dist: mlflow; extra == 'importers'
+Requires-Dist: polars; extra == 'importers'
+Requires-Dist: rich; extra == 'importers'
+Requires-Dist: tenacity; extra == 'importers'
 Provides-Extra: kubeflow
-Requires-Dist: kubernetes ; extra == 'kubeflow'
-Requires-Dist: minio ; extra == 'kubeflow'
-Requires-Dist: google-cloud-storage ; extra == 'kubeflow'
-Requires-Dist: sh ; extra == 'kubeflow'
+Requires-Dist: google-cloud-storage; extra == 'kubeflow'
+Requires-Dist: kubernetes; extra == 'kubeflow'
+Requires-Dist: minio; extra == 'kubeflow'
+Requires-Dist: sh; extra == 'kubeflow'
 Provides-Extra: launch
-Requires-Dist: awscli ; extra == 'launch'
-Requires-Dist: azure-identity ; extra == 'launch'
-Requires-Dist: azure-containerregistry ; extra == 'launch'
-Requires-Dist: azure-storage-blob ; extra == 'launch'
-Requires-Dist: boto3 ; extra == 'launch'
-Requires-Dist: botocore ; extra == 'launch'
-Requires-Dist: chardet ; extra == 'launch'
-Requires-Dist: google-auth ; extra == 'launch'
-Requires-Dist: google-cloud-aiplatform ; extra == 'launch'
-Requires-Dist: google-cloud-artifact-registry ; extra == 'launch'
-Requires-Dist: google-cloud-compute ; extra == 'launch'
-Requires-Dist: google-cloud-storage ; extra == 'launch'
-Requires-Dist: iso8601 ; extra == 'launch'
-Requires-Dist: kubernetes ; extra == 'launch'
-Requires-Dist: kubernetes-asyncio ; extra == 'launch'
-Requires-Dist: optuna ; extra == 'launch'
-Requires-Dist: nbconvert ; extra == 'launch'
-Requires-Dist: nbformat ; extra == 'launch'
-Requires-Dist: pydantic ; extra == 'launch'
-Requires-Dist: typing-extensions ; extra == 'launch'
-Requires-Dist: tomli ; extra == 'launch'
-Requires-Dist: PyYAML >=6.0.0 ; extra == 'launch'
+Requires-Dist: awscli; extra == 'launch'
+Requires-Dist: azure-containerregistry; extra == 'launch'
+Requires-Dist: azure-identity; extra == 'launch'
+Requires-Dist: azure-storage-blob; extra == 'launch'
+Requires-Dist: boto3; extra == 'launch'
+Requires-Dist: botocore; extra == 'launch'
+Requires-Dist: chardet; extra == 'launch'
+Requires-Dist: google-auth; extra == 'launch'
+Requires-Dist: google-cloud-aiplatform; extra == 'launch'
+Requires-Dist: google-cloud-artifact-registry; extra == 'launch'
+Requires-Dist: google-cloud-compute; extra == 'launch'
+Requires-Dist: google-cloud-storage; extra == 'launch'
+Requires-Dist: iso8601; extra == 'launch'
+Requires-Dist: kubernetes; extra == 'launch'
+Requires-Dist: kubernetes-asyncio; extra == 'launch'
+Requires-Dist: nbconvert; extra == 'launch'
+Requires-Dist: nbformat; extra == 'launch'
+Requires-Dist: optuna; extra == 'launch'
+Requires-Dist: pydantic; extra == 'launch'
+Requires-Dist: pyyaml>=6.0.0; extra == 'launch'
+Requires-Dist: tomli; extra == 'launch'
+Requires-Dist: typing-extensions; extra == 'launch'
 Provides-Extra: media
-Requires-Dist: numpy ; extra == 'media'
-Requires-Dist: moviepy ; extra == 'media'
-Requires-Dist: pillow ; extra == 'media'
-Requires-Dist: bokeh ; extra == 'media'
-Requires-Dist: soundfile ; extra == 'media'
-Requires-Dist: plotly >=5.18.0 ; extra == 'media'
-Requires-Dist: rdkit-pypi ; extra == 'media'
+Requires-Dist: bokeh; extra == 'media'
+Requires-Dist: moviepy; extra == 'media'
+Requires-Dist: numpy; extra == 'media'
+Requires-Dist: pillow; extra == 'media'
+Requires-Dist: plotly>=5.18.0; extra == 'media'
+Requires-Dist: rdkit-pypi; extra == 'media'
+Requires-Dist: soundfile; extra == 'media'
 Provides-Extra: models
-Requires-Dist: cloudpickle ; extra == 'models'
+Requires-Dist: cloudpickle; extra == 'models'
 Provides-Extra: perf
-Requires-Dist: orjson ; extra == 'perf'
+Requires-Dist: orjson; extra == 'perf'
 Provides-Extra: reports
-Requires-Dist: pydantic >=2.0.0 ; extra == 'reports'
+Requires-Dist: pydantic>=2.0.0; extra == 'reports'
 Provides-Extra: sweeps
-Requires-Dist: sweeps >=0.2.0 ; extra == 'sweeps'
+Requires-Dist: sweeps>=0.2.0; extra == 'sweeps'
+Description-Content-Type: text/markdown
 
 <div align="center">
   <img src="https://i.imgur.com/RUtiVzH.png" width="600" /><br><br>