wandb 0.19.6rc4__py3-none-macosx_11_0_arm64.whl → 0.19.8__py3-none-macosx_11_0_arm64.whl

wandb/sdk/wandb_sync.py

@@ -6,7 +6,6 @@ from wandb.errors.term import termerror, termlog
 
 from . import wandb_setup
 from .backend.backend import Backend
-from .lib.mailbox import Mailbox
 from .lib.runid import generate_id
 
 if TYPE_CHECKING:
@@ -49,21 +48,14 @@ def _sync(
     service = wl.ensure_service()
     service.inform_init(settings=settings, run_id=stream_id)
 
-    mailbox = Mailbox()
-    backend = Backend(
-        settings=wl.settings,
-        service=service,
-        mailbox=mailbox,
-    )
+    backend = Backend(settings=wl.settings, service=service)
     backend.ensure_launched()
 
     assert backend.interface
     backend.interface._stream_id = stream_id  # type: ignore
 
-    mailbox.enable_keepalive()
     handle = backend.interface.deliver_finish_sync()
-    result = handle.wait(timeout=-1)
-    assert result and result.response
+    result = handle.wait_or(timeout=None)
     response = result.response.sync_response
     if response.url:
         termlog(f"Synced {p} to {util.app_url(response.url)}")

wandb/util.py

@@ -788,6 +788,8 @@ class JSONEncoderUncompressed(json.JSONEncoder):
     def default(self, obj: Any) -> Any:
         if is_numpy_array(obj):
             return obj.tolist()
+        elif np and isinstance(obj, np.number):
+            return obj.item()
         elif np and isinstance(obj, np.generic):
             obj = obj.item()
         return json.JSONEncoder.default(self, obj)
@@ -872,7 +874,7 @@ def parse_backend_error_messages(response: requests.Response) -> List[str]:
     # Backend error values are returned in one of two ways:
     # - A string containing the error message
     # - A JSON object with a "message" field that is a string
-    def get_message(err: Any) -> Optional[str]:
+    def get_message(error: Any) -> Optional[str]:
         if isinstance(error, str):
             return error
         elif (

{wandb-0.19.6rc4.dist-info → wandb-0.19.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: wandb
-Version: 0.19.6rc4
+Version: 0.19.8
 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
@@ -110,7 +110,7 @@ Requires-Dist: typing-extensions; extra == 'launch'
 Provides-Extra: media
 Requires-Dist: bokeh; extra == 'media'
 Requires-Dist: imageio; extra == 'media'
-Requires-Dist: moviepy; extra == 'media'
+Requires-Dist: moviepy>=1.0.0; extra == 'media'
 Requires-Dist: numpy; extra == 'media'
 Requires-Dist: pillow; extra == 'media'
 Requires-Dist: plotly>=5.18.0; extra == 'media'

{wandb-0.19.6rc4.dist-info → wandb-0.19.8.dist-info}/RECORD

@@ -1,17 +1,17 @@
 package_readme.md,sha256=U9047nyMDICgctm1HLm4HfXwFnFKsEn2m77hsYPUZ1I,4298
-wandb-0.19.6rc4.dist-info/RECORD,,
-wandb-0.19.6rc4.dist-info/WHEEL,sha256=B_HTF0nESsbkeA2UeB8YgKrtPrxyKFORYURx7Qxu8Xw,101
-wandb-0.19.6rc4.dist-info/entry_points.txt,sha256=v4FCOZ9gW7Pc6KLsmgQqpCiKTrA1wh2XHmNf-NUP1-I,67
-wandb-0.19.6rc4.dist-info/METADATA,sha256=AgyaK7K_qLQ3C-FNzzP2cgKJ8pNsIax5PS3JaHY4aJE,10285
-wandb-0.19.6rc4.dist-info/licenses/LICENSE,sha256=izOKRJpGOx1PrJiGOKR0HsNdlB5JdH2d0Z4P7a7ssTc,1081
-wandb/env.py,sha256=ImVRezUi1wpXf2ogJ4IY9YNaELqon3k3S6-vCMNNzwQ,13505
-wandb/__init__.pyi,sha256=Rd4ARb64AK401vM56lDpn7qwfgKR7bsiFo7fRTSAv2A,46671
+wandb-0.19.8.dist-info/RECORD,,
+wandb-0.19.8.dist-info/WHEEL,sha256=B_HTF0nESsbkeA2UeB8YgKrtPrxyKFORYURx7Qxu8Xw,101
+wandb-0.19.8.dist-info/entry_points.txt,sha256=v4FCOZ9gW7Pc6KLsmgQqpCiKTrA1wh2XHmNf-NUP1-I,67
+wandb-0.19.8.dist-info/METADATA,sha256=qm19tpiMFdQkDIVM5-x0ERqDuSNC_pAOatXyuDVxJqE,10289
+wandb-0.19.8.dist-info/licenses/LICENSE,sha256=izOKRJpGOx1PrJiGOKR0HsNdlB5JdH2d0Z4P7a7ssTc,1081
+wandb/env.py,sha256=NtyTiZbU3xrpNiISvrTVO5SDBXVqHl58wmrhpB4MCSA,13736
+wandb/__init__.pyi,sha256=D74u5n7I6svyFW-yISHSee9DUiT_yK-73wbjtuqc1sY,48317
 wandb/_globals.py,sha256=CccwOAls5bxJArYHg12b08ZeKR8Qu9u57GtYWjBH0o0,702
-wandb/util.py,sha256=CoQPRhzGtCYp4QYPTyBy4AHTU5oqSL8QlZnUTDuIaGk,63445
+wandb/util.py,sha256=tlaNaz9W6r2T0bs0-KVDcuSPE1PiZ7642GjCjAx8Mvw,63525
 wandb/wandb_run.py,sha256=CNh9S6uubFk8FphQjzkbvedyyGCN9aBEsRBKjy8tqqs,155
 wandb/_iterutils.py,sha256=bOhOrOENEDzOMtyQqHysivzHri9axcVlCfXt4Ai1ot8,1545
-wandb/__init__.py,sha256=z9qa4RemyCngKb-KmUVUAG8XeG7s0lKgwOaEWkO3BoI,6991
-wandb/data_types.py,sha256=tjxcQ8padGuGxST192PyEDX_nhU__izHcAK-kaSyevI,2276
+wandb/__init__.py,sha256=_gHKjYc0NbTN_pwUJWnp4yA6uRDPpWfz93VliZmKZGw,6988
+wandb/data_types.py,sha256=M-wqAO0FtIqvj28556u3h4nzSwlRcbxhXoX0B9jlJSo,2283
 wandb/wandb_controller.py,sha256=SksJdgwn14PpnUoIaBjJ9Ki4Nksl9BpQGGn42hT0xZg,24936
 wandb/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wandb/wandb_agent.py,sha256=0TKjcJ8jxEqaXMrs_dIflXfWFFOaulfSjzp_fpYiXD0,20985
@@ -43,15 +43,23 @@ wandb/apis/public/query_generator.py,sha256=WfyaQuBcTHEECOJ2zN_0iqkUufki9MX28fj3
 wandb/apis/public/teams.py,sha256=YJAsJLRVM5r6UT48tdGGVA2V_f7nayDe5Q5NRTFQ8m4,5474
 wandb/apis/public/__init__.py,sha256=ldsy9V8UQlRnsIf9E4qRMtYIdH89oQ8eiYThQsphXks,1070
 wandb/apis/public/jobs.py,sha256=Ewwh-KK2mNoOXIz3p4Q1fMoWqAkGDl1vGbTBZIkOxPc,22372
-wandb/apis/public/api.py,sha256=jF_zmKLWY8p2uY_dhrOxSowLxkDoZZBGqG4S3nk4BZc,51748
-wandb/apis/public/utils.py,sha256=3VZ_oKACQ3wR2QG0x89hea4GOdKkjsP-cYm5qZt7gqc,2103
+wandb/apis/public/api.py,sha256=OGDy0Iw8cRQJYVmHDVuMb8idxoNbEZLITrgaGWo0-E0,58882
+wandb/apis/public/utils.py,sha256=6jfEut3u4qc1KiD4fR04YM2X6tmNW997MT44ptL7bTU,3681
 wandb/apis/public/reports.py,sha256=xUm3UO8OnCGXMMqXw-YUhOi143HccrDPS8VO7B0xW0s,15278
 wandb/apis/public/sweeps.py,sha256=FYWvA-h-xqCnZ8PYXRwOEvRHgl3BS0RhuZpTftIMNrc,6547
-wandb/apis/public/artifacts.py,sha256=ZO6vt4Qs2N8GerR8vM5Wv8Bx-mfG8tAA-FXHs5czDRk,33902
+wandb/apis/public/artifacts.py,sha256=rBzZJQsftgyImSu_r8abSjguqA0Q-SDQtkYZALJ1cYE,33366
 wandb/apis/public/projects.py,sha256=F5cPDxAbZD4-oVB_BxBCTsZk6k1tVL0cPU3Z0YEUqzo,4322
 wandb/apis/public/runs.py,sha256=_v9F-Onzx_LJha8xxWDxXk3BWEK1Gg0Yq9XJSxtlx_c,35202
+wandb/apis/public/registries.py,sha256=CaL1vD2dH74aMu916vVfgqSsKFsQWMVZf5jNfXiboa0,18404
 wandb/apis/public/const.py,sha256=icNtcS3gTCtvevLWuTOCqm0FHEfLQ0P80mA69dWeEXs,121
 wandb/apis/public/history.py,sha256=4gwe9HJ_NH9SSPtLtP7ELw4nIsxLPrY6ji13EK1siyM,4636
+wandb/apis/public/_generated/typing_compat.py,sha256=qV4yFJH1utKxh-vl-U3JhPHyRSqxZ8d8niMSBF5K370,303
+wandb/apis/public/_generated/enums.py,sha256=vPRi0TPBkZS4D4to-WzhesolsPj3a2B1Okr1xmjRy5k,124
+wandb/apis/public/_generated/input_types.py,sha256=vPRi0TPBkZS4D4to-WzhesolsPj3a2B1Okr1xmjRy5k,124
+wandb/apis/public/_generated/__init__.py,sha256=m19wAWXw_4qFC1pNaJPp1XBngTOjCab-4Eds3YmDT3U,528
+wandb/apis/public/_generated/operations.py,sha256=e_jeUmwP9mnaFe2WUssdmKw2AeN-Iuf3j3FkG9fDnUY,245
+wandb/apis/public/_generated/server_features_query.py,sha256=VcqMZTC-_Xq9nhVUv5TD_na6O4PzuDPA2YK6lREOQmM,641
+wandb/apis/public/_generated/base.py,sha256=zaD3C8aYH-WNw5pL1d-UarEm0NjmEW7lYF2eMraHFpk,4017
 wandb/apis/reports/__init__.py,sha256=5ZkKvOqwks3UmVeaPuwIZYwyV8_afsGbtybDOXYjSsg,32
 wandb/apis/reports/v1/__init__.py,sha256=L5l1BqMaap-w5wXCzn8pC8RXB7_72MsjTx8HeLuC3XU,263
 wandb/apis/reports/v2/__init__.py,sha256=jdLiTqf5QIRxnWH8drWV9Y06bZEhI5N6uGGSKAT7tv0,263
@@ -76,25 +84,25 @@ wandb/proto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wandb/proto/wandb_generate_proto.py,sha256=KO1hlAUBGHQRNKsddhcSXvh5a6rmFM3kshKTWftbWwY,1278
 wandb/proto/wandb_deprecated.py,sha256=CGD8fZII6QOjVPJZ2zCGG8Cx95A5KZVYHYwFPz1Cz6k,1978
 wandb/proto/wandb_telemetry_pb2.py,sha256=ReY9N2qSt46o7m1ou1khvYSYYImrzo5ro90g_9m8QsM,322
-wandb/proto/v5/wandb_settings_pb2.py,sha256=KKbLjQkNwX-9QHxgQHgSts8lwg3w7Eo999ytTsiYjc4,17853
-wandb/proto/v5/wandb_server_pb2.py,sha256=piGeVPCrHNCMMc9yIYKQtFRNSvfvB2fQKs8ANwApJu8,7463
+wandb/proto/v5/wandb_settings_pb2.py,sha256=qDaa79-Rm4A0-1BiDlCP4Tf69XhZyJA_Or5XpQa1keg,17976
+wandb/proto/v5/wandb_server_pb2.py,sha256=f55mk3Z77xUnq_tAbTd_Q75CiOJz6ppPbZT7CoeF7-I,7535
 wandb/proto/v5/wandb_base_pb2.py,sha256=u7VVWdExJ4WDkYNMV-xvWvyQ-NTIbAqToTKtgEqt_Lg,1572
-wandb/proto/v5/wandb_internal_pb2.py,sha256=JNGSa8ynettM2C6xFzo_q2A5Ej2olXtfryzKZcWuUKs,56810
-wandb/proto/v5/wandb_telemetry_pb2.py,sha256=tYmNIX5Mtw2AybnuX8n9KbwdxT2w4GKPvvokFfKfP8M,11522
-wandb/proto/v4/wandb_settings_pb2.py,sha256=O0U7x3aM6ih8Riiyv8CTIS4fUsMi3wSe8UkpL373myQ,17497
-wandb/proto/v4/wandb_server_pb2.py,sha256=MjXs9_8Gy8PKlIZuNZILp0fh2GDXctwKDL5BZhXydsA,6905
+wandb/proto/v5/wandb_internal_pb2.py,sha256=FGXDcOhWMVRkFTBF0uRSOi6YG6rJciJi0HuMlxbxmBI,57498
+wandb/proto/v5/wandb_telemetry_pb2.py,sha256=PCTcpV-BmgR-5qb2vA30rRhLv_V4A-8Rt3cEgczgjs8,11578
+wandb/proto/v4/wandb_settings_pb2.py,sha256=hRf0tgWk8NSpeM1677mXaZOahjaG8W78qTJrct_j3r8,17620
+wandb/proto/v4/wandb_server_pb2.py,sha256=NiG6Xv9CDvxc8v8NmnM237jyYqMuRyE_nqmLAMZVdI8,6977
 wandb/proto/v4/wandb_base_pb2.py,sha256=El5lnZMRpkc1W4SlZgeKoDPqWiJau0BG8QfigMMt6bM,1422
-wandb/proto/v4/wandb_internal_pb2.py,sha256=aFpVLemWazsb3XeV7z4qi9vOBc98FkoazbA6Lx6my7o,52511
+wandb/proto/v4/wandb_internal_pb2.py,sha256=LV8iKL7ufkIFyNjfG4PjDbg_u2EoGbYuMG7Qn5XaABw,53151
 wandb/proto/v4/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wandb/proto/v4/wandb_telemetry_pb2.py,sha256=sd9xk2SdYw3SBOYw77GT5pFa0YHl9v-P17-854RMMxg,11252
-wandb/proto/v3/wandb_settings_pb2.py,sha256=oaZH14fElK1x3Ze2i5A2VzU7YcagWF64MfkYA6Mj3yI,21122
-wandb/proto/v3/wandb_server_pb2.py,sha256=v_A6xNDUCKToBeUTgr1b6sqo51-PdqBEgLLKu8oloSU,15442
+wandb/proto/v4/wandb_telemetry_pb2.py,sha256=9ALCD8ZYH_Sr4Sw5uJ_gsPDOxwquxjkMMXZQD124Dq4,11308
+wandb/proto/v3/wandb_settings_pb2.py,sha256=Iiq2lnpfLiqk1su_dpD6SHH3zmlvOaU9kRxtyzMGMAo,21245
+wandb/proto/v3/wandb_server_pb2.py,sha256=LQCUmuy0NFDo20N9uCUiE5VPyhP5r929WKtniae4CdQ,15514
 wandb/proto/v3/wandb_base_pb2.py,sha256=_nsr_HW4Fdz62-KiXGo6Dw0_9bwdXz07auZkkH2QfSI,2355
-wandb/proto/v3/wandb_internal_pb2.py,sha256=-zsy_FpMea-pu-e8iGOSyoXt_aOWaJZ-wkb-oLk5u4E,114598
+wandb/proto/v3/wandb_internal_pb2.py,sha256=X5m7_mikGKgCRpD84q311n64q6-HMMFMhVop6XeGgHM,116112
 wandb/proto/v3/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wandb/proto/v3/wandb_telemetry_pb2.py,sha256=tQQJ7s3fbFYJc60JLnrkdf8yxCqKK0zJMdJaXxA3phM,13805
-wandb/bin/wandb-core,sha256=HfQsX4Zlt-V78vEh6LOmuOo-2v--d4f5qyMgxPCaR4g,47309922
-wandb/bin/gpu_stats,sha256=yad1MzqpoqGUmiUcr6o5MB7w08xcC9XFharEw0BIXTw,9441344
+wandb/proto/v3/wandb_telemetry_pb2.py,sha256=c3y5sNd6dc-POBGJRmqNych_7L7nd0Cdt7ILXuW7RPM,13861
+wandb/bin/wandb-core,sha256=bIPb-K45R0ogMa-lVxiANAruRCrXvnnngvZ4IHOzOnE,47308498
+wandb/bin/gpu_stats,sha256=TaQw4YTbAxSPTAPOtEE2ZXCKY5y1I_a8aXEd3PSjSRI,9461072
 wandb/integration/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wandb/integration/yolov8/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wandb/integration/yolov8/yolov8.py,sha256=srryantZqO65pj1x52MiKD5m2SzdQc4nIsj4o1GPK8o,11371
@@ -188,48 +196,46 @@ wandb/old/summary.py,sha256=eqpzQB5CfzTKZta6tB-nVUh0ZmiLwISqjSAt4yvwocE,13952
 wandb/old/settings.py,sha256=IrS10skC9gCTb8NgTWeRFgsTYL1bw36e-rG_xECk-yo,6374
 wandb/cli/beta.py,sha256=Um35Q9CeoQPS80g0wryo2WtSaD7VA0XqF_YCQNDNiSY,5438
 wandb/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wandb/cli/cli.py,sha256=LOeIp_FQ9PgknW2bquEEI7nnlJEKv2GP2ggloxGNlwA,92777
+wandb/cli/cli.py,sha256=eggnDh2vtLOP44F4VGEJEvoGFZZFzhUaOHtPp804NEw,92622
 wandb/mpmain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wandb/mpmain/__main__.py,sha256=bLhspPeHQvNMyRNR7xi9v-02XfW1mhJY2yBWI3bYtbg,57
-wandb/sdk/wandb_metadata.py,sha256=wNvnpkY8WUWctVffZGbu1SZUkKP1plAJr65nNKNjk4s,20331
+wandb/sdk/wandb_metadata.py,sha256=NLMhT6pp16JqzKatmqeo9utEcZfil33mC7vBOxM1Ato,20218
 wandb/sdk/wandb_config.py,sha256=b7kxQVnIh5HCBZXb2pOGZ4c02xCVlW4IQiAu3N-8Opg,10856
 wandb/sdk/wandb_alerts.py,sha256=SwBPBiXRxknMTMGbsVoMMWqWK65UWMcKAdTWZtdwAeo,193
-wandb/sdk/wandb_run.py,sha256=m3zHyYUDcDBr5wGGei74GCPV85ZvFKPtiNRhfgsDMTo,154325
-wandb/sdk/wandb_sync.py,sha256=GmIjtydDNiUT3FHkrFcIZzSlqiRerXA8kY3q72ZbL1k,2275
+wandb/sdk/wandb_run.py,sha256=4MLT0N5MYvdL9xETEKhM7nKnJQNZqaPzZpAK45-wEnU,153621
+wandb/sdk/wandb_sync.py,sha256=GNC8xYqT56ws4aw6Vd_mV_Ck-InEbqnyioSqHIybDug,2106
 wandb/sdk/__init__.py,sha256=N-GTAC0AzbZF2J8RzB33DTmYk9u-jubllCwvhWrPgsE,813
-wandb/sdk/wandb_init.py,sha256=ZlbLSBpfgqNtpv7hsAc3LScs-pw6k9QQEPRiBF7515A,59687
+wandb/sdk/wandb_init.py,sha256=bRO4JbxWZ1DXW2-43OyQQRjYMm62vFOXzB46MmmK_W8,59497
 wandb/sdk/wandb_helper.py,sha256=IbJ7opO8UkfwCDekSjRYIrGBblUxnTPBfp1EdesfF4U,1824
-wandb/sdk/wandb_settings.py,sha256=MdniU_qPcjeAVBATwUcxYKQUnpGlQwsJLf6H8vzTpPQ,55514
+wandb/sdk/wandb_settings.py,sha256=Zbns21r4MCgMq0HjPp-qrFcYxUhrLUFEbpbAiyc9K-E,55657
 wandb/sdk/wandb_summary.py,sha256=yQdOVIPrZaZanhBQ7yuSfPLX0x6dxwkN_KAn4SgjSZU,4536
 wandb/sdk/wandb_watch.py,sha256=F7S9CLbw9PddUp1qBjPRKsOiVFU8LPaq6A9taV3TJKI,4840
-wandb/sdk/wandb_login.py,sha256=qYDpk2e8DVFecZfEV-KaRVqSLeuVGHnHBp5413nqT_M,10863
+wandb/sdk/wandb_login.py,sha256=-2EvouBgBOQgOnh_8Hlb5LCGg6Lz2OR44ieWRRgqdZ8,10952
 wandb/sdk/wandb_metric.py,sha256=a3GiQXr6H18m81uobYjlJaC8CL8iANzI42qxkxfZsDs,3268
-wandb/sdk/wandb_require.py,sha256=Y0ib8h27__t7hXos1F2srfsQzVfzH4BB6wq8E1aRbRA,2950
+wandb/sdk/wandb_require.py,sha256=eYDG4oH1pwik2tHJVEdvivo4R2RypoITFomgr7umMhc,3104
 wandb/sdk/wandb_sweep.py,sha256=Sg_JqxVzmjUBvii41azpdr-c6RPwHOBnSha8k7jrRhk,4028
-wandb/sdk/wandb_setup.py,sha256=IadkZxlkOiMRTS24NstInMHmoSb7q0KeIn1WtjPNajM,12569
+wandb/sdk/wandb_setup.py,sha256=L6xoj3DWS_F_jfQ-i8kISd443AiqO8OSaOx8yCXSWzE,12938
 wandb/sdk/wandb_require_helpers.py,sha256=ZmKv5aXXHDTTU6nYHMLKW4_pt9X-PlaMtbRJl77kHX8,1331
 wandb/sdk/integration_utils/data_logging.py,sha256=DDFtDaUu50aeTTgxCHHYd2f85guqqf2xfEOburRlwwQ,19533
 wandb/sdk/integration_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wandb/sdk/integration_utils/auto_logging.py,sha256=sblNn9BIG7memxTj23UfrGXyA06t39FEt5igMwS0-zU,8233
-wandb/sdk/interface/interface_relay.py,sha256=vQUrk5KESKInZsXpOxWF4YcWRZFLJjNz1mdNywbWbbE,1514
-wandb/sdk/interface/router_queue.py,sha256=4RDXvCcm1OQ7R7eSrZQi7qpaXyd_DzSStYs6CbgXHkQ,1041
-wandb/sdk/interface/message_future_poll.py,sha256=drjrcBKswYPYJJQLFlj7UDXq7_zg7KNcObFVetsbvhQ,1410
+wandb/sdk/interface/interface_relay.py,sha256=lM5ZgJc6m8BS-nWOuzMCwAK0ve9UBGgOD2rgJ9flIhI,867
+wandb/sdk/interface/router_queue.py,sha256=wnApY1iXa_3L0-HO6t42Nbsh68zyEw3FWN3oI6PmoXM,1049
 wandb/sdk/interface/summary_record.py,sha256=-wDv_zLYueeUY8IzyF9NPYnYwF3iBpUWbrsGcHAd2YM,1677
 wandb/sdk/interface/constants.py,sha256=NJNBFr7LkshLI837D3LU3JuEURLzBwza9H-kxcy4ihw,60
-wandb/sdk/interface/interface.py,sha256=xPY5MHwMf1go_pDnDYs1bkQdpSbIF43mm5N-85U1m-Y,36576
+wandb/sdk/interface/interface.py,sha256=oRyy1hs78rZyjf3jn7pHKAqZUA7Swz25I90yDmuJ4vA,37716
 wandb/sdk/interface/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wandb/sdk/interface/router_sock.py,sha256=7-J3SFhnaqPohYaUyljwpbsvYPBXA-ZTENYnXxzbh9U,1060
-wandb/sdk/interface/interface_shared.py,sha256=F6kkd8Aaq_nZALIprMgul0xDkSB8uG0HXJ0-94uEZNw,21276
-wandb/sdk/interface/interface_queue.py,sha256=7lbQz6VYamqYlNpkC-Ckrn5az_Yc9CrbnmwKNFtta44,1700
-wandb/sdk/interface/router_relay.py,sha256=uf0KA_WJ25xMwLsH_3RU_ZhRPNS5qujo1aFU2d_rfzs,953
-wandb/sdk/interface/message_future.py,sha256=5OMApIUKTXLHP98ph_jCdshuPZB_E0Uf3UGZpOQ8sik,685
-wandb/sdk/interface/router.py,sha256=uTgmF0TaQnYq8Frgs7pncUBucA6B7XXCmKcRoYEZV10,3419
-wandb/sdk/interface/interface_sock.py,sha256=dCTpyTQHsWhs2x5v3hY2NY3PK97ta0Pl4B1i9lil_ag,1802
+wandb/sdk/interface/router_sock.py,sha256=S15CRwMoDFYKnEzwqAjvPQC6BlKtJpxE_rZHeaxM2V4,1002
+wandb/sdk/interface/interface_shared.py,sha256=XmEg78CyUHI4XPNtgyx59aQIlzJzQjZYEIeQVECHXSE,18645
+wandb/sdk/interface/interface_queue.py,sha256=zOjUB2nf_wNMoEeXGWP5di-u4-UL2_jvsVZw53EbEsk,1257
+wandb/sdk/interface/router_relay.py,sha256=tOBQYEX7PGw9-qE91CdNhLEQb9KbJiQnlHg7XBOsijA,1390
+wandb/sdk/interface/router.py,sha256=hlYzz7DjDt3sYDJQYLE1NHkxGFDezHgmfznnPRHPcHU,2630
+wandb/sdk/interface/interface_sock.py,sha256=rz6uwKNzLpGDCCVGSlC6ABspmkYD5LnGX_cd9dvaAwU,1043
 wandb/sdk/artifacts/storage_policy.py,sha256=BGIC8QRfryCNzF-72uk53MvQQbZVoKwQGhn2zX3NaRE,2090
 wandb/sdk/artifacts/artifact_state.py,sha256=6fxISSPsK62VeMau_5CI4a9lQXTaUJXcruzk74WUIBE,236
 wandb/sdk/artifacts/artifact_ttl.py,sha256=kNRr9JZ5P2iuo0ofoNKlSddGJkIQXRUjwgO8YUdyN4g,86
 wandb/sdk/artifacts/storage_layout.py,sha256=No2cLJEuU3Dr8rJ5Pq-e-36S6p-WKoYcCG24DKAKzro,73
-wandb/sdk/artifacts/artifact.py,sha256=afIes8o0NyaCZsB1ROc-AsKVrQHdH98PCPa0DVju3EY,91248
+wandb/sdk/artifacts/artifact.py,sha256=Fpo7my8k83DzCXDDxyardKG1CJkl2dPupyEvcuprybI,90432
 wandb/sdk/artifacts/_validators.py,sha256=QlN8yn7muPn1NZMSslHVYM-mcGEEd_4XLfRdv3Wf91Y,4171
 wandb/sdk/artifacts/artifact_file_cache.py,sha256=503z2x4CoTLXk_IfC9jRKBU9fr-X9PMiyG66h0INzQg,9744
 wandb/sdk/artifacts/artifact_saver.py,sha256=2x_n6KKSdntz7XBIDzQt3BqTXQgxfpyHe4U0GHe0nI8,9329
@@ -241,6 +247,7 @@ wandb/sdk/artifacts/exceptions.py,sha256=h9tSi023BhW85_FfjLYjeNtlVgUTqZRfnYM9Wgd
 wandb/sdk/artifacts/storage_handler.py,sha256=ebMCKXvvn7z_eYunXriTOYescstuBbNbxn7MDW9qZAw,1834
 wandb/sdk/artifacts/staging.py,sha256=_U3oH2S7fEevRkcYgo4nmwvdaoJnZR4V-bNiJVWlowA,854
 wandb/sdk/artifacts/artifact_download_logger.py,sha256=XEVxmktMInrd1p2z86CKt3QyTA9vscxJL5-8_eupknI,1501
+wandb/sdk/artifacts/_graphql_fragments.py,sha256=fHvyiABSU5RDa0--FN9D9Dfra_t9vnNzb2MD-S4JYzs,3080
 wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py,sha256=m2s67vmRySPw0bkI69HoIUTO2hyYAo-Nl22jv32Rl90,14002
 wandb/sdk/artifacts/storage_policies/register.py,sha256=xT7kUxubtLqyE-9S6U9E4mCo1PtXl0ZEJ6gVQiS-kGQ,49
 wandb/sdk/artifacts/storage_policies/__init__.py,sha256=bgpWKElL-3iHcLO8pF-L8oezG-dQbp_6vcCYo7CEFAU,226
@@ -268,9 +275,9 @@ wandb/sdk/data_types/utils.py,sha256=Qdj-gUOrrc9wVjBD7Mtmt2o6bHadr1IgzAWF4ijdrz4
 wandb/sdk/data_types/audio.py,sha256=VcbZRjY-9szxIKR-x1Eg375H1Lm5R6znErT4Q8FYrWM,5364
 wandb/sdk/data_types/table.py,sha256=DjppFy0-ii1iMoVOg-h_F1YEUbgk5wGPJDXlQURXLQw,44683
 wandb/sdk/data_types/bokeh.py,sha256=0Aq4CJECPpoAkl7vw5LCGYhdCb9pS8i46jHB9PxTj1k,2534
-wandb/sdk/data_types/saved_model.py,sha256=aKM7CGO5c8VE_7klwyYeC9tnl3XKIunVT03vyrXGFaQ,16398
+wandb/sdk/data_types/saved_model.py,sha256=p9k92Jc6ihqTjir-GesTKpPiC92m6nen0zqUEcw0W5E,16254
 wandb/sdk/data_types/plotly.py,sha256=LEIHRqtHHpSou04AEUsyH8_UMXP4adrEHHaKoEtQkxo,2913
-wandb/sdk/data_types/video.py,sha256=q0vvyZHzHnxjYmX6VFVuTGT72m52YdYGnW-jAwqJSJg,9715
+wandb/sdk/data_types/video.py,sha256=iug92X8xW5JIquk1QvMom8-N9SKqEO3-dv9vxA28M_s,9344
 wandb/sdk/data_types/image.py,sha256=o_ZwqB4uO1-R8QLuO9J6aiUuyANMz2qPQJAskushozQ,31239
 wandb/sdk/data_types/_private.py,sha256=zp2NRarTlIq4Hk3R2xp7j_qPGNzBMnaGHrZUN82shaY,299
 wandb/sdk/data_types/base_types/media.py,sha256=QO6aQdNGYtwr6qZYUQ9-WbTpkB3adCBTXWSqO3ZCJjI,14284
@@ -278,11 +285,16 @@ wandb/sdk/data_types/base_types/json_metadata.py,sha256=oKpimndUQvDW30n15tb2pbyy
 wandb/sdk/data_types/base_types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wandb/sdk/data_types/base_types/wb_value.py,sha256=HtV2VblGg_nnSBQ1rRSjlRuY-NyDTbeDVCJzlQ49SUc,11460
 wandb/sdk/data_types/helper_types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wandb/sdk/data_types/helper_types/bounding_boxes_2d.py,sha256=hzVq4wTxx3-rJk4zaPKHR6-v0cTwj46JC9yUMTwuhUU,13010
+wandb/sdk/data_types/helper_types/bounding_boxes_2d.py,sha256=z_cssRDjzWSoDopThg4W3oUQ6Cy17RvrO4WhB4Ha_V0,13272
 wandb/sdk/data_types/helper_types/classes.py,sha256=cg-AwUgULbb3T_-ptOdliXGvYjXZYRj8rvSmT7fl9ds,5520
-wandb/sdk/data_types/helper_types/image_mask.py,sha256=JwgXaYTFkw4gEGe35ptIScnv4KA5ViqrUlVD8SZwmRQ,8601
+wandb/sdk/data_types/helper_types/image_mask.py,sha256=kx77gxfNnhNIk-Nvaw2SC-NjnDlD5x7RXEB0DBuZCis,8727
 wandb/sdk/verify/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wandb/sdk/verify/verify.py,sha256=9vGbgfu4lZIl3_wdDO91H1sCK-mpZZFemtbhs8G-5zc,16703
+wandb/sdk/verify/verify.py,sha256=Bz7DKW7zmhNAPJQNAAQszAmjBilxs2WJJ9rVHRml6LM,18299
+wandb/sdk/mailbox/mailbox_handle.py,sha256=YLKNMkytIAoIpMuV5Nu_g04JdTGzm8P9d9iw2EPJhV4,3743
+wandb/sdk/mailbox/__init__.py,sha256=C4x3ym_hymymKFZiI1Q0HPlGx5x5CzeKPVF229Z2V8w,856
+wandb/sdk/mailbox/response_handle.py,sha256=kZWVvVbpc92u3u23FTwAod-QCPFENUYREpjXm-wkh6Q,5031
+wandb/sdk/mailbox/wait_with_progress.py,sha256=O3yGgUC3ztpfmdLk_CyirJVeXiCzF8nuDI2CqNiak90,4126
+wandb/sdk/mailbox/mailbox.py,sha256=jUEP2Ltgi9rCZmiDtfb5ZESaqwB2wH-sy72OLRvATZk,4314
 wandb/sdk/launch/wandb_reference.py,sha256=t4REjZz5lwB9fjDW2eo8uRgw9KeLsPeZ1Uu8tiFDBfA,4253
 wandb/sdk/launch/_launch.py,sha256=FZ2JjZc1rzAIW3biWpVKHzxTwWmRx1ZK3n5S-YUQMKE,11829
 wandb/sdk/launch/__init__.py,sha256=moXY557JibPbvE1GSSh3nGFiyelTVaJZMdFSv61Dn2k,399
@@ -334,12 +346,12 @@ wandb/sdk/launch/inputs/internal.py,sha256=xtEZ7znfUCL2ME7xZv5bD8I5C6hNpN2EqjgNd
 wandb/sdk/launch/inputs/files.py,sha256=BWGDmAfDPLja7zz0bUzYLaF3wH4cTPesD9LqDuJRUoU,4685
 wandb/sdk/launch/inputs/manage.py,sha256=O0IsTWhjUftihohRfK7DkT186LjHMMcE2NSMeG8IFBY,5003
 wandb/sdk/launch/inputs/schema.py,sha256=Fs4CcxI0ifFagazbNp47uJP15JL3v43PAvVNO93f9XY,1418
-wandb/sdk/internal/sender.py,sha256=1oJX1f1FqxloyCIgy-A3bM8XLThdcPNZGkRn1PhAN1w,65287
+wandb/sdk/internal/sender.py,sha256=Bwit4q53suRX37KKTiONT65V1f6fm87o1RjnkLIYgi8,64514
 wandb/sdk/internal/internal.py,sha256=o9_-6C99LMOx_YB5xuZ1I1KgpeHk3jUgll63sEe9gCY,12182
 wandb/sdk/internal/run.py,sha256=8OhVy2vfgPC8pNFq0tJ4CkQHETOBfQsFDghw50ccSXc,682
 wandb/sdk/internal/job_builder.py,sha256=_L-DfGFEzDl0JLXPo535KPK3Bk0vqXsTQVF7buXeaiY,22962
-wandb/sdk/internal/internal_api.py,sha256=WrWaeVasXrSyTj9eABz3ULa3vIAIb2lGzZDWK8ng9KQ,159641
-wandb/sdk/internal/handler.py,sha256=k3b5pHh8NQgh8BqegaKLqawt9d_cnxKW64i_Z6cw3pk,34183
+wandb/sdk/internal/internal_api.py,sha256=R9UF5K2xrSGEQ-46Ch_Ab0CW6EHouOHCk7Z1GVyyRIM,160084
+wandb/sdk/internal/handler.py,sha256=zquR4k6Ah5ohg9n5XJC_bKAulz3aSUuMjUlHqiIOfFE,34280
 wandb/sdk/internal/thread_local_settings.py,sha256=UqD6kfjsy6mvxIWcjhd-vJWkNRCeU1whuRe_-VGIklQ,527
 wandb/sdk/internal/sender_config.py,sha256=qEuXwOskca3sYyDIRsswlXmj9StCCS0WKQ1qrBXbIjw,6767
 wandb/sdk/internal/settings_static.py,sha256=1Dc2MoVCIskHo7rKafjtrVxY9sQWZMNeee39LYBe9t0,3668
@@ -373,16 +385,16 @@ wandb/sdk/internal/system/assets/tpu.py,sha256=nwv3C9a6dgD1hc0461Rh28NQkWbNQ5fJO
 wandb/sdk/internal/system/assets/network.py,sha256=n3buH2-InLGT35PVlvhY5eHv_Hm5Kl_N8jH576OiVi8,3392
 wandb/sdk/internal/system/assets/open_metrics.py,sha256=VqOncOSnPztqSyQ0g-zeJWcjc_lircRk6F76rwoBWR4,9631
 wandb/sdk/internal/system/assets/gpu.py,sha256=DnnfH9fvvfiRXzMAXjCgHZZZt_Kkn2br74T3Y_oyyhw,13722
-wandb/sdk/backend/backend.py,sha256=c0vdeiQr6D3Ua8hvKKlYoELfDTRxPrHIX4E3qRZsX00,7492
+wandb/sdk/backend/backend.py,sha256=w9aV2IEEjUFjjAzAe8UX7Pfz49hp3dcI-fXRhpQWq9A,7791
 wandb/sdk/backend/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wandb/sdk/lib/deprecate.py,sha256=r8kT4UY2DSG7V5N78MG46kQOzycBp3_8lH49KnDXxXM,1462
 wandb/sdk/lib/exit_hooks.py,sha256=_4oozaRQCJi8NJfZvHsA8livvFb0trZKLOGB8_UcHGk,1540
 wandb/sdk/lib/server.py,sha256=dn0RXjAnpmigAULLyq75XHYyHxw17MLCJ9fhPq2v22I,1623
 wandb/sdk/lib/paths.py,sha256=YiEE5mkYB5ahMuI4C27IsNvejC3z6MI5JPW1iISi864,4529
-wandb/sdk/lib/sock_client.py,sha256=-B2V0rTRM_pnZsqV0GuCjPTvEOAjfvjICTz5nohEIGY,10351
+wandb/sdk/lib/sock_client.py,sha256=YzR4uFMgFBA95ySl9h2QwIqXKMEAyb36Y09vpf70IPU,8383
 wandb/sdk/lib/credentials.py,sha256=WmVdzL1rFy7S0WIHf1ZYd98_eaHTxPKUobReRSPQgBM,4737
 wandb/sdk/lib/runid.py,sha256=Qa-5ft4B85YUazkV_18OYwf9JhMaAVp0JAInZzxzX5o,392
-wandb/sdk/lib/redirect.py,sha256=eTNvQp4SFLgN5Kxwmh1BP0du8_7YZxcgoXsmE_2IHSM,26132
+wandb/sdk/lib/redirect.py,sha256=-7iM7Bg12LzsLivkkVL3wO1pT4ofKeShz1L6NTHoSVc,27249
 wandb/sdk/lib/preinit.py,sha256=11QkGmBpoGazNaUGvyDIzBJA4QTggj7fGQdugpCvOiw,1450
 wandb/sdk/lib/file_stream_utils.py,sha256=NN4qaSyNufvlkg96a1YUjQ8-pYtCVU7xagJaW8mqAyI,4018
 wandb/sdk/lib/interrupt.py,sha256=Ea7aSVHcY0wLYnyMfm4jZbQcwGPgRcuPuoBSe0Swuds,1353
@@ -400,30 +412,31 @@ wandb/sdk/lib/retry.py,sha256=BAgPgJb1SxRguor2KTfOFnQpTlyNE4M_X8VcEc7YfMs,10173
 wandb/sdk/lib/json_util.py,sha256=fAsYfaw8iMmx3KJ0wSthUSj_XpF1iAysad4XZY1kQdo,2584
 wandb/sdk/lib/gitlib.py,sha256=RjlbRyZpIRqc0toVJQSpNxVgPpwx1LSjeF9LSuf4bAY,7836
 wandb/sdk/lib/proto_util.py,sha256=JxDldi8j6dfF_Lx9zOwqYL6LQZhYYGgKt_AfZtYHAW0,2894
+wandb/sdk/lib/console_capture.py,sha256=eC1bThCHa9DiQTk21ALdK8w1ETeydBpiTOYKN31EjVs,4923
 wandb/sdk/lib/ipython.py,sha256=I2iT6vY90Aaws7tmjzW_CrUdQFmfV9-WhAPk8at9LQg,3782
 wandb/sdk/lib/service_token.py,sha256=c4olk_g4Suo0zTTDaoR78oHLwpt3NgDzQeF94raigME,2471
 wandb/sdk/lib/run_moment.py,sha256=HMx8jYIaWwmSPNViP54ZYMbxLKBn2m_ezYCa67UC6H8,2464
-wandb/sdk/lib/mailbox.py,sha256=iuxunHXc2JzxFyubyS6CqxMXsrSpSFswBfxASF-hmss,13832
 wandb/sdk/lib/hashutil.py,sha256=aU8YsAgkTcYDlqshfcDjItiO2s9lqq2YdKt7RoA2adQ,2763
 wandb/sdk/lib/module.py,sha256=PWxpFqOYmLyKPF-VgfINZXzkFxDcoQVunVDVNWNnbxQ,2098
 wandb/sdk/lib/handler_util.py,sha256=mT9CKsmboq4lPWElsi4uo9ORHhx6OYTr7KY2QtgbM6M,589
-wandb/sdk/lib/service_connection.py,sha256=z4zMzqMxD86Qs8YVJOb_N_GJmkCOYnoBsBGEY9093S4,6504
-wandb/sdk/lib/progress.py,sha256=GN3AhhjUK6D2SGCXSercS6pSepyHqVvE-X7f5W1WnW4,8609
+wandb/sdk/lib/service_connection.py,sha256=pf1T0isbDwTkv9MvuIpuk1MHTVkCebK4N8n9pXF4SSg,7455
+wandb/sdk/lib/progress.py,sha256=GakG9R_mJxVx8zG8atVHKydDZt0uRFEPwls7b3hUpsc,10652
 wandb/sdk/lib/gql_request.py,sha256=-KRTtT2zqn7pTOBcl41IBdbvQ25YCrxVurROpRqrWqw,2399
+wandb/sdk/lib/asyncio_compat.py,sha256=AokSd8OXVYF1ALggThRr4RrgD4UiUtDjP3LRE4HLEm4,6584
 wandb/sdk/lib/config_util.py,sha256=YdYvk-KbTdTa-84XegpvbqMuQfdlOREFiVR7m3q6e3Q,2917
-wandb/sdk/lib/apikey.py,sha256=1nCkfu7G-mOdJYTyByDxeuQHFZmYeT-UUm4OL_mgSGU,8715
+wandb/sdk/lib/apikey.py,sha256=a3m7IaNyx7_yP-rB_PI-DiVJgtv27gqapvYYHgogx_A,10508
 wandb/sdk/lib/printer.py,sha256=v3T2wqere8HQIVba9LYd8xCIUtV8e8l21Fw4XUEPbXM,15289
 wandb/sdk/lib/fsm.py,sha256=RLr_wp6_9vSRQbnoP6aqOXYghi53FV93eUyAMMEjcTg,5175
 wandb/sdk/lib/lazyloader.py,sha256=4Seis4C8Ph6a2-3scfYqeHh7KjKG6PO0WJ2ZPW30O0o,1891
 wandb/sdk/service/service.py,sha256=tfw6zDHLv9q7E_W25g5eddC6nuyvFPLZvpsTZv1SBM0,8614
 wandb/sdk/service/server.py,sha256=_0MAn4Y1hs2AnapyQb5heYiixKQNxFENfkxIGoiiIB8,3460
-wandb/sdk/service/streams.py,sha256=DR7vtHV0q4rBMGJwnS-mYoOfHix9IS8knpWrCIKNv1g,14642
+wandb/sdk/service/streams.py,sha256=lNtn6aujS5GlhzsG1Uw3JpF7GL5tWewN1dW39uP5hFo,14801
 wandb/sdk/service/_startup_debug.py,sha256=A6P5iCYRoNGcVulziUc7Mw_Y2MCG8PAIAUVjzzMUTPs,602
-wandb/sdk/service/server_sock.py,sha256=lpeVRPI0ehRZHoNyG0dCGmBdLnlv_Xl1Znf4qqZHYs4,9583
+wandb/sdk/service/server_sock.py,sha256=HiuncxAaNHUpii7CCIvYr1XVirBHNE7d3WBK2tVVsxY,9695
 wandb/sdk/service/port_file.py,sha256=_VeftD2LjNBeQH6nfzd9Dyq2u4fT2YRLE7D1ikIhYmU,1546
 wandb/sdk/service/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wandb/filesync/step_checksum.py,sha256=AIz9ALGk59OiR3NtxGj3CJKnsKSQ_1f9PmrbIzxuarE,4690
-wandb/filesync/dir_watcher.py,sha256=8dct43c0syDAeFFHhRAetlgH247MItjCCV7gcXo5xVQ,16345
+wandb/filesync/dir_watcher.py,sha256=9Y57-5-hWZFzCuBL6HQVtm-MIE9p8kCRU_g_YZUVHvo,16345
 wandb/filesync/upload_job.py,sha256=t-QgMheUcxFE8C9TIJ5Wp9Cl_4jlpCzCKfsRoPzrHbs,5493
 wandb/filesync/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wandb/filesync/stats.py,sha256=bjVBoCU9I9ke_XbEqtUgmKJVSmFxxAs2JciBzGWPhDg,3050

wandb/sdk/interface/message_future.py

@@ -1,27 +0,0 @@
-"""MessageFuture - represents a message result of an asynchronous operation.
-
-Base class MessageFuture for MessageFutureObject and MessageFuturePoll
-
-"""
-
-import threading
-from abc import abstractmethod
-from typing import Optional
-
-from wandb.proto import wandb_internal_pb2 as pb
-
-
-class MessageFuture:
-    _object: Optional[pb.Result]
-
-    def __init__(self) -> None:
-        self._object = None
-        self._object_ready = threading.Event()
-
-    def _set_object(self, obj: pb.Result) -> None:
-        self._object = obj
-        self._object_ready.set()
-
-    @abstractmethod
-    def get(self, timeout: Optional[int] = None) -> Optional[pb.Result]:
-        raise NotImplementedError

wandb/sdk/interface/message_future_poll.py

@@ -1,50 +0,0 @@
-"""MessageFuturePoll - Derived from MessageFuture but implementing polling loop.
-
-MessageFuture represents a message result of an asynchronous operation.
-
-MessageFuturePoll implements a polling loop to periodically query for a
-completed async operation.
-
-"""
-
-import time
-from typing import Any, Optional
-
-from wandb.proto import wandb_internal_pb2 as pb
-
-from .message_future import MessageFuture
-
-
-class MessageFuturePoll(MessageFuture):
-    _fn: Any
-    _xid: str
-
-    def __init__(self, fn: Any, xid: str) -> None:
-        super().__init__()
-        self._fn = fn
-        self._xid = xid
-
-    def get(self, timeout: Optional[int] = None) -> Optional[pb.Result]:
-        self._poll(timeout=timeout)
-        if self._object_ready.is_set():
-            return self._object
-        return None
-
-    def _poll(self, timeout: Optional[int] = None) -> None:
-        if self._object_ready.is_set():
-            return
-        done = False
-        start_time = time.time()
-        sleep_time = 0.5
-        while not done:
-            result = self._fn(xid=self._xid)
-            if result:
-                self._set_object(result)
-                done = True
-                continue
-            now_time = time.time()
-            if timeout and start_time - now_time > timeout:
-                done = True
-                continue
-            time.sleep(sleep_time)
-            sleep_time = min(sleep_time * 2, 5)