wandb 0.15.3__py3-none-any.whl → 0.15.5__py3-none-any.whl

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/analytics/sentry.py

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

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/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/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)