wandb 0.15.4__tar.gz → 0.15.5__tar.gz

{wandb-0.15.4 → wandb-0.15.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wandb
-Version: 0.15.4
+Version: 0.15.5
 Summary: A CLI and library for interacting with the Weights and Biases API.
 Home-page: https://github.com/wandb/wandb
 Author: Weights & Biases

{wandb-0.15.4 → wandb-0.15.5}/README.md

@@ -26,57 +26,64 @@ Want to use Weights & Biases for seamless collaboration between your ML or Data
 <p align='center'>
 <a target="_blank" href="https://docs.wandb.ai/guides/track?utm_source=github&utm_medium=code&utm_campaign=wandb&utm_content=readme">
 <picture>
-  <source media="(prefers-color-scheme: dark)" srcset="./docs/README_images/Product_Icons_dark_background/experiments-dark.svg" width="12.0%">
-  <source media="(prefers-color-scheme: light)" srcset="./docs/README_images/Product_Icons_light/experiments-light.svg" width="12.0%">
+  <source media="(prefers-color-scheme: dark)" srcset="./docs/README_images/Product_Icons_dark_background/experiments-dark.svg" width="14.0%">
+  <source media="(prefers-color-scheme: light)" srcset="./docs/README_images/Product_Icons_light/experiments-light.svg" width="14.0%">
   <img alt="Weights and Biases Experiments" src="">
 </picture>
 </a>
 <a target="_blank" href="https://docs.wandb.ai/guides/reports?utm_source=github&utm_medium=code&utm_campaign=wandb&utm_content=readme">
 <picture>
-  <source media="(prefers-color-scheme: dark)" srcset="./docs/README_images/Product_Icons_dark_background/report-dark.svg" width="12.0%">
-  <source media="(prefers-color-scheme: light)" srcset="./docs/README_images/Product_Icons_light/report-light.svg" width="12.0%">
+  <source media="(prefers-color-scheme: dark)" srcset="./docs/README_images/Product_Icons_dark_background/report-dark.svg" width="14.0%">
+  <source media="(prefers-color-scheme: light)" srcset="./docs/README_images/Product_Icons_light/report-light.svg" width="14.0%">
   <img alt="Weights and Biases Reports" src="">
 </picture>
 </a>
 <a target="_blank" href="https://docs.wandb.ai/guides/artifacts?utm_source=github&utm_medium=code&utm_campaign=wandb&utm_content=readme">
 <picture>
-  <source media="(prefers-color-scheme: dark)" srcset="./docs/README_images/Product_Icons_dark_background/artifacts-dark.svg" width="12.0%">
-  <source media="(prefers-color-scheme: light)" srcset="./docs/README_images/Product_Icons_light/artifacts-light.svg" width="12.0%">
+  <source media="(prefers-color-scheme: dark)" srcset="./docs/README_images/Product_Icons_dark_background/artifacts-dark.svg" width="14.0%">
+  <source media="(prefers-color-scheme: light)" srcset="./docs/README_images/Product_Icons_light/artifacts-light.svg" width="14.0%">
   <img alt="Weights and Biases Artifacts" src="">
 </picture>
 </a>
 <a target="_blank" href="https://docs.wandb.ai/guides/data-vis?utm_source=github&utm_medium=code&utm_campaign=wandb&utm_content=readme">
 <picture>
-  <source media="(prefers-color-scheme: dark)" srcset="./docs/README_images/Product_Icons_dark_background/tables-dark.svg" width="12.0%">
-  <source media="(prefers-color-scheme: light)" srcset="./docs/README_images/Product_Icons_light/tables-light.svg" width="12.0%">
+  <source media="(prefers-color-scheme: dark)" srcset="./docs/README_images/Product_Icons_dark_background/tables-dark.svg" width="14.0%">
+  <source media="(prefers-color-scheme: light)" srcset="./docs/README_images/Product_Icons_light/tables-light.svg" width="14.0%">
   <img alt="Weights and Biases Tables" src="">
 </picture>
 </a>
 <a target="_blank" href="https://docs.wandb.ai/guides/sweeps?utm_source=github&utm_medium=code&utm_campaign=wandb&utm_content=readme">
 <picture>
-  <source media="(prefers-color-scheme: dark)" srcset="./docs/README_images/Product_Icons_dark_background/sweeps-dark.svg" width="12.0%">
-  <source media="(prefers-color-scheme: light)" srcset="./docs/README_images/Product_Icons_light/sweeps-light.svg" width="12.0%">
+  <source media="(prefers-color-scheme: dark)" srcset="./docs/README_images/Product_Icons_dark_background/sweeps-dark.svg" width="14.0%">
+  <source media="(prefers-color-scheme: light)" srcset="./docs/README_images/Product_Icons_light/sweeps-light.svg" width="14.0%">
   <img alt="Weights and Biases Sweeps" src="">
 </picture>
 </a>
-<a target="_blank" href="https://docs.wandb.ai/guides/models?utm_source=github&utm_medium=code&utm_campaign=wandb&utm_content=readme">
+<a target="_blank" href="https://docs.wandb.ai/guides/launch?utm_source=github&utm_medium=code&utm_campaign=wandb&utm_content=readme">
 <picture>
-  <source media="(prefers-color-scheme: dark)" srcset="./docs/README_images/Product_Icons_dark_background/models-dark.svg" width="12.0%">
-  <source media="(prefers-color-scheme: light)" srcset="./docs/README_images/Product_Icons_light/models-light.svg" width="12.0%">
-  <img alt="Weights and Biases Model Management" src="">
+  <source media="(prefers-color-scheme: dark)" srcset="./docs/README_images/Product_Icons_dark_background/launch-dark.svg" width="14.0%">
+  <source media="(prefers-color-scheme: light)" srcset="./docs/README_images/Product_Icons_light/launch-light.svg" width="14.0%">
+  <img alt="Weights and Biases Launch" src="">
 </picture>
 </a>
-<a target="_blank" href="https://docs.wandb.ai/guides/launch?utm_source=github&utm_medium=code&utm_campaign=wandb&utm_content=readme">
+<a target="_blank" href="https://docs.wandb.ai/guides/models?utm_source=github&utm_medium=code&utm_campaign=wandb&utm_content=readme">
 <picture>
-  <source media="(prefers-color-scheme: dark)" srcset="./docs/README_images/Product_Icons_dark_background/launch-dark.svg" width="12.0%">
-  <source media="(prefers-color-scheme: light)" srcset="./docs/README_images/Product_Icons_light/launch-light.svg" width="12.0%">
-  <img alt="Weights and Biases Launch" src="">
+  <source media="(prefers-color-scheme: dark)" srcset="./docs/README_images/Product_Icons_dark_background/models-dark.svg" width="14.0%">
+  <source media="(prefers-color-scheme: light)" srcset="./docs/README_images/Product_Icons_light/models-light.svg" width="14.0%">
+  <img alt="Weights and Biases Model Management" src="">
 </picture>
 </a>
+</a>
 <a target="_blank" href="https://docs.wandb.ai/guides/prompts?utm_source=github&utm_medium=code&utm_campaign=wandb&utm_content=readme">
 <picture>
-  <source media="(prefers-color-scheme: dark)" srcset="./docs/README_images/Product_Icons_dark_background/prompts-dark.svg" width="12.0%">
-  <source media="(prefers-color-scheme: light)" srcset="./docs/README_images/Product_Icons_light/prompts-light.svg" width="12.0%">
+  <source media="(prefers-color-scheme: dark)" srcset="./docs/README_images/Product_Icons_dark_background/prompts-dark.svg" width="14.0%">
+  <source media="(prefers-color-scheme: light)" srcset="./docs/README_images/Product_Icons_light/prompts-light.svg" width="14.0%">
+  <img alt="Weights and Biases Prompts" src="">
+</picture>
+<a target="_blank" href="https://github.com/wandb/weave">
+<picture>
+  <source media="(prefers-color-scheme: dark)" srcset="./docs/README_images/Product_Icons_dark_background/weave-dark.svg" width="14.0%">
+  <source media="(prefers-color-scheme: light)" srcset="./docs/README_images/Product_Icons_light/weave-light.svg" width="14.0%">
   <img alt="Weights and Biases Prompts" src="">
 </picture>
 </p>

{wandb-0.15.4 → wandb-0.15.5}/pyproject.toml

@@ -29,7 +29,7 @@ skip_gitignore = "True"
 [tool.commitizen]
 name = "cz_conventional_commits"
 major_version_zero = true
-version = "0.15.4"
+version = "0.15.5"
 version_files = ["setup.py", "setup.cfg", "wandb/__init__.py"]
 tag_format = "v$major.$minor.$patch$prerelease"
 changelog_incremental = true

{wandb-0.15.4 → wandb-0.15.5}/requirements.txt

@@ -1,4 +1,4 @@
-Click>=7.0,!=8.0.0  # click 8.0.0 is broken
+Click>=7.1,!=8.0.0  # click 8.0.0 is broken
 GitPython>=1.0.0,!=3.1.29  # CVE-2022-24439
 requests>=2.0.0,<3
 psutil>=5.0.0

{wandb-0.15.4 → wandb-0.15.5}/setup.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.15.4
+current_version = 0.15.5
 commit = True
 tag = False
 parse = (?P<major>\d+)\.(?P<minor>\d+)\.(?P<patch>\d+)((?P<prekind>[a-z]+)(?P<pre>\d+))?(\.(?P<devkind>[a-z]+)(?P<dev>\d+))?

{wandb-0.15.4 → wandb-0.15.5}/setup.py

@@ -28,19 +28,22 @@ media_requirements = [
 ]
 launch_requirements = [
     "awscli",
-    "nbconvert",
-    "nbformat",
-    "chardet",
-    "iso8601",
-    "typing_extensions",
+    "azure-identity",
+    "azure-containerregistry",
+    "azure-storage-blob",
     "boto3",
     "botocore",
+    "chardet",
     "google-auth",
+    "google-cloud-artifact-registry",
     "google-cloud-compute",
     "google-cloud-storage",
-    "google-cloud-artifact-registry",
+    "iso8601",
     "kubernetes",
     "optuna",
+    "nbconvert",
+    "nbformat",
+    "typing_extensions",
 ]
 
 models_requirements = ["cloudpickle"]
@@ -52,7 +55,7 @@ async_requirements = [
 
 setup(
     name="wandb",
-    version="0.15.4",
+    version="0.15.5",
     description="A CLI and library for interacting with the Weights and Biases API.",
     long_description=readme,
     long_description_content_type="text/markdown",

{wandb-0.15.4 → wandb-0.15.5}/wandb/__init__.py

@@ -11,7 +11,7 @@ For scripts and interactive notebooks, see https://github.com/wandb/examples.
 
 For reference documentation, see https://docs.wandb.com/ref/python.
 """
-__version__ = "0.15.4"
+__version__ = "0.15.5"
 
 # Used with pypi checks and other messages related to pip
 _wandb_module = "wandb"

{wandb-0.15.4 → wandb-0.15.5}/wandb/analytics/sentry.py

@@ -207,6 +207,7 @@ class Sentry:
             "sweep_id",
             "deployment",
             "_disable_service",
+            "_require_nexus",
             "launch",
         )
 

{wandb-0.15.4 → wandb-0.15.5}/wandb/apis/internal.py

@@ -209,6 +209,9 @@ class Api:
     def fail_run_queue_item(self, *args, **kwargs):
         return self.api.fail_run_queue_item(*args, **kwargs)
 
+    def update_run_queue_item_warning(self, *args, **kwargs):
+        return self.api.update_run_queue_item_warning(*args, **kwargs)
+
     def get_launch_agent(self, *args, **kwargs):
         return self.api.get_launch_agent(*args, **kwargs)
 

{wandb-0.15.4 → wandb-0.15.5}/wandb/apis/public.py

@@ -41,7 +41,6 @@ from wandb import __version__, env, util
 from wandb.apis.internal import Api as InternalApi
 from wandb.apis.normalize import normalize_exceptions
 from wandb.errors import CommError
-from wandb.sdk import artifacts
 from wandb.sdk.data_types._dtypes import InvalidType, Type, TypeRegistry
 from wandb.sdk.internal.thread_local_settings import _thread_local_api_settings
 from wandb.sdk.launch.errors import LaunchError
@@ -909,14 +908,13 @@ class Api:
 
     @normalize_exceptions
     def artifact(self, name, type=None):
-        """Return a single artifact by parsing path in the form `entity/project/run_id`.
+        """Return a single artifact by parsing path in the form `entity/project/name`.
 
         Arguments:
             name: (str) An artifact name. May be prefixed with entity/project. Valid names
                 can be in the following forms:
                     name:version
                     name:alias
-                    digest
             type: (str, optional) The type of artifact to fetch.
 
         Returns:
@@ -925,7 +923,9 @@ class Api:
         if name is None:
             raise ValueError("You must specify name= to fetch an artifact.")
         entity, project, artifact_name = self._parse_artifact_path(name)
-        artifact = artifacts.PublicArtifact(self.client, entity, project, artifact_name)
+        artifact = wandb.Artifact._from_name(
+            entity, project, artifact_name, self.client
+        )
         if type is not None and artifact.type != type:
             raise ValueError(
                 f"type {type} specified but this artifact is of type {artifact.type}"
@@ -2140,10 +2140,10 @@ class Run(Attrs):
         )
         api.set_current_run_id(self.id)
 
-        if isinstance(artifact, artifacts.PublicArtifact):
+        if isinstance(artifact, wandb.Artifact) and not artifact.is_draft():
             api.use_artifact(artifact.id, use_as=use_as or artifact.name)
             return artifact
-        elif isinstance(artifact, wandb.Artifact):
+        elif isinstance(artifact, wandb.Artifact) and artifact.is_draft():
             raise ValueError(
                 "Only existing artifacts are accepted by this api. "
                 "Manually create one with `wandb artifacts put`"
@@ -2168,7 +2168,7 @@ class Run(Attrs):
         )
         api.set_current_run_id(self.id)
 
-        if isinstance(artifact, artifacts.PublicArtifact):
+        if isinstance(artifact, wandb.Artifact) and not artifact.is_draft():
             artifact_collection_name = artifact.name.split(":")[0]
             api.create_artifact(
                 artifact.type,
@@ -2177,7 +2177,7 @@ class Run(Attrs):
                 aliases=aliases,
             )
             return artifact
-        elif isinstance(artifact, wandb.Artifact):
+        elif isinstance(artifact, wandb.Artifact) and artifact.is_draft():
             raise ValueError(
                 "Only existing artifacts are accepted by this api. "
                 "Manually create one with `wandb artifacts put`"
@@ -3814,7 +3814,7 @@ class RunArtifacts(Paginator):
             }
             %s
             """
-            % artifacts.ARTIFACT_FRAGMENT
+            % wandb.Artifact._GQL_FRAGMENT
         )
 
         input_query = gql(
@@ -3842,7 +3842,7 @@ class RunArtifacts(Paginator):
             }
             %s
             """
-            % artifacts.ARTIFACT_FRAGMENT
+            % wandb.Artifact._GQL_FRAGMENT
         )
 
         self.run = run
@@ -3890,14 +3890,14 @@ class RunArtifacts(Paginator):
 
     def convert_objects(self):
         return [
-            artifacts.PublicArtifact(
-                self.client,
+            wandb.Artifact._from_attrs(
                 self.run.entity,
                 self.run.project,
                 "{}:v{}".format(
                     r["node"]["artifactSequence"]["name"], r["node"]["versionIndex"]
                 ),
                 r["node"],
+                self.client,
             )
             for r in self.last_response["project"]["run"][self.run_key]["edges"]
         ]
@@ -4144,7 +4144,7 @@ class ArtifactVersions(Paginator):
                 artifact_collection_edge_name(
                     server_supports_artifact_collections_gql_edges(client)
                 ),
-                artifacts.ARTIFACT_FRAGMENT,
+                wandb.Artifact._GQL_FRAGMENT,
             )
         )
         super().__init__(client, variables, per_page)
@@ -4180,12 +4180,12 @@ class ArtifactVersions(Paginator):
         if self.last_response["project"]["artifactType"]["artifactCollection"] is None:
             return []
         return [
-            artifacts.PublicArtifact(
-                self.client,
+            wandb.Artifact._from_attrs(
                 self.entity,
                 self.project,
                 self.collection_name + ":" + a["version"],
                 a["node"],
+                self.client,
             )
             for a in self.last_response["project"]["artifactType"][
                 "artifactCollection"
@@ -4221,7 +4221,7 @@ class ArtifactFiles(Paginator):
     def __init__(
         self,
         client: Client,
-        artifact: "artifacts.PublicArtifact",
+        artifact: "wandb.Artifact",
         names: Optional[Sequence[str]] = None,
         per_page: int = 50,
     ):
@@ -4335,9 +4335,7 @@ class Job:
     def _get_code_artifact(self, artifact_string):
         artifact_string, base_url, is_id = util.parse_artifact_string(artifact_string)
         if is_id:
-            code_artifact = artifacts.PublicArtifact.from_id(
-                artifact_string, self._api._client
-            )
+            code_artifact = wandb.Artifact._from_id(artifact_string, self._api._client)
         else:
             code_artifact = self._api.artifact(name=artifact_string, type="code")
         if code_artifact is None:
@@ -4412,7 +4410,7 @@ class Job:
         run_config = {}
         for key, item in config.items():
             if util._is_artifact_object(item):
-                if isinstance(item, wandb.Artifact) and item.id is None:
+                if isinstance(item, wandb.Artifact) and item.is_draft():
                     raise ValueError("Cannot queue jobs with unlogged artifacts")
                 run_config[key] = util.artifact_to_json(item)
 

{wandb-0.15.4 → wandb-0.15.5}/wandb/beta/workflows.py

@@ -5,12 +5,12 @@ from typing import Any, Dict, List, Optional, Union
 import wandb
 import wandb.data_types as data_types
 from wandb.data_types import _SavedModel
-from wandb.sdk.artifacts.artifact import Artifact as ArtifactInterface
+from wandb.sdk.artifacts.artifact import Artifact
 from wandb.sdk.artifacts.artifact_manifest_entry import ArtifactManifestEntry
 
 
 def _add_any(
-    artifact: ArtifactInterface,
+    artifact: Artifact,
     path_or_obj: Union[
         str, ArtifactManifestEntry, data_types.WBValue
     ],  # todo: add dataframe
@@ -23,8 +23,7 @@ def _add_any(
     be moved to the Artifact class in the future.
 
     Args:
-        artifact: `ArtifactInterface` - most likely a LocalArtifact created with
-            `wandb.Artifact(...)`
+        artifact: `Artifact` - artifact created with `wandb.Artifact(...)`
         path_or_obj: `Union[str, ArtifactManifestEntry, data_types.WBValue]` - either a
             str or valid object which indicates what to add to an artifact.
 
@@ -62,7 +61,7 @@ def _log_artifact_version(
     project: Optional[str] = None,
     scope_project: Optional[bool] = None,
     job_type: str = "auto",
-) -> ArtifactInterface:
+) -> Artifact:
     """Create an artifact, populate it, and log it with a run.
 
     If a run is not present, we create one.
@@ -83,7 +82,7 @@ def _log_artifact_version(
             Used to identify runs of a certain job type, i.e "evaluation".
 
     Returns:
-        ArtifactInterface
+        Artifact
 
     """
     if wandb.run is None:

{wandb-0.15.4 → wandb-0.15.5}/wandb/cli/cli.py

@@ -949,6 +949,9 @@ def launch_sweep(
         wandb.termerror("A project must be configured when using launch")
         return
 
+    # get personal username, not team name or service account, default to entity
+    author = api.viewer().get("username") or entity
+
     # if not sweep_config XOR resume_id
     if not (config or resume_id):
         wandb.termerror("'config' and/or 'resume_id' required")
@@ -993,40 +996,37 @@ def launch_sweep(
         if not found:
             wandb.termerror(f"Could not find sweep {entity}/{project}/{resume_id}")
             return
+
+        if found.get("state") == "RUNNING":
+            wandb.termerror(
+                f"Cannot resume sweep {entity}/{project}/{resume_id}, it is already running"
+            )
+            return
+
         sweep_obj_id = found["id"]
         sweep_config = yaml.safe_load(found["config"])
         wandb.termlog(f"Resuming from existing sweep {entity}/{project}/{resume_id}")
         if len(parsed_user_config.keys()) > 0:
             wandb.termwarn(
-                "Sweep params loaded from resumed sweep, ignoring provided config"
+                "Sweep parameters loaded from resumed sweep, ignoring provided config"
             )
 
-        # Is the resuming scheduler executing from a job? grab it to confirm
-        # the inputted scheduler job matches the resumed scheduler job
-        raw_scheduler_args = found.get("scheduler") or "{}"
-        prev_sweep_scheduler = json.loads(raw_scheduler_args)
-        prev_sweep_run_spec = json.loads(prev_sweep_scheduler.get("run_spec", "{}"))
-        if prev_sweep_run_spec.get("job"):
-            if scheduler_args and not scheduler_job:
-                wandb.termwarn(
-                    f"Resuming a launch sweep that was previously run from a scheduler job, but no scheduler job found in config. Defaulting to previous scheduler job: {prev_sweep_run_spec['job']}"
-                )
-                scheduler_job = prev_sweep_run_spec["job"]
-
-            if scheduler_job and scheduler_job != prev_sweep_run_spec["job"]:
-                wandb.termerror(
-                    f"Resuming a launch sweep with a different scheduler job is not supported. Loaded from sweep: {prev_sweep_run_spec['job']}, Provided in config: {scheduler_job}"
-                )
-                return False
-
-        # grab the queue from previously run scheduler if not specified
-        if not queue and prev_sweep_run_spec.get("overrides", {}).get(
-            "run_config", {}
-        ).get("sweep_args", {}).get("queue"):
-            queue = prev_sweep_run_spec["overrides"]["run_config"]["sweep_args"][
-                "queue"
-            ]
+        prev_scheduler = json.loads(found.get("scheduler") or "{}")
+        run_spec = json.loads(prev_scheduler.get("run_spec", "{}"))
+        if (
+            scheduler_job
+            and run_spec.get("job")
+            and run_spec.get("job") != scheduler_job
+        ):
+            wandb.termerror(
+                f"Resuming a launch sweep with a different scheduler job is not supported. Job loaded from sweep: {run_spec.get('job')}, job in config: {scheduler_job}"
+            )
+            return
 
+        prev_scheduler_args, prev_settings = sweep_utils.get_previous_args(run_spec)
+        # Passed in scheduler_args and settings override previous
+        scheduler_args.update(prev_scheduler_args)
+        settings.update(prev_settings)
     if not queue:
         wandb.termerror(
             "Launch-sweeps require setting a 'queue', use --queue option or a 'queue' key in the 'launch' section in the config"
@@ -1039,7 +1039,7 @@ def launch_sweep(
         sweep_config=sweep_config,
         queue=queue,
         project=project,
-        author=entity,
+        author=author,
     )
     if not args:
         return

{wandb-0.15.4 → wandb-0.15.5}/wandb/data_types.py

@@ -47,6 +47,7 @@ from .sdk.data_types.molecule import Molecule
 from .sdk.data_types.object_3d import Object3D
 from .sdk.data_types.plotly import Plotly
 from .sdk.data_types.saved_model import _SavedModel
+from .sdk.data_types.trace_tree import WBTraceTree
 from .sdk.data_types.video import Video
 from .sdk.lib import runid
 
@@ -67,6 +68,7 @@ __all__ = [
     "Object3D",
     "Plotly",
     "Video",
+    "WBTraceTree",
     "_SavedModel",
     # Typed Legacy Exports (I'd like to remove these)
     "ImageMask",

wandb-0.15.5/wandb/integration/langchain/wandb_tracer.py

@@ -0,0 +1,47 @@
+"""This module contains an integration with the LangChain library.
+
+Specifically, it exposes a `WandbTracer` class that can be used to stream
+LangChain activity to W&B. The intended usage pattern is to call
+`tracer = WandbTracer()` at the top of the script/notebook, and call
+`tracer.finish()` at the end of the script/notebook.
+ This will stream all LangChain activity to W&B.
+
+Technical Note:
+LangChain is in very rapid development - meaning their APIs and schemas are actively changing.
+As a matter of precaution, any call to LangChain apis, or use of their returned data is wrapped
+in a try/except block. This is to ensure that if a breaking change is introduced, the W&B
+integration will not break user code. The one exception to the rule is at import time. If
+LangChain is not installed, or the symbols are not in the same place, the appropriate error
+will be raised when importing this module.
+"""
+from packaging import version
+
+import wandb.util
+from wandb.sdk.lib import deprecate
+
+langchain = wandb.util.get_module(
+    name="langchain",
+    required="To use the LangChain WandbTracer you need to have the `langchain` python "
+    "package installed. Please install it with `pip install langchain`.",
+)
+
+if version.parse(langchain.__version__) < version.parse("0.0.188"):
+    raise ValueError(
+        "The Weights & Biases Langchain integration does not support versions 0.0.187 and lower. "
+        "To ensure proper functionality, please use version 0.0.188 or higher."
+    )
+
+# isort: off
+from langchain.callbacks.tracers import WandbTracer  # noqa: E402, I001
+
+
+class WandbTracer(WandbTracer):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        deprecate.deprecate(
+            field_name=deprecate.Deprecated.langchain_tracer,
+            warning_message="This feature is deprecated and has been moved to `langchain`. Enable tracing by setting "
+            "LANGCHAIN_WANDB_TRACING=true in your environment. See the documentation at "
+            "https://python.langchain.com/docs/ecosystem/integrations/agent_with_wandb_tracing for guidance. "
+            "Replace your current import with `from langchain.callbacks.tracers import WandbTracer`.",
+        )

{wandb-0.15.4 → wandb-0.15.5}/wandb/integration/sagemaker/config.py

@@ -20,9 +20,9 @@ def parse_sm_config() -> Dict[str, Any]:
         # Hyperparameter searches quote configs...
         for k, v in json.load(open(sm_files.SM_PARAM_CONFIG)).items():
             cast = v.strip('"')
-            if re.match(r"^[-\d]+$", cast):
+            if re.match(r"^-?[\d]+$", cast):
                 cast = int(cast)
-            elif re.match(r"^[-.\d]+$", cast):
+            elif re.match(r"^-?[.\d]+$", cast):
                 cast = float(cast)
             conf[k] = cast
     return conf

{wandb-0.15.4 → wandb-0.15.5}/wandb/integration/tensorboard/log.py

@@ -9,7 +9,7 @@ from wandb.sdk.lib import telemetry
 from wandb.viz import custom_chart
 
 if TYPE_CHECKING:
-    import numpy as np  # type: ignore
+    import numpy as np
 
     from wandb.sdk.internal.tb_watcher import TBHistory
 
@@ -42,7 +42,7 @@ def make_ndarray(tensor: Any) -> Optional["np.ndarray"]:
         if res.dtype == "object":
             return None
         else:
-            return res
+            return res  # type: ignore
     else:
         wandb.termwarn(
             "Can't convert tensor summary, upgrade tensorboard with `pip"
@@ -167,7 +167,7 @@ def tf_summary_to_dict(  # noqa: C901
                     try:
                         # TODO: we should just re-bin if there are too many buckets
                         values[namespaced_tag(value.tag, namespace)] = wandb.Histogram(
-                            np_histogram=(counts, bins)
+                            np_histogram=(counts, bins)  # type: ignore
                         )
                     except ValueError:
                         wandb.termwarn(
@@ -238,7 +238,7 @@ def tf_summary_to_dict(  # noqa: C901
                 )
                 try:
                     # TODO: we should just re-bin if there are too many buckets
-                    values[tag] = wandb.Histogram(np_histogram=np_histogram)
+                    values[tag] = wandb.Histogram(np_histogram=np_histogram)  # type: ignore
                 except ValueError:
                     wandb.termwarn(
                         f"Not logging key {tag!r}. "