wandb 0.15.3__tar.gz → 0.15.5__tar.gz

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

@@ -1,11 +1,12 @@
 Metadata-Version: 2.1
 Name: wandb
-Version: 0.15.3
+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
 Author-email: support@wandb.com
 License: MIT license
+Platform: UNKNOWN
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
@@ -26,7 +27,6 @@ Provides-Extra: kubeflow
 Provides-Extra: gcp
 Provides-Extra: aws
 Provides-Extra: azure
-Provides-Extra: service
 Provides-Extra: grpc
 Provides-Extra: media
 Provides-Extra: sweeps
@@ -130,3 +130,5 @@ Got questions, feedback or want to join a community of ML engineers working on e
 <a href="https://bit.ly/wb-slack"><img src="https://svgshare.com/i/M93.svg" alt="slack" width="55"/></a> Join our [slack](https://bit.ly/wb-slack) community.
 
 [![Twitter](https://img.shields.io/twitter/follow/weights_biases?style=social)](https://twitter.com/weights_biases) Follow us on [Twitter](https://twitter.com/weights_biases).
+
+

{wandb-0.15.3 → 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.3 → 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.3"
+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.3 → 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.3 → wandb-0.15.5}/setup.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.15.3
+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.3 → wandb-0.15.5}/setup.py

@@ -12,14 +12,10 @@ with open("requirements.txt") as requirements_file:
 with open("requirements_sweeps.txt") as sweeps_requirements_file:
     sweeps_requirements = sweeps_requirements_file.read().splitlines()
 
-
-test_requirements = ["mock>=2.0.0", "tox-pyenv>=1.0.3"]
-
 gcp_requirements = ["google-cloud-storage"]
 aws_requirements = ["boto3"]
 azure_requirements = ["azure-identity", "azure-storage-blob"]
 grpc_requirements = ["grpcio>=1.27.2"]
-service_requirements = []
 kubeflow_requirements = ["kubernetes", "minio", "google-cloud-storage", "sh"]
 media_requirements = [
     "numpy",
@@ -32,18 +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"]
@@ -55,7 +55,7 @@ async_requirements = [
 
 setup(
     name="wandb",
-    version="0.15.3",
+    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",
@@ -75,7 +75,6 @@ setup(
     install_requires=requirements,
     license="MIT license",
     zip_safe=False,
-    # keywords='wandb',
     python_requires=">=3.6",
     classifiers=[
         "Development Status :: 5 - Production/Stable",
@@ -93,14 +92,11 @@ setup(
         "Topic :: System :: Logging",
         "Topic :: System :: Monitoring",
     ],
-    test_suite="tests",
-    tests_require=test_requirements,
     extras_require={
         "kubeflow": kubeflow_requirements,
         "gcp": gcp_requirements,
         "aws": aws_requirements,
         "azure": azure_requirements,
-        "service": service_requirements,
         "grpc": grpc_requirements,
         "media": media_requirements,
         "sweeps": sweeps_requirements,
@@ -109,9 +105,3 @@ setup(
         "async": async_requirements,
     },
 )
-
-# if os.name == "nt" and sys.version_info >= (3, 6):
-#     legacy_env_var = "PYTHONLEGACYWINDOWSSTDIO"
-#     if legacy_env_var not in os.environ:
-#         if os.system("setx " + legacy_env_var + " 1") != 0:
-#             raise Exception("Error setting environment variable " + legacy_env_var)

{wandb-0.15.3 → 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.3"
+__version__ = "0.15.5"
 
 # Used with pypi checks and other messages related to pip
 _wandb_module = "wandb"

{wandb-0.15.3 → 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.3 → wandb-0.15.5}/wandb/apis/importers/base.py

@@ -1,5 +1,6 @@
 import json
 import platform
+import re
 from abc import ABC, abstractmethod
 from concurrent.futures import ProcessPoolExecutor, as_completed
 from contextlib import contextmanager
@@ -193,7 +194,8 @@ class ImporterRun:
         )
 
     def _make_artifact_record(self) -> pb.Record:
-        art = wandb.Artifact(self.display_name(), "imported-artifacts")
+        artifact_name = self._handle_incompatible_strings(self.display_name())
+        art = wandb.Artifact(artifact_name, "imported-artifacts")
         artifacts = self.artifacts()
         if artifacts is not None:
             for name, path in artifacts:
@@ -260,6 +262,13 @@ class ImporterRun:
         with open(f"{run_dir}/files/wandb-metadata.json", "w") as f:
             f.write(json.dumps(d))
 
+    @staticmethod
+    def _handle_incompatible_strings(s):
+        valid_chars = r"[^a-zA-Z0-9_\-\.]"
+        replacement = "__"
+
+        return re.sub(valid_chars, replacement, s)
+
 
 class Importer(ABC):
     @abstractmethod
@@ -282,10 +291,16 @@ class Importer(ABC):
                 }
                 for future in as_completed(futures):
                     run = futures[future]
-                    pbar.update(1)
-                    pbar.set_description(
-                        f"Imported Run: {run.run_group()} {run.display_name()}"
-                    )
+                    try:
+                        future.result()
+                    except Exception as exc:
+                        wandb.termerror(f"Failed to import {run.display_name()}: {exc}")
+                    else:
+                        pbar.set_description(
+                            f"Imported Run: {run.run_group()} {run.display_name()}"
+                        )
+                    finally:
+                        pbar.update(1)
 
     def import_one(
         self,

{wandb-0.15.3 → wandb-0.15.5}/wandb/apis/importers/mlflow.py

@@ -75,7 +75,13 @@ class MlflowRun(ImporterRun):
         return self.run.info.start_time // 1000
 
     def runtime(self):
-        return self.run.info.end_time // 1_000 - self.start_time()
+        end_time = (
+            self.run.info.end_time // 1000
+            if self.run.info.end_time is not None
+            else self.start_time()
+        )
+
+        return end_time - self.start_time()
 
     def git(self):
         ...

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

@@ -91,6 +91,9 @@ class Api:
             project, id=id, description=description, entity=entity
         )
 
+    def upsert_run(self, *args, **kwargs):
+        return self.api.upsert_run(*args, **kwargs)
+
     def settings(self, *args, **kwargs):
         return self.api.settings(*args, **kwargs)
 
@@ -155,12 +158,18 @@ class Api:
     def create_artifact(self, *args, **kwargs):
         return self.api.create_artifact(*args, **kwargs)
 
+    def complete_multipart_upload_artifact(self, *args, **kwargs):
+        return self.api.complete_multipart_upload_artifact(*args, **kwargs)
+
     def run_config(self, *args, **kwargs):
         return self.api.run_config(*args, **kwargs)
 
     def upload_file_retry(self, *args, **kwargs):
         return self.api.upload_file_retry(*args, **kwargs)
 
+    def upload_multipart_file_chunk_retry(self, *args, **kwargs):
+        return self.api.upload_multipart_file_chunk_retry(*args, **kwargs)
+
     async def upload_file_retry_async(self, *args, **kwargs):
         return await self.api.upload_file_retry_async(*args, **kwargs)
 
@@ -200,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)