wandb 0.19.11__py3-none-any.whl → 0.19.12rc1__py3-none-any.whl

wandb/__init__.py

@@ -10,7 +10,7 @@ For reference documentation, see https://docs.wandb.com/ref/python.
 """
 from __future__ import annotations
 
-__version__ = "0.19.11"
+__version__ = "0.19.12rc1"
 
 
 from wandb.errors import Error

wandb/__init__.pyi

@@ -106,7 +106,7 @@ if TYPE_CHECKING:
     import wandb
     from wandb.plot import CustomChart
 
-__version__: str = "0.19.11"
+__version__: str = "0.19.12rc1"
 
 run: Run | None
 config: wandb_config.Config

wandb/sdk/lib/service_connection.py

@@ -18,10 +18,6 @@ from wandb.sdk.mailbox import HandleAbandonedError, Mailbox, MailboxClosedError
 from wandb.sdk.service import service
 
 
-class WandbServiceNotOwnedError(Exception):
-    """Raised when the current process does not own the service process."""
-
-
 class WandbServiceConnectionError(Exception):
     """Raised on failure to connect to the service process."""
 
@@ -196,35 +192,35 @@ class ServiceConnection:
         request._info.stream_id = run_id
         self._client.send_server_request(spb.ServerRequest(inform_start=request))
 
-    def teardown(self, exit_code: int) -> int:
-        """Shuts down the service process and returns its exit code.
+    def teardown(self, exit_code: int) -> int | None:
+        """Close the connection.
+
+        Stop reading responses on the connection, and if this connection owns
+        the service process, send a teardown message and wait for it to shut
+        down.
 
         This may only be called once.
 
         Returns:
-            The exit code of the service process.
-
-        Raises:
-            WandbServiceNotOwnedError: If the current process did not start
-                the service process.
+            The exit code of the service process, or None if the process was
+            not owned by this connection.
         """
-        if not self._proc:
-            raise WandbServiceNotOwnedError(
-                "Cannot tear down service started by different process",
-            )
-
-        assert not self._torn_down
+        if self._torn_down:
+            raise AssertionError("Already torn down.")
         self._torn_down = True
 
         if self._cleanup:
             self._cleanup()
 
-        # Clear the service token to prevent new connections from being made.
-        service_token.clear_service_token()
-
         # Stop reading responses on the socket.
         self._router.join()
 
+        if not self._proc:
+            return None
+
+        # Clear the service token to prevent new connections to the process.
+        service_token.clear_service_token()
+
         self._client.send_server_request(
             spb.ServerRequest(
                 inform_teardown=spb.ServerInformTeardownRequest(

wandb/sdk/wandb_setup.py

@@ -307,12 +307,12 @@ class _WandbSetup:
         if not self._connection:
             return
 
-        internal_exit_code = self._connection.teardown(exit_code or 0)
-
         # Reset to None so that setup() creates a new connection.
+        connection = self._connection
         self._connection = None
 
-        if internal_exit_code != 0:
+        internal_exit_code = connection.teardown(exit_code or 0)
+        if internal_exit_code not in (None, 0):
             sys.exit(internal_exit_code)
 
     def ensure_service(self) -> ServiceConnection:

{wandb-0.19.11.dist-info → wandb-0.19.12rc1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wandb
-Version: 0.19.11
+Version: 0.19.12rc1
 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

{wandb-0.19.11.dist-info → wandb-0.19.12rc1.dist-info}/RECORD

@@ -1,6 +1,6 @@
 package_readme.md,sha256=U9047nyMDICgctm1HLm4HfXwFnFKsEn2m77hsYPUZ1I,4298
-wandb/__init__.py,sha256=F1oW4Oflh9R5-OZ5Xj2y-pxEjIrlC33lsCf-_zYhdEQ,7148
-wandb/__init__.pyi,sha256=3VFHLCEDLBaqfL6m8E67cewPBXc0pptuEGqBKj9doD4,48325
+wandb/__init__.py,sha256=h-BqHsfabAKDyqLt8M3IOEPcElSA0IXpkhPfdRpM7nk,7151
+wandb/__init__.pyi,sha256=4ngOvMcsz4c6J-4lu7LOE1R6WfMPW15H1ywVc2UhQpI,48328
 wandb/__main__.py,sha256=gripuDgB7J8wMMeJt4CIBRjn1BMSFr5zvsrt585Pnj4,64
 wandb/_iterutils.py,sha256=bOhOrOENEDzOMtyQqHysivzHri9axcVlCfXt4Ai1ot8,1545
 wandb/data_types.py,sha256=M-wqAO0FtIqvj28556u3h4nzSwlRcbxhXoX0B9jlJSo,2283
@@ -253,7 +253,7 @@ wandb/sdk/wandb_require.py,sha256=eYDG4oH1pwik2tHJVEdvivo4R2RypoITFomgr7umMhc,31
 wandb/sdk/wandb_require_helpers.py,sha256=ZmKv5aXXHDTTU6nYHMLKW4_pt9X-PlaMtbRJl77kHX8,1331
 wandb/sdk/wandb_run.py,sha256=mIPxXWDc8Xdi7yLYpLGt7BvqLvFfsYoLpfXe1VfYFO0,158032
 wandb/sdk/wandb_settings.py,sha256=y3T8UeSdNBqrKVOlmAXpwCREXhf1aZ-cWsFpQGooyuk,64830
-wandb/sdk/wandb_setup.py,sha256=3dvnkFlSuoa8BxzyKp_7qtj2tC2L26ge-rWolKQzfgY,15197
+wandb/sdk/wandb_setup.py,sha256=Yk9GlIik9AAChofQUvtkivDnlCiqiHM9-K1eJ03OWtw,15240
 wandb/sdk/wandb_summary.py,sha256=yQdOVIPrZaZanhBQ7yuSfPLX0x6dxwkN_KAn4SgjSZU,4536
 wandb/sdk/wandb_sweep.py,sha256=Sg_JqxVzmjUBvii41azpdr-c6RPwHOBnSha8k7jrRhk,4028
 wandb/sdk/wandb_sync.py,sha256=GNC8xYqT56ws4aw6Vd_mV_Ck-InEbqnyioSqHIybDug,2106
@@ -487,7 +487,7 @@ wandb/sdk/lib/retry.py,sha256=BAgPgJb1SxRguor2KTfOFnQpTlyNE4M_X8VcEc7YfMs,10173
 wandb/sdk/lib/run_moment.py,sha256=jMkfNAHOrYHlb6_nisqjfMu2Onm1bDy6lIubOJ2WxCM,2421
 wandb/sdk/lib/runid.py,sha256=Qa-5ft4B85YUazkV_18OYwf9JhMaAVp0JAInZzxzX5o,392
 wandb/sdk/lib/server.py,sha256=dn0RXjAnpmigAULLyq75XHYyHxw17MLCJ9fhPq2v22I,1623
-wandb/sdk/lib/service_connection.py,sha256=pf1T0isbDwTkv9MvuIpuk1MHTVkCebK4N8n9pXF4SSg,7455
+wandb/sdk/lib/service_connection.py,sha256=J6WHS95CdloYT-C_KgkuQa9pJBXkp23mobhAG1BC9JQ,7352
 wandb/sdk/lib/service_token.py,sha256=c4olk_g4Suo0zTTDaoR78oHLwpt3NgDzQeF94raigME,2471
 wandb/sdk/lib/sock_client.py,sha256=YzR4uFMgFBA95ySl9h2QwIqXKMEAyb36Y09vpf70IPU,8383
 wandb/sdk/lib/sparkline.py,sha256=9xQkuZ0z1DM7rHE2jFNDy5vBdPirnurd__A_bC6-Bgc,1363
@@ -904,8 +904,8 @@ wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/importlib2.py,sha256=cJIaJ2EQso
 wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/platform.py,sha256=UORYTNVcUSE2NpFcq9UVLIS-tsS0TS_Qw8akhKxn2eY,1506
 wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/unicode_paths.py,sha256=UWX8DB97ygkEeSxWQUYCHR4MahNilux7vl5TCTQtPPk,2190
 wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/win32stat.py,sha256=ZOevOTbSo8NRiIxkuBVGaG4yigWnPoO0goxAi-jsBkM,3828
-wandb-0.19.11.dist-info/METADATA,sha256=gVmAK68XIW9ZWlg4--FcNF2qbJIq8fZzsQ_NjV2RWSU,10259
-wandb-0.19.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-wandb-0.19.11.dist-info/entry_points.txt,sha256=v4FCOZ9gW7Pc6KLsmgQqpCiKTrA1wh2XHmNf-NUP1-I,67
-wandb-0.19.11.dist-info/licenses/LICENSE,sha256=izOKRJpGOx1PrJiGOKR0HsNdlB5JdH2d0Z4P7a7ssTc,1081
-wandb-0.19.11.dist-info/RECORD,,
+wandb-0.19.12rc1.dist-info/METADATA,sha256=vMgGUYCKJEfrEq3d_3Eo0rpiBIXYwdatPwvHavX4ot8,10262
+wandb-0.19.12rc1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+wandb-0.19.12rc1.dist-info/entry_points.txt,sha256=v4FCOZ9gW7Pc6KLsmgQqpCiKTrA1wh2XHmNf-NUP1-I,67
+wandb-0.19.12rc1.dist-info/licenses/LICENSE,sha256=izOKRJpGOx1PrJiGOKR0HsNdlB5JdH2d0Z4P7a7ssTc,1081
+wandb-0.19.12rc1.dist-info/RECORD,,