wandb 0.17.1__py3-none-manylinux_2_17_aarch64.manylinux2014_aarch64.whl → 0.17.2__py3-none-manylinux_2_17_aarch64.manylinux2014_aarch64.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.17.1"
+__version__ = "0.17.2"
 
 
 # Used with pypi checks and other messages related to pip
@@ -214,9 +214,12 @@ from .analytics import Sentry as _Sentry
 if "dev" in __version__:
     import os
 
+    # disable error reporting in dev versions for the python client
     os.environ["WANDB_ERROR_REPORTING"] = os.environ.get(
         "WANDB_ERROR_REPORTING", "false"
     )
+    # turn on wandb-core for dev versions
+    os.environ["WANDB__REQUIRE_CORE"] = os.environ.get("WANDB__REQUIRE_CORE", "true")
 
 _sentry = _Sentry()
 _sentry.setup()

wandb/cli/cli.py

@@ -880,6 +880,14 @@ def sync(
     default=False,
     help="Resume a sweep to continue running new runs.",
 )
+@click.option(
+    "--prior_run",
+    "-R",
+    "prior_runs",
+    multiple=True,
+    default=None,
+    help="ID of an existing run to add to this sweep",
+)
 @click.argument("config_yaml_or_sweep_id")
 @click.pass_context
 @display_error
@@ -897,6 +905,7 @@ def sweep(
     cancel,
     pause,
     resume,
+    prior_runs,
     config_yaml_or_sweep_id,
 ):
     state_args = "stop", "cancel", "pause", "resume"
@@ -1038,6 +1047,7 @@ def sweep(
         project=project,
         entity=entity,
         obj_id=sweep_obj_id,
+        prior_runs=prior_runs,
     )
     sweep_utils.handle_sweep_config_violations(warnings)
 
@@ -1104,6 +1114,14 @@ def sweep(
     default=None,
     help="Resume a launch sweep by passing an 8-char sweep id. Queue required",
 )
+@click.option(
+    "--prior_run",
+    "-R",
+    "prior_runs",
+    multiple=True,
+    default=None,
+    help="ID of an existing run to add to this sweep",
+)
 @click.argument("config", required=False, type=click.Path(exists=True))
 @click.pass_context
 @display_error
@@ -1114,6 +1132,7 @@ def launch_sweep(
     queue,
     config,
     resume_id,
+    prior_runs,
 ):
     api = _get_cling_api()
     env = os.environ
@@ -1298,6 +1317,7 @@ def launch_sweep(
         obj_id=sweep_obj_id,  # if resuming
         launch_scheduler=launch_scheduler_with_queue,
         state="PENDING",
+        prior_runs=prior_runs,
     )
     sweep_utils.handle_sweep_config_violations(warnings)
     # Log nicely formatted sweep information

wandb/old/summary.py

@@ -390,6 +390,11 @@ class HTTPSummary(Summary):
         self._client = client
         self._started = time.time()
 
+    def __delitem__(self, key):
+        if key not in self._json_dict:
+            raise KeyError(key)
+        del self._json_dict[key]
+
     def load(self):
         pass
 

wandb/sdk/artifacts/artifact.py

@@ -2019,16 +2019,23 @@ class Artifact:
     def delete(self, delete_aliases: bool = False) -> None:
         """Delete an artifact and its files.
 
+        If called on a linked artifact (i.e. a member of a portfolio collection): only the link is deleted, and the
+        source artifact is unaffected.
+
         Arguments:
             delete_aliases: If set to `True`, deletes all aliases associated with the artifact.
                 Otherwise, this raises an exception if the artifact has existing
                 aliases.
+                This parameter is ignored if the artifact is linked (i.e. a member of a portfolio collection).
 
         Raises:
             ArtifactNotLoggedError: If the artifact is not logged.
         """
         self._ensure_logged("delete")
-        self._delete(delete_aliases)
+        if self.collection.is_sequence():
+            self._delete(delete_aliases)
+        else:
+            self._unlink()
 
     @normalize_exceptions
     def _delete(self, delete_aliases: bool = False) -> None:

wandb/sdk/data_types/_dtypes.py

@@ -390,10 +390,13 @@ if np:
     NumberType.types.append(np.uintp)
     NumberType.types.append(np.float32)
     NumberType.types.append(np.float64)
-    NumberType.types.append(np.float_)
     NumberType.types.append(np.complex64)
     NumberType.types.append(np.complex128)
-    NumberType.types.append(np.complex_)
+
+    numpy_major_version = np.__version__.split(".")[0]
+    if int(numpy_major_version) < 2:
+        NumberType.types.append(np.float_)
+        NumberType.types.append(np.complex_)
 
 
 class TimestampType(Type):

wandb/sdk/data_types/html.py

@@ -33,7 +33,7 @@ class Html(BatchableMedia):
         if data_is_path:
             assert isinstance(data, str)
             data_path = data
-            with open(data_path) as file:
+            with open(data_path, encoding="utf-8") as file:
                 self.html = file.read()
         elif isinstance(data, str):
             self.html = data

wandb/sdk/data_types/object_3d.py

@@ -36,7 +36,7 @@ if TYPE_CHECKING:  # pragma: no cover
 
     from ..wandb_run import Run as LocalRun
 
-    numeric = Union[int, float, np.integer, np.float_]
+    numeric = Union[int, float, np.integer, np.float64]
     FileFormat3D = Literal[
         "obj",
         "gltf",

wandb/sdk/internal/internal_api.py

@@ -1180,6 +1180,7 @@ class Api:
             project_name (str): The project to download, (can include bucket)
             name (str): The run to download
         """
+        # Pulling wandbConfig.start_time is required so that we can determine if a run has actually started
         query = gql(
             """
         query RunResumeStatus($project: String, $entity: String, $name: String!) {
@@ -1203,6 +1204,7 @@ class Api:
                     eventsTail
                     config
                     tags
+                    wandbConfig(keys: ["t"])
                 }
             }
         }
@@ -3036,6 +3038,7 @@ class Api:
         project: Optional[str] = None,
         entity: Optional[str] = None,
         state: Optional[str] = None,
+        prior_runs: Optional[List[str]] = None,
     ) -> Tuple[str, List[str]]:
         """Upsert a sweep object.
 
@@ -3048,6 +3051,7 @@ class Api:
             project (str): project to use
             entity (str): entity to use
             state (str): state
+            prior_runs (list): IDs of existing runs to add to the sweep
         """
         project_query = """
             project {
@@ -3068,7 +3072,8 @@ class Api:
             $projectName: String,
             $controller: JSONString,
             $scheduler: JSONString,
-            $state: String
+            $state: String,
+            $priorRunsFilters: JSONString,
         ) {
             upsertSweep(input: {
                 id: $id,
@@ -3078,7 +3083,8 @@ class Api:
                 projectName: $projectName,
                 controller: $controller,
                 scheduler: $scheduler,
-                state: $state
+                state: $state,
+                priorRunsFilters: $priorRunsFilters,
             }) {
                 sweep {
                     name
@@ -3127,6 +3133,9 @@ class Api:
         config_str = yaml.dump(
             json.loads(json.dumps(config)), Dumper=util.NonOctalStringDumper
         )
+        filters = None
+        if prior_runs:
+            filters = json.dumps({"$or": [{"name": r} for r in prior_runs]})
 
         err: Optional[Exception] = None
         for mutation in mutations:
@@ -3140,6 +3149,7 @@ class Api:
                     "controller": controller,
                     "launchScheduler": launch_scheduler,
                     "scheduler": scheduler,
+                    "priorRunsFilters": filters,
                 }
                 if state:
                     variables["state"] = state

wandb/sdk/internal/sender.py

@@ -754,14 +754,14 @@ class SendManager:
             project_name=run.project,
             name=run.run_id,
         )
-
-        if not resume_status:
+        # No resume status = run does not exist; No t key in wandbConfig = run exists but hasn't been inited
+        if not resume_status or '"t":' not in resume_status.get("wandbConfig", ""):
             if self._settings.resume == "must":
                 error = wandb_internal_pb2.ErrorInfo()
                 error.code = wandb_internal_pb2.ErrorInfo.ErrorCode.USAGE
                 error.message = (
                     "You provided an invalid value for the `resume` argument."
-                    f" The value 'must' is not a valid option for resuming a run ({run.run_id}) that does not exist."
+                    f" The value 'must' is not a valid option for resuming a run ({run.run_id}) that has not been initialized."
                     " Please check your inputs and try again with a valid run ID."
                     " If you are trying to start a new run, please omit the `resume` argument or use `resume='allow'`."
                 )

wandb/sdk/internal/system/assets/gpu.py

@@ -410,5 +410,7 @@ class GPU:
 
         except pynvml.NVMLError:
             pass
+        except Exception as e:
+            logger.error(f"Error Probing GPU: {e}")
 
         return info

wandb/sdk/launch/_project_spec.py

@@ -141,7 +141,7 @@ class LaunchProject:
         elif self.job is not None:
             self.source = LaunchSource.JOB
             self.project_dir = tempfile.mkdtemp()
-        if self.uri and self.uri.startswith("placeholder"):
+        elif self.uri and self.uri.startswith("placeholder"):
             self.source = LaunchSource.SCHEDULER
             self.project_dir = os.getcwd()
             self._entry_point = self.override_entrypoint

wandb/sdk/wandb_run.py

@@ -2614,11 +2614,6 @@ class Run:
         exit_handle = self._backend.interface.deliver_exit(self._exit_code)
         exit_handle.add_probe(on_probe=self._on_probe_exit)
 
-        # this message is confusing, we should remove it
-        # self._footer_exit_status_info(
-        #     self._exit_code, settings=self._settings, printer=self._printer
-        # )
-
         # wait for the exit to complete
         _ = exit_handle.wait(timeout=-1, on_progress=self._on_progress_exit)
 
@@ -3764,6 +3759,11 @@ class Run:
             settings=settings,
             printer=printer,
         )
+        Run._footer_notify_wandb_core(
+            quiet=quiet,
+            settings=settings,
+            printer=printer,
+        )
         Run._footer_local_warn(
             server_info_response=server_info_response,
             quiet=quiet,
@@ -3786,26 +3786,6 @@ class Run:
             printer=printer,
         )
 
-    @staticmethod
-    def _footer_exit_status_info(
-        exit_code: Optional[int],
-        *,
-        settings: "Settings",
-        printer: Union["PrinterTerm", "PrinterJupyter"],
-    ) -> None:
-        if settings.silent:
-            return
-
-        status = "(success)." if not exit_code else f"(failed {exit_code})."
-        info = [
-            f"Waiting for W&B process to finish... {printer.status(status, bool(exit_code))}"
-        ]
-
-        if not settings._offline and exit_code:
-            info.append(f"Press {printer.abort()} to abort syncing.")
-
-        printer.display(f'{" ".join(info)}')
-
     # fixme: Temporary hack until we move to rich which allows multiple spinners
     @staticmethod
     def _footer_file_pusher_status_info(
@@ -4154,6 +4134,24 @@ class Run:
         if package_problem and check_version.upgrade_message:
             printer.display(check_version.upgrade_message)
 
+    @staticmethod
+    def _footer_notify_wandb_core(
+        *,
+        quiet: Optional[bool] = None,
+        settings: "Settings",
+        printer: Union["PrinterTerm", "PrinterJupyter"],
+    ) -> None:
+        """Prints a message advertising the upcoming core release."""
+        if quiet or settings._require_core:
+            return
+
+        printer.display(
+            "The new W&B backend becomes opt-out in version 0.18.0;"
+            ' try it out with `wandb.require("core")`!'
+            " See https://wandb.me/wandb-core for more information.",
+            level="warn",
+        )
+
     @staticmethod
     def _footer_reporter_warn_err(
         reporter: Optional[Reporter] = None,

wandb/util.py

@@ -1,6 +1,7 @@
 import colorsys
 import contextlib
 import dataclasses
+import enum
 import functools
 import gzip
 import importlib
@@ -637,6 +638,8 @@ def json_friendly(  # noqa: C901
     elif isinstance(obj, set):
         # set is not json serializable, so we convert it to tuple
         obj = tuple(obj)
+    elif isinstance(obj, enum.Enum):
+        obj = obj.name
     else:
         converted = False
     if getsizeof(obj) > VALUE_BYTES_LIMIT:

{wandb-0.17.1.dist-info → wandb-0.17.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wandb
-Version: 0.17.1
+Version: 0.17.2
 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