wandb 0.19.7__py3-none-any.whl → 0.19.9__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (90) hide show
  1. wandb/__init__.py +5 -1
  2. wandb/__init__.pyi +43 -9
  3. wandb/_pydantic/__init__.py +23 -0
  4. wandb/_pydantic/base.py +113 -0
  5. wandb/_pydantic/v1_compat.py +262 -0
  6. wandb/apis/paginator.py +82 -38
  7. wandb/apis/public/api.py +10 -64
  8. wandb/apis/public/artifacts.py +73 -17
  9. wandb/apis/public/files.py +2 -2
  10. wandb/apis/public/projects.py +3 -2
  11. wandb/apis/public/reports.py +2 -2
  12. wandb/apis/public/runs.py +19 -11
  13. wandb/bin/gpu_stats +0 -0
  14. wandb/data_types.py +1 -1
  15. wandb/filesync/dir_watcher.py +2 -1
  16. wandb/integration/metaflow/metaflow.py +19 -17
  17. wandb/integration/sacred/__init__.py +1 -1
  18. wandb/jupyter.py +18 -15
  19. wandb/proto/v3/wandb_internal_pb2.py +7 -3
  20. wandb/proto/v3/wandb_settings_pb2.py +2 -2
  21. wandb/proto/v3/wandb_telemetry_pb2.py +10 -10
  22. wandb/proto/v4/wandb_internal_pb2.py +3 -3
  23. wandb/proto/v4/wandb_settings_pb2.py +2 -2
  24. wandb/proto/v4/wandb_telemetry_pb2.py +10 -10
  25. wandb/proto/v5/wandb_internal_pb2.py +3 -3
  26. wandb/proto/v5/wandb_settings_pb2.py +2 -2
  27. wandb/proto/v5/wandb_telemetry_pb2.py +10 -10
  28. wandb/proto/wandb_deprecated.py +2 -0
  29. wandb/sdk/artifacts/_graphql_fragments.py +18 -20
  30. wandb/sdk/artifacts/_validators.py +1 -0
  31. wandb/sdk/artifacts/artifact.py +81 -46
  32. wandb/sdk/artifacts/artifact_saver.py +16 -2
  33. wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py +23 -2
  34. wandb/sdk/backend/backend.py +16 -5
  35. wandb/sdk/data_types/audio.py +1 -3
  36. wandb/sdk/data_types/base_types/media.py +11 -4
  37. wandb/sdk/data_types/image.py +44 -25
  38. wandb/sdk/data_types/molecule.py +1 -5
  39. wandb/sdk/data_types/object_3d.py +2 -1
  40. wandb/sdk/data_types/saved_model.py +7 -9
  41. wandb/sdk/data_types/video.py +1 -4
  42. wandb/sdk/interface/interface.py +65 -43
  43. wandb/sdk/interface/interface_queue.py +0 -7
  44. wandb/sdk/interface/interface_relay.py +6 -16
  45. wandb/sdk/interface/interface_shared.py +47 -40
  46. wandb/sdk/interface/interface_sock.py +1 -8
  47. wandb/sdk/interface/router.py +22 -54
  48. wandb/sdk/interface/router_queue.py +11 -10
  49. wandb/sdk/interface/router_relay.py +24 -12
  50. wandb/sdk/interface/router_sock.py +6 -11
  51. wandb/{apis/public → sdk/internal}/_generated/__init__.py +0 -6
  52. wandb/sdk/internal/_generated/base.py +226 -0
  53. wandb/{apis/public → sdk/internal}/_generated/server_features_query.py +3 -3
  54. wandb/{apis/public → sdk/internal}/_generated/typing_compat.py +1 -1
  55. wandb/sdk/internal/internal_api.py +138 -47
  56. wandb/sdk/internal/sender.py +5 -1
  57. wandb/sdk/internal/sender_config.py +8 -11
  58. wandb/sdk/internal/settings_static.py +24 -2
  59. wandb/sdk/lib/apikey.py +15 -16
  60. wandb/sdk/lib/console_capture.py +172 -0
  61. wandb/sdk/lib/redirect.py +102 -76
  62. wandb/sdk/lib/run_moment.py +4 -6
  63. wandb/sdk/lib/service_connection.py +37 -17
  64. wandb/sdk/lib/sock_client.py +2 -52
  65. wandb/sdk/lib/wb_logging.py +161 -0
  66. wandb/sdk/mailbox/__init__.py +3 -3
  67. wandb/sdk/mailbox/mailbox.py +31 -17
  68. wandb/sdk/mailbox/mailbox_handle.py +127 -0
  69. wandb/sdk/mailbox/{handles.py → response_handle.py} +34 -66
  70. wandb/sdk/mailbox/wait_with_progress.py +16 -15
  71. wandb/sdk/service/server_sock.py +4 -2
  72. wandb/sdk/service/streams.py +10 -5
  73. wandb/sdk/wandb_config.py +44 -43
  74. wandb/sdk/wandb_init.py +151 -92
  75. wandb/sdk/wandb_metadata.py +107 -91
  76. wandb/sdk/wandb_run.py +160 -54
  77. wandb/sdk/wandb_settings.py +410 -202
  78. wandb/sdk/wandb_setup.py +3 -1
  79. wandb/sdk/wandb_sync.py +1 -7
  80. {wandb-0.19.7.dist-info → wandb-0.19.9.dist-info}/METADATA +3 -3
  81. {wandb-0.19.7.dist-info → wandb-0.19.9.dist-info}/RECORD +87 -83
  82. wandb/apis/public/_generated/base.py +0 -128
  83. wandb/sdk/interface/message_future.py +0 -27
  84. wandb/sdk/interface/message_future_poll.py +0 -50
  85. /wandb/{apis/public → sdk/internal}/_generated/enums.py +0 -0
  86. /wandb/{apis/public → sdk/internal}/_generated/input_types.py +0 -0
  87. /wandb/{apis/public → sdk/internal}/_generated/operations.py +0 -0
  88. {wandb-0.19.7.dist-info → wandb-0.19.9.dist-info}/WHEEL +0 -0
  89. {wandb-0.19.7.dist-info → wandb-0.19.9.dist-info}/entry_points.txt +0 -0
  90. {wandb-0.19.7.dist-info → wandb-0.19.9.dist-info}/licenses/LICENSE +0 -0
wandb/sdk/wandb_setup.py CHANGED
@@ -21,7 +21,7 @@ from typing import TYPE_CHECKING, Any, Union
21
21
 
22
22
  import wandb
23
23
  import wandb.integration.sagemaker as sagemaker
24
- from wandb.sdk.lib import import_hooks
24
+ from wandb.sdk.lib import import_hooks, wb_logging
25
25
 
26
26
  from . import wandb_settings
27
27
  from .lib import config_util, server
@@ -294,6 +294,7 @@ def singleton() -> _WandbSetup | None:
294
294
  return None
295
295
 
296
296
 
297
+ @wb_logging.log_to_all_runs()
297
298
  def _setup(
298
299
  settings: Settings | None = None,
299
300
  start_service: bool = True,
@@ -382,6 +383,7 @@ def setup(settings: Settings | None = None) -> _WandbSetup:
382
383
  return _setup(settings=settings)
383
384
 
384
385
 
386
+ @wb_logging.log_to_all_runs()
385
387
  def teardown(exit_code: int | None = None) -> None:
386
388
  """Waits for wandb to finish and frees resources.
387
389
 
wandb/sdk/wandb_sync.py CHANGED
@@ -7,7 +7,6 @@ from wandb.errors.term import termerror, termlog
7
7
  from . import wandb_setup
8
8
  from .backend.backend import Backend
9
9
  from .lib.runid import generate_id
10
- from .mailbox import Mailbox
11
10
 
12
11
  if TYPE_CHECKING:
13
12
  from wandb.proto import wandb_internal_pb2
@@ -49,12 +48,7 @@ def _sync(
49
48
  service = wl.ensure_service()
50
49
  service.inform_init(settings=settings, run_id=stream_id)
51
50
 
52
- mailbox = Mailbox()
53
- backend = Backend(
54
- settings=wl.settings,
55
- service=service,
56
- mailbox=mailbox,
57
- )
51
+ backend = Backend(settings=wl.settings, service=service)
58
52
  backend.ensure_launched()
59
53
 
60
54
  assert backend.interface
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: wandb
3
- Version: 0.19.7
3
+ Version: 0.19.9
4
4
  Summary: A CLI and library for interacting with the Weights & Biases API.
5
5
  Project-URL: Source, https://github.com/wandb/wandb
6
6
  Project-URL: Bug Reports, https://github.com/wandb/wandb/issues
@@ -57,13 +57,13 @@ Requires-Dist: protobuf!=4.21.0,!=5.28.0,<6,>=3.15.0; python_version == '3.9' an
57
57
  Requires-Dist: protobuf!=4.21.0,!=5.28.0,<6,>=3.19.0; python_version > '3.9' and sys_platform == 'linux'
58
58
  Requires-Dist: protobuf!=4.21.0,!=5.28.0,<6,>=3.19.0; sys_platform != 'linux'
59
59
  Requires-Dist: psutil>=5.0.0
60
- Requires-Dist: pydantic<3,>=2.6
60
+ Requires-Dist: pydantic<3
61
61
  Requires-Dist: pyyaml
62
62
  Requires-Dist: requests<3,>=2.0.0
63
63
  Requires-Dist: sentry-sdk>=2.0.0
64
64
  Requires-Dist: setproctitle
65
65
  Requires-Dist: setuptools
66
- Requires-Dist: typing-extensions<5,>=4.4; python_version < '3.12'
66
+ Requires-Dist: typing-extensions<5,>=4.4
67
67
  Provides-Extra: aws
68
68
  Requires-Dist: boto3; extra == 'aws'
69
69
  Requires-Dist: botocore>=1.5.76; extra == 'aws'
@@ -1,12 +1,12 @@
1
1
  package_readme.md,sha256=U9047nyMDICgctm1HLm4HfXwFnFKsEn2m77hsYPUZ1I,4298
2
- wandb/__init__.py,sha256=i4gYgAkoE1xFoiJmMxR4Hh985FID5705UOsnTJ08rXY,6988
3
- wandb/__init__.pyi,sha256=QAnvRB-jx3-ZSf71XK5Zz3ataVKfgEiN24a-F0-wVYM,47259
2
+ wandb/__init__.py,sha256=e31yaHVH78RZDBC-FKl-IttiNnT7bxTdpcqXPAi8Sfg,7147
3
+ wandb/__init__.pyi,sha256=pstgzL37MQKvtnejJA0C4dKyNE_ikcYTM-AwQTKc7mw,48110
4
4
  wandb/__main__.py,sha256=gripuDgB7J8wMMeJt4CIBRjn1BMSFr5zvsrt585Pnj4,64
5
5
  wandb/_globals.py,sha256=CccwOAls5bxJArYHg12b08ZeKR8Qu9u57GtYWjBH0o0,702
6
6
  wandb/_iterutils.py,sha256=bOhOrOENEDzOMtyQqHysivzHri9axcVlCfXt4Ai1ot8,1545
7
- wandb/data_types.py,sha256=tjxcQ8padGuGxST192PyEDX_nhU__izHcAK-kaSyevI,2276
7
+ wandb/data_types.py,sha256=M-wqAO0FtIqvj28556u3h4nzSwlRcbxhXoX0B9jlJSo,2283
8
8
  wandb/env.py,sha256=NtyTiZbU3xrpNiISvrTVO5SDBXVqHl58wmrhpB4MCSA,13736
9
- wandb/jupyter.py,sha256=ip6ukYdoFzvPiHO-IUNnbMMxRnYl3ebEoC0X143_VEw,17274
9
+ wandb/jupyter.py,sha256=TdnXHFBZ9Kwi3jc5uHYvDWzjZWThWXFUG1UsqoTSXao,17352
10
10
  wandb/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
11
11
  wandb/sklearn.py,sha256=hbPkefhS39A1RRymn0nHZZmKM2TrOd4xjlkthTZe9pY,803
12
12
  wandb/trigger.py,sha256=PaitU3sX6ekGkd2R8iD6d_VtI72ypF7LaPBXh3rXY7Q,615
@@ -14,6 +14,9 @@ wandb/util.py,sha256=tlaNaz9W6r2T0bs0-KVDcuSPE1PiZ7642GjCjAx8Mvw,63525
14
14
  wandb/wandb_agent.py,sha256=0TKjcJ8jxEqaXMrs_dIflXfWFFOaulfSjzp_fpYiXD0,20985
15
15
  wandb/wandb_controller.py,sha256=SksJdgwn14PpnUoIaBjJ9Ki4Nksl9BpQGGn42hT0xZg,24936
16
16
  wandb/wandb_run.py,sha256=CNh9S6uubFk8FphQjzkbvedyyGCN9aBEsRBKjy8tqqs,155
17
+ wandb/_pydantic/__init__.py,sha256=nZMbUoWR_JvYkvjdUcgt_NvfzQdvMuC3RVHTC9IKDMk,452
18
+ wandb/_pydantic/base.py,sha256=4sZFuMToGU1OmDI_1X4H9Kc0LIT-MErhwtYT_L0t2rA,3398
19
+ wandb/_pydantic/v1_compat.py,sha256=wz8RhjcGAxWdKE0MjYJ6F6G_ZvxiYHjSh00ILIUB1_8,9435
17
20
  wandb/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
18
21
  wandb/agents/pyagent.py,sha256=paZ04bMdruGEoApHHIBTJVJphK2I-ZHMNQ51nnR4ZZ4,13384
19
22
  wandb/analytics/__init__.py,sha256=WG_Mh20Hr8d3vDmGMcfDXCMEIew3uzDYZAJwFsrbAug,50
@@ -22,7 +25,7 @@ wandb/apis/__init__.py,sha256=uPZKqlX8Y9YcHoERJshWRGS3ukMOIVWi4sDnkO5_CYY,1338
22
25
  wandb/apis/attrs.py,sha256=MRle0pRB8QGV31MRjVfqE7l8XJOA04FFKCtOOuDt0xE,1445
23
26
  wandb/apis/internal.py,sha256=Whl1x29iH_DdIu47vUVvDSyOavEVaovdmlqXmu4-yOw,7622
24
27
  wandb/apis/normalize.py,sha256=mamR8PquPZU7YAVNP1k_Q941l7FRwEoQhJhTH3SgSJQ,2676
25
- wandb/apis/paginator.py,sha256=18qGrzVzjamd-8AlrwvhUq11Bk31qCjxWWVcYzijGa8,2118
28
+ wandb/apis/paginator.py,sha256=ZxG7oJkw5DPjN9u-GAFS_kBokmUiIW21QLH8SMVQ8YI,3550
26
29
  wandb/apis/importers/__init__.py,sha256=ewph-RBUC9FnxL6uWbuuW0Hmdv2cHsxfqnZi0546ds4,38
27
30
  wandb/apis/importers/mlflow.py,sha256=uoxzm3bSOqjQyUkgOBwuzpaBjTdGp4drPaqE9f7Ny5w,8256
28
31
  wandb/apis/importers/validation.py,sha256=fVILb8fXKk_oj8BbBhUOnzSPwoUuYfF8lflwJklviYY,3190
@@ -31,34 +34,27 @@ wandb/apis/importers/internals/internal.py,sha256=hJAe3iYTU1uM3MJ5KLm7e4XLMzx5EI
31
34
  wandb/apis/importers/internals/protocols.py,sha256=shZBD2bRMo81xPSUQdK2d9KK8qpop4CmMkizg8FI3WM,2885
32
35
  wandb/apis/importers/internals/util.py,sha256=HvphFwhSxuYqg0omNhfP1GwZV6UzZN-m85CNaSUZXQs,2083
33
36
  wandb/apis/public/__init__.py,sha256=ldsy9V8UQlRnsIf9E4qRMtYIdH89oQ8eiYThQsphXks,1070
34
- wandb/apis/public/api.py,sha256=OGDy0Iw8cRQJYVmHDVuMb8idxoNbEZLITrgaGWo0-E0,58882
35
- wandb/apis/public/artifacts.py,sha256=rBzZJQsftgyImSu_r8abSjguqA0Q-SDQtkYZALJ1cYE,33366
37
+ wandb/apis/public/api.py,sha256=2mlZOQ4rR2nfNICFoMZMyDcm3leRnlRDN5DNmRh7EDA,56720
38
+ wandb/apis/public/artifacts.py,sha256=D4e9CQ_C-PvkeM8j5yLR56TJw6O30WV8W0Of3qbQaYI,35908
36
39
  wandb/apis/public/const.py,sha256=icNtcS3gTCtvevLWuTOCqm0FHEfLQ0P80mA69dWeEXs,121
37
- wandb/apis/public/files.py,sha256=z3QLBT_boqrncCTWjsfmW2vmX1BY6Q8e4KkGaywYe2E,8479
40
+ wandb/apis/public/files.py,sha256=sQNGU18jmcpc6XccK5-jUXFhnqjOJgu9-fFljsTfg4Y,8497
38
41
  wandb/apis/public/history.py,sha256=4gwe9HJ_NH9SSPtLtP7ELw4nIsxLPrY6ji13EK1siyM,4636
39
42
  wandb/apis/public/jobs.py,sha256=Ewwh-KK2mNoOXIz3p4Q1fMoWqAkGDl1vGbTBZIkOxPc,22372
40
- wandb/apis/public/projects.py,sha256=F5cPDxAbZD4-oVB_BxBCTsZk6k1tVL0cPU3Z0YEUqzo,4322
43
+ wandb/apis/public/projects.py,sha256=rglvcBD7npD6McZa3VKdbxz8DuHPraZVBwlRq57Yy2E,4420
41
44
  wandb/apis/public/query_generator.py,sha256=WfyaQuBcTHEECOJ2zN_0iqkUufki9MX28fj3axYZCLU,5977
42
45
  wandb/apis/public/registries.py,sha256=CaL1vD2dH74aMu916vVfgqSsKFsQWMVZf5jNfXiboa0,18404
43
- wandb/apis/public/reports.py,sha256=xUm3UO8OnCGXMMqXw-YUhOi143HccrDPS8VO7B0xW0s,15278
44
- wandb/apis/public/runs.py,sha256=_v9F-Onzx_LJha8xxWDxXk3BWEK1Gg0Yq9XJSxtlx_c,35202
46
+ wandb/apis/public/reports.py,sha256=DR_Kw3dznuy1UeaiJ_ENLbimV1j1kSerYr_vqoPtWNY,15302
47
+ wandb/apis/public/runs.py,sha256=0RIX-HChtT5yug-AeilTnmnadE6rqjROCW9gBkya3pI,35356
45
48
  wandb/apis/public/sweeps.py,sha256=FYWvA-h-xqCnZ8PYXRwOEvRHgl3BS0RhuZpTftIMNrc,6547
46
49
  wandb/apis/public/teams.py,sha256=YJAsJLRVM5r6UT48tdGGVA2V_f7nayDe5Q5NRTFQ8m4,5474
47
50
  wandb/apis/public/users.py,sha256=nbguWN0QJWvhsmgVx6IrkLYbjJcyI-eJQAUVaoY5Qjc,3796
48
51
  wandb/apis/public/utils.py,sha256=6jfEut3u4qc1KiD4fR04YM2X6tmNW997MT44ptL7bTU,3681
49
- wandb/apis/public/_generated/__init__.py,sha256=m19wAWXw_4qFC1pNaJPp1XBngTOjCab-4Eds3YmDT3U,528
50
- wandb/apis/public/_generated/base.py,sha256=zaD3C8aYH-WNw5pL1d-UarEm0NjmEW7lYF2eMraHFpk,4017
51
- wandb/apis/public/_generated/enums.py,sha256=vPRi0TPBkZS4D4to-WzhesolsPj3a2B1Okr1xmjRy5k,124
52
- wandb/apis/public/_generated/input_types.py,sha256=vPRi0TPBkZS4D4to-WzhesolsPj3a2B1Okr1xmjRy5k,124
53
- wandb/apis/public/_generated/operations.py,sha256=e_jeUmwP9mnaFe2WUssdmKw2AeN-Iuf3j3FkG9fDnUY,245
54
- wandb/apis/public/_generated/server_features_query.py,sha256=VcqMZTC-_Xq9nhVUv5TD_na6O4PzuDPA2YK6lREOQmM,641
55
- wandb/apis/public/_generated/typing_compat.py,sha256=qV4yFJH1utKxh-vl-U3JhPHyRSqxZ8d8niMSBF5K370,303
56
52
  wandb/apis/reports/__init__.py,sha256=5ZkKvOqwks3UmVeaPuwIZYwyV8_afsGbtybDOXYjSsg,32
57
53
  wandb/apis/reports/v1/__init__.py,sha256=L5l1BqMaap-w5wXCzn8pC8RXB7_72MsjTx8HeLuC3XU,263
58
54
  wandb/apis/reports/v2/__init__.py,sha256=jdLiTqf5QIRxnWH8drWV9Y06bZEhI5N6uGGSKAT7tv0,263
59
55
  wandb/apis/workspaces/__init__.py,sha256=buvqtqELpCFWC7sHzEuslJgx24ZWDj-5lo2mdV6aHdg,265
60
56
  wandb/beta/workflows.py,sha256=bk12HDWnxI4uuP0KyUbfclrTSoRVXrJibAuO_QBB5tI,10239
61
- wandb/bin/gpu_stats,sha256=gUPEzm32MHhTy1J7j5tASxZGkp0pQX7mtY8_hLSY5DA,11027584
57
+ wandb/bin/gpu_stats,sha256=oRYGJZBDYVfIDiqSP1IhsvARkUDFU0if7yjBnDOYxwA,10811344
62
58
  wandb/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
63
59
  wandb/cli/beta.py,sha256=Um35Q9CeoQPS80g0wryo2WtSaD7VA0XqF_YCQNDNiSY,5438
64
60
  wandb/cli/cli.py,sha256=eggnDh2vtLOP44F4VGEJEvoGFZZFzhUaOHtPp804NEw,92622
@@ -73,7 +69,7 @@ wandb/errors/term.py,sha256=kH_noQvl9BDsGBXm7fQfH5u0ERmI6bOZa8Rp49DoAPk,12207
73
69
  wandb/errors/util.py,sha256=EtipkN-l12IT5OAyqhmZyTqYiqplS-Tz5SPp2bjfsJo,1712
74
70
  wandb/errors/warnings.py,sha256=kyLP3bfXSmlztp8nOepLtfTdM-03N-i7Ho1Y568BOtk,57
75
71
  wandb/filesync/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
76
- wandb/filesync/dir_watcher.py,sha256=8dct43c0syDAeFFHhRAetlgH247MItjCCV7gcXo5xVQ,16345
72
+ wandb/filesync/dir_watcher.py,sha256=9Y57-5-hWZFzCuBL6HQVtm-MIE9p8kCRU_g_YZUVHvo,16345
77
73
  wandb/filesync/stats.py,sha256=bjVBoCU9I9ke_XbEqtUgmKJVSmFxxAs2JciBzGWPhDg,3050
78
74
  wandb/filesync/step_checksum.py,sha256=AIz9ALGk59OiR3NtxGj3CJKnsKSQ_1f9PmrbIzxuarE,4690
79
75
  wandb/filesync/step_prepare.py,sha256=CmGgLMbDtwgqC-ooDrLIfWP1NZWAf0Icl3sx3KNXnCc,5495
@@ -113,14 +109,14 @@ wandb/integration/lightning/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMp
113
109
  wandb/integration/lightning/fabric/__init__.py,sha256=RFhr2VkC0D8b6bAXUMZzcVzqDQaoOozfP_HOogAWi9o,94
114
110
  wandb/integration/lightning/fabric/logger.py,sha256=ev8xFbCohaRp4MvrBn1x_5ypuvLe2FJdUWY7Jjw0ylk,27251
115
111
  wandb/integration/metaflow/__init__.py,sha256=nYn3ubiX9Ay6PFxr7r7F8Ia_2dLImb63rpYDmuDlkkQ,109
116
- wandb/integration/metaflow/metaflow.py,sha256=qlzrdkTmT_KTg6sUE3PV8PStuZyFGYEdQzNNNil05jI,11804
112
+ wandb/integration/metaflow/metaflow.py,sha256=E4EVeqpTHdDlqtlZz4sw8x4YEafxzeEBOmfBPPosQFQ,11814
117
113
  wandb/integration/openai/__init__.py,sha256=T6x9lIFfw2NzOSi46wi3otY_-DSwyMgnIjMIOIod7iU,66
118
114
  wandb/integration/openai/fine_tuning.py,sha256=w9_F3bLTG72KdNkD2rAPRdrq6rc58EtltFxwedyI690,18555
119
115
  wandb/integration/openai/openai.py,sha256=zGD1kj6yjDP1IfAAYRcltuXD5Bf569CSnMHMf4rDm54,496
120
116
  wandb/integration/openai/resolver.py,sha256=KrndobRki0C9SNRL8SgPISea0-D7-1g8kvJMcFMN7SA,8164
121
117
  wandb/integration/prodigy/__init__.py,sha256=1-Hg98Y4T1kSNAbrlR9TUrr7dwL1Gxxa-Exu0fsfxl0,66
122
118
  wandb/integration/prodigy/prodigy.py,sha256=RQBptD6zlcyuHM5hHU8XUYUZG3smHPRYtgonmW5tirw,11368
123
- wandb/integration/sacred/__init__.py,sha256=r-jZjCdU8PwBUHyC9n0Y6-ix5UfsYFGE7w8_pizEiYU,5691
119
+ wandb/integration/sacred/__init__.py,sha256=Zzi-m0yRvtbyEgB0N67xSUAWZc1eYBz8Nni3e3OxcHg,5751
124
120
  wandb/integration/sagemaker/__init__.py,sha256=Eyd7rArw0IKGR2ll6zGiTXBi6y7E5nYYuyrVABt_9sI,360
125
121
  wandb/integration/sagemaker/auth.py,sha256=OmCkirtBj9IpHCRD_NSE1ou3TaOT-OqIkJstkuV4X9o,969
126
122
  wandb/integration/sagemaker/config.py,sha256=XLtB9n0cEUe3hA1N9OAA51E9rKco5UZGRteGv_9SCfE,1748
@@ -184,7 +180,7 @@ wandb/plot/utils.py,sha256=Ri1W9_NYQijFcPv9BfGLGMKdQk_AZ8dnw1TiHJ0LANE,6803
184
180
  wandb/plot/viz.py,sha256=ni1MTkt3k1TxSqArdIR_sVcD1cV0j9eKmJdX0QlBcOM,943
185
181
  wandb/proto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
186
182
  wandb/proto/wandb_base_pb2.py,sha256=HjakgSUcYSm6X7B-nvckrmMjaUKGuUh_KE97AuFBat8,307
187
- wandb/proto/wandb_deprecated.py,sha256=CGD8fZII6QOjVPJZ2zCGG8Cx95A5KZVYHYwFPz1Cz6k,1978
183
+ wandb/proto/wandb_deprecated.py,sha256=ycCrPSC0odwFFkhRMpqNsrMsCjfNYSRTj-dMc7KWD_I,2065
188
184
  wandb/proto/wandb_generate_deprecated.py,sha256=Iyf7PwIL_2B7XohrckYLbjjT09lccwbqknxtWelJpJM,1002
189
185
  wandb/proto/wandb_generate_proto.py,sha256=KO1hlAUBGHQRNKsddhcSXvh5a6rmFM3kshKTWftbWwY,1278
190
186
  wandb/proto/wandb_internal_pb2.py,sha256=nIbQS1O4z7n2U0SWGxViHUnjReEBblhht3rZ7pQ0MTs,542
@@ -193,48 +189,48 @@ wandb/proto/wandb_settings_pb2.py,sha256=g5NT8BLXrE1Y7zWMM768zRoW98dBLE9TNuwttXn
193
189
  wandb/proto/wandb_telemetry_pb2.py,sha256=ReY9N2qSt46o7m1ou1khvYSYYImrzo5ro90g_9m8QsM,322
194
190
  wandb/proto/v3/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
195
191
  wandb/proto/v3/wandb_base_pb2.py,sha256=_nsr_HW4Fdz62-KiXGo6Dw0_9bwdXz07auZkkH2QfSI,2355
196
- wandb/proto/v3/wandb_internal_pb2.py,sha256=X5m7_mikGKgCRpD84q311n64q6-HMMFMhVop6XeGgHM,116112
192
+ wandb/proto/v3/wandb_internal_pb2.py,sha256=TwAdOUWcpTeTsh2JDo6sxcyYbC_ZXX9T-L_wjwjp5To,116533
197
193
  wandb/proto/v3/wandb_server_pb2.py,sha256=LQCUmuy0NFDo20N9uCUiE5VPyhP5r929WKtniae4CdQ,15514
198
- wandb/proto/v3/wandb_settings_pb2.py,sha256=jBSvc-eCXhQxGf6NI21atkwrmrerYePyo5yhFUAmTHE,21146
199
- wandb/proto/v3/wandb_telemetry_pb2.py,sha256=tQQJ7s3fbFYJc60JLnrkdf8yxCqKK0zJMdJaXxA3phM,13805
194
+ wandb/proto/v3/wandb_settings_pb2.py,sha256=3q2U2JdfaGE3iVi4IgvVGkMjDjgO8VKWvFkpoYqQzak,21197
195
+ wandb/proto/v3/wandb_telemetry_pb2.py,sha256=ZznK36COb2ZzWhi55YXKbg-WZxPYJLtrafN-7ENCSJw,13909
200
196
  wandb/proto/v4/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
201
197
  wandb/proto/v4/wandb_base_pb2.py,sha256=El5lnZMRpkc1W4SlZgeKoDPqWiJau0BG8QfigMMt6bM,1422
202
- wandb/proto/v4/wandb_internal_pb2.py,sha256=LV8iKL7ufkIFyNjfG4PjDbg_u2EoGbYuMG7Qn5XaABw,53151
198
+ wandb/proto/v4/wandb_internal_pb2.py,sha256=SlqrjYQ6m53PrIthGy0x6-i7tNsZGJQ7idQLNJ6t0_c,53393
203
199
  wandb/proto/v4/wandb_server_pb2.py,sha256=NiG6Xv9CDvxc8v8NmnM237jyYqMuRyE_nqmLAMZVdI8,6977
204
- wandb/proto/v4/wandb_settings_pb2.py,sha256=_5Vak6XbTRCRtUp22T7xdwqNl2RMdDTgQlwWyXsDhdY,17521
205
- wandb/proto/v4/wandb_telemetry_pb2.py,sha256=sd9xk2SdYw3SBOYw77GT5pFa0YHl9v-P17-854RMMxg,11252
200
+ wandb/proto/v4/wandb_settings_pb2.py,sha256=pOU_zBpU8XqL6dCKboUSaqeSXB4MSn4year4z2wzBX4,17572
201
+ wandb/proto/v4/wandb_telemetry_pb2.py,sha256=6QtWG9fzlAJyzzDtk-sWv6LetC1cYBatceKEu4TmepE,11356
206
202
  wandb/proto/v5/wandb_base_pb2.py,sha256=u7VVWdExJ4WDkYNMV-xvWvyQ-NTIbAqToTKtgEqt_Lg,1572
207
- wandb/proto/v5/wandb_internal_pb2.py,sha256=FGXDcOhWMVRkFTBF0uRSOi6YG6rJciJi0HuMlxbxmBI,57498
203
+ wandb/proto/v5/wandb_internal_pb2.py,sha256=S-raVx_sCQ-OmOICRjCacmEIZz83XGXHknXGjHDY9uY,57740
208
204
  wandb/proto/v5/wandb_server_pb2.py,sha256=f55mk3Z77xUnq_tAbTd_Q75CiOJz6ppPbZT7CoeF7-I,7535
209
- wandb/proto/v5/wandb_settings_pb2.py,sha256=iEaW53am009xLqmyIBvx9YH-JvvaJfGbf0up9X1zlkg,17877
210
- wandb/proto/v5/wandb_telemetry_pb2.py,sha256=tYmNIX5Mtw2AybnuX8n9KbwdxT2w4GKPvvokFfKfP8M,11522
205
+ wandb/proto/v5/wandb_settings_pb2.py,sha256=7B1GCfP7mcfaxhXlP-QcmOEPjKkDeTml9cVgDTGOF1c,17928
206
+ wandb/proto/v5/wandb_telemetry_pb2.py,sha256=d6qjnhCbLT5fdTwvhKoHtqaLop0Kbbh6eppcb4KjoSg,11626
211
207
  wandb/sdk/__init__.py,sha256=N-GTAC0AzbZF2J8RzB33DTmYk9u-jubllCwvhWrPgsE,813
212
208
  wandb/sdk/wandb_alerts.py,sha256=SwBPBiXRxknMTMGbsVoMMWqWK65UWMcKAdTWZtdwAeo,193
213
- wandb/sdk/wandb_config.py,sha256=b7kxQVnIh5HCBZXb2pOGZ4c02xCVlW4IQiAu3N-8Opg,10856
209
+ wandb/sdk/wandb_config.py,sha256=imPAYPMzP3rdhmYCK6zYWeZ1GbM7Jw5w7_PTJdoRmcc,10693
214
210
  wandb/sdk/wandb_helper.py,sha256=IbJ7opO8UkfwCDekSjRYIrGBblUxnTPBfp1EdesfF4U,1824
215
- wandb/sdk/wandb_init.py,sha256=hUlqaAkN7GFIyo7fqYDX6GLki6nZ-v5sWaJU6z6tjgw,59407
211
+ wandb/sdk/wandb_init.py,sha256=agD3n4OQ2MWIgtG7OEPCOYM9VDJAdZDl0rts3k_SPhw,61404
216
212
  wandb/sdk/wandb_login.py,sha256=-2EvouBgBOQgOnh_8Hlb5LCGg6Lz2OR44ieWRRgqdZ8,10952
217
- wandb/sdk/wandb_metadata.py,sha256=NLMhT6pp16JqzKatmqeo9utEcZfil33mC7vBOxM1Ato,20218
213
+ wandb/sdk/wandb_metadata.py,sha256=wnwKexZjJ7zRtAcfYbOZtaoK6rfVVU5oKDWT5-UerOk,20886
218
214
  wandb/sdk/wandb_metric.py,sha256=a3GiQXr6H18m81uobYjlJaC8CL8iANzI42qxkxfZsDs,3268
219
215
  wandb/sdk/wandb_require.py,sha256=eYDG4oH1pwik2tHJVEdvivo4R2RypoITFomgr7umMhc,3104
220
216
  wandb/sdk/wandb_require_helpers.py,sha256=ZmKv5aXXHDTTU6nYHMLKW4_pt9X-PlaMtbRJl77kHX8,1331
221
- wandb/sdk/wandb_run.py,sha256=LRO84qI1dVnT5f3PujRTW_THyH3BF13YtAWNKscnv2c,153637
222
- wandb/sdk/wandb_settings.py,sha256=zECCD-UV42WIgwhqzxK9_f5QjF7p88-B4xGDe-JnErM,55457
223
- wandb/sdk/wandb_setup.py,sha256=L6xoj3DWS_F_jfQ-i8kISd443AiqO8OSaOx8yCXSWzE,12938
217
+ wandb/sdk/wandb_run.py,sha256=yNcr8j5Q5p1EmWCP5tMQpSQhnlQz7lu7pCtDSedrakI,156504
218
+ wandb/sdk/wandb_settings.py,sha256=UTLsLl9ho4lOJnI0ZAPmkWC3h4zJo3ZCvKzqHLQjJnI,64047
219
+ wandb/sdk/wandb_setup.py,sha256=bznp9ZHKofN2B_J34m-oAkf1bFg1M6MeW6UhiEpY_Ks,13010
224
220
  wandb/sdk/wandb_summary.py,sha256=yQdOVIPrZaZanhBQ7yuSfPLX0x6dxwkN_KAn4SgjSZU,4536
225
221
  wandb/sdk/wandb_sweep.py,sha256=Sg_JqxVzmjUBvii41azpdr-c6RPwHOBnSha8k7jrRhk,4028
226
- wandb/sdk/wandb_sync.py,sha256=tcXTzfCs6j0CQ87vBx_8de1de3PuI9jggM8DOkvQIlk,2207
222
+ wandb/sdk/wandb_sync.py,sha256=GNC8xYqT56ws4aw6Vd_mV_Ck-InEbqnyioSqHIybDug,2106
227
223
  wandb/sdk/wandb_watch.py,sha256=F7S9CLbw9PddUp1qBjPRKsOiVFU8LPaq6A9taV3TJKI,4840
228
224
  wandb/sdk/artifacts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
229
- wandb/sdk/artifacts/_graphql_fragments.py,sha256=fHvyiABSU5RDa0--FN9D9Dfra_t9vnNzb2MD-S4JYzs,3080
230
- wandb/sdk/artifacts/_validators.py,sha256=QlN8yn7muPn1NZMSslHVYM-mcGEEd_4XLfRdv3Wf91Y,4171
231
- wandb/sdk/artifacts/artifact.py,sha256=HcNzpoj6jxCpz9rc_IokU7lWx1fd51rXDIy2-CgmDvk,90181
225
+ wandb/sdk/artifacts/_graphql_fragments.py,sha256=mH_wkTAZz2Qn-rlkaq84XZIQ5x3zLF0bqRaTuM5mbXE,2926
226
+ wandb/sdk/artifacts/_validators.py,sha256=p74NPaBhRPJ-Z_2o85RdGcfDM8Tqok7DrqmIvuUbN9Y,4216
227
+ wandb/sdk/artifacts/artifact.py,sha256=T279vMZYJsHH0fcdpzwAJqQoJwXsQkn8W3q_deamC8w,92458
232
228
  wandb/sdk/artifacts/artifact_download_logger.py,sha256=XEVxmktMInrd1p2z86CKt3QyTA9vscxJL5-8_eupknI,1501
233
229
  wandb/sdk/artifacts/artifact_file_cache.py,sha256=503z2x4CoTLXk_IfC9jRKBU9fr-X9PMiyG66h0INzQg,9744
234
230
  wandb/sdk/artifacts/artifact_instance_cache.py,sha256=atSUDvKjkp6by6QY0w2FLMrFUg9UKc8TIrMVWGauV64,501
235
231
  wandb/sdk/artifacts/artifact_manifest.py,sha256=YYC6o2OooTcVfkzqQIIVcxVxqdSN9TELG-qWgaSI2Qo,2545
236
232
  wandb/sdk/artifacts/artifact_manifest_entry.py,sha256=jskWFLdJ81Y6Z_rNbHLZu26Os2EE2BUPzKx4o6GMw-4,8355
237
- wandb/sdk/artifacts/artifact_saver.py,sha256=2x_n6KKSdntz7XBIDzQt3BqTXQgxfpyHe4U0GHe0nI8,9329
233
+ wandb/sdk/artifacts/artifact_saver.py,sha256=Dw8gT3NVMlyoVuDkIpROYdZMU5rGduVZzRTFSOnebak,9718
238
234
  wandb/sdk/artifacts/artifact_state.py,sha256=6fxISSPsK62VeMau_5CI4a9lQXTaUJXcruzk74WUIBE,236
239
235
  wandb/sdk/artifacts/artifact_ttl.py,sha256=kNRr9JZ5P2iuo0ofoNKlSddGJkIQXRUjwgO8YUdyN4g,86
240
236
  wandb/sdk/artifacts/exceptions.py,sha256=h9tSi023BhW85_FfjLYjeNtlVgUTqZRfnYM9WgdAMbM,2023
@@ -256,29 +252,29 @@ wandb/sdk/artifacts/storage_handlers/wb_artifact_handler.py,sha256=MVLbOi-py7Fbw
256
252
  wandb/sdk/artifacts/storage_handlers/wb_local_artifact_handler.py,sha256=JgQVx0G5wocuycEJXxXKlXAl6B7fRVlrvlBEX8_BzP0,2552
257
253
  wandb/sdk/artifacts/storage_policies/__init__.py,sha256=bgpWKElL-3iHcLO8pF-L8oezG-dQbp_6vcCYo7CEFAU,226
258
254
  wandb/sdk/artifacts/storage_policies/register.py,sha256=xT7kUxubtLqyE-9S6U9E4mCo1PtXl0ZEJ6gVQiS-kGQ,49
259
- wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py,sha256=m2s67vmRySPw0bkI69HoIUTO2hyYAo-Nl22jv32Rl90,14002
255
+ wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py,sha256=VMy9B6Wm6I4WtNtGOTb4QGIDSlRVoVIakG0UpFNUNqo,14883
260
256
  wandb/sdk/backend/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
261
- wandb/sdk/backend/backend.py,sha256=taUs81J5U4XxTkl4TuMDz8YUxlt8I2FZqhHFV7XHBxY,7488
257
+ wandb/sdk/backend/backend.py,sha256=w9aV2IEEjUFjjAzAe8UX7Pfz49hp3dcI-fXRhpQWq9A,7791
262
258
  wandb/sdk/data_types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
263
259
  wandb/sdk/data_types/_dtypes.py,sha256=NdRfTto95YxLMMK89PONQvAz_Jaey8D7caTsZ_OlvW8,30092
264
260
  wandb/sdk/data_types/_private.py,sha256=zp2NRarTlIq4Hk3R2xp7j_qPGNzBMnaGHrZUN82shaY,299
265
- wandb/sdk/data_types/audio.py,sha256=VcbZRjY-9szxIKR-x1Eg375H1Lm5R6znErT4Q8FYrWM,5364
261
+ wandb/sdk/data_types/audio.py,sha256=q6buCUd3NYmHLs6UaQwzU0eMULKigofq4WOlIJKGg8U,5305
266
262
  wandb/sdk/data_types/bokeh.py,sha256=0Aq4CJECPpoAkl7vw5LCGYhdCb9pS8i46jHB9PxTj1k,2534
267
263
  wandb/sdk/data_types/graph.py,sha256=_P_z2dNj2mCJeSvusIfVHkSkVB9AL0AEO-6KaguC1ys,12089
268
264
  wandb/sdk/data_types/histogram.py,sha256=syx3fUDoP6vYqA3r5m7bHXK6MAWo2MkRhpCKasihwvg,3145
269
265
  wandb/sdk/data_types/html.py,sha256=CBgxEybPGDY6vpQ1fHoKtQfGNGuqSkFQtUZF-qG9p6c,3504
270
- wandb/sdk/data_types/image.py,sha256=o_ZwqB4uO1-R8QLuO9J6aiUuyANMz2qPQJAskushozQ,31239
271
- wandb/sdk/data_types/molecule.py,sha256=lTScaG-_B-eGh20bRfLNA2fECDYQOuD0zdnGvQjEI8o,8583
272
- wandb/sdk/data_types/object_3d.py,sha256=xQw-9mq_cXMxXoW817BiByh7C5wvMuC6jliJEzKEAPg,15948
266
+ wandb/sdk/data_types/image.py,sha256=L9FiBr1OL_e2bJlqoWS84Cz9CJR4zRCQCsbSemGBpcY,31792
267
+ wandb/sdk/data_types/molecule.py,sha256=rTvaAfHLYZ3u7TZsHxEClKHA6fwNl81iolAf74KwIvQ,8490
268
+ wandb/sdk/data_types/object_3d.py,sha256=6WlsiefTzDErb84M9mVRE8aN-GXFoX1v3S4YReVOzR8,16002
273
269
  wandb/sdk/data_types/plotly.py,sha256=LEIHRqtHHpSou04AEUsyH8_UMXP4adrEHHaKoEtQkxo,2913
274
- wandb/sdk/data_types/saved_model.py,sha256=p9k92Jc6ihqTjir-GesTKpPiC92m6nen0zqUEcw0W5E,16254
270
+ wandb/sdk/data_types/saved_model.py,sha256=XtlrAp8PHb4Pd8-xG97I_iSvFe8fme9iHmHMR9G2KG8,16212
275
271
  wandb/sdk/data_types/table.py,sha256=DjppFy0-ii1iMoVOg-h_F1YEUbgk5wGPJDXlQURXLQw,44683
276
272
  wandb/sdk/data_types/trace_tree.py,sha256=HHJE6QHmlfvWE090OMW6CxMebMT03rF_qnTSGmX0-EE,14700
277
273
  wandb/sdk/data_types/utils.py,sha256=Qdj-gUOrrc9wVjBD7Mtmt2o6bHadr1IgzAWF4ijdrz4,7633
278
- wandb/sdk/data_types/video.py,sha256=iug92X8xW5JIquk1QvMom8-N9SKqEO3-dv9vxA28M_s,9344
274
+ wandb/sdk/data_types/video.py,sha256=pfgxOEfx7tGugyjpO5fVM7X_yXE4_wjMbIx1XH84Y9Y,9252
279
275
  wandb/sdk/data_types/base_types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
280
276
  wandb/sdk/data_types/base_types/json_metadata.py,sha256=oKpimndUQvDW30n15tb2pbyyyKhhWwm8wsYNBEot060,1553
281
- wandb/sdk/data_types/base_types/media.py,sha256=QO6aQdNGYtwr6qZYUQ9-WbTpkB3adCBTXWSqO3ZCJjI,14284
277
+ wandb/sdk/data_types/base_types/media.py,sha256=z50vmzEbWgiCyp8Wjt7e-8-AzQeGPlo-O0N8pcQuKVw,14468
282
278
  wandb/sdk/data_types/base_types/wb_value.py,sha256=HtV2VblGg_nnSBQ1rRSjlRuY-NyDTbeDVCJzlQ49SUc,11460
283
279
  wandb/sdk/data_types/helper_types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
284
280
  wandb/sdk/data_types/helper_types/bounding_boxes_2d.py,sha256=z_cssRDjzWSoDopThg4W3oUQ6Cy17RvrO4WhB4Ha_V0,13272
@@ -289,17 +285,15 @@ wandb/sdk/integration_utils/auto_logging.py,sha256=sblNn9BIG7memxTj23UfrGXyA06t3
289
285
  wandb/sdk/integration_utils/data_logging.py,sha256=DDFtDaUu50aeTTgxCHHYd2f85guqqf2xfEOburRlwwQ,19533
290
286
  wandb/sdk/interface/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
291
287
  wandb/sdk/interface/constants.py,sha256=NJNBFr7LkshLI837D3LU3JuEURLzBwza9H-kxcy4ihw,60
292
- wandb/sdk/interface/interface.py,sha256=tgNdWliszE1hfuztGXaYJeDZ6JiyYW0YhYuC1nB-CNI,37161
293
- wandb/sdk/interface/interface_queue.py,sha256=ebHJQSwGiA2N2CRT5ubDttyjZfpbi9ZXzLQIx87GBsc,1512
294
- wandb/sdk/interface/interface_relay.py,sha256=lQEetJfxFk1T1kEo4uec_6AufG1W8BULhTFtpyC0Nt4,1317
295
- wandb/sdk/interface/interface_shared.py,sha256=cvDM-ryAzRKf6rBqmG3TH6Yc135ZcIqjsmhGpzv6wzE,18543
296
- wandb/sdk/interface/interface_sock.py,sha256=K958Geq33GVUJvpzTruLO1KJp6Qb_3MjYcCahkOCwEY,1321
297
- wandb/sdk/interface/message_future.py,sha256=5OMApIUKTXLHP98ph_jCdshuPZB_E0Uf3UGZpOQ8sik,685
298
- wandb/sdk/interface/message_future_poll.py,sha256=drjrcBKswYPYJJQLFlj7UDXq7_zg7KNcObFVetsbvhQ,1410
299
- wandb/sdk/interface/router.py,sha256=A-9ctBt9LpgX4uK8VMgibTnp5QSte_zn3TOb2hEtQlc,3596
300
- wandb/sdk/interface/router_queue.py,sha256=N2ky8m4fcLOGuNAiEuCbTcQtkzo1F5ZUWrmJoNqZu44,1046
301
- wandb/sdk/interface/router_relay.py,sha256=imeQIhQinZ5FN_yUvN_OUXhR7Z0mSYCNhaB8voMoaqk,958
302
- wandb/sdk/interface/router_sock.py,sha256=BOnSoKu0Di8lIjdL0Z-F1RiSRRpZMSHwuInF-sI0wP4,1085
288
+ wandb/sdk/interface/interface.py,sha256=oRyy1hs78rZyjf3jn7pHKAqZUA7Swz25I90yDmuJ4vA,37716
289
+ wandb/sdk/interface/interface_queue.py,sha256=zOjUB2nf_wNMoEeXGWP5di-u4-UL2_jvsVZw53EbEsk,1257
290
+ wandb/sdk/interface/interface_relay.py,sha256=lM5ZgJc6m8BS-nWOuzMCwAK0ve9UBGgOD2rgJ9flIhI,867
291
+ wandb/sdk/interface/interface_shared.py,sha256=XmEg78CyUHI4XPNtgyx59aQIlzJzQjZYEIeQVECHXSE,18645
292
+ wandb/sdk/interface/interface_sock.py,sha256=rz6uwKNzLpGDCCVGSlC6ABspmkYD5LnGX_cd9dvaAwU,1043
293
+ wandb/sdk/interface/router.py,sha256=hlYzz7DjDt3sYDJQYLE1NHkxGFDezHgmfznnPRHPcHU,2630
294
+ wandb/sdk/interface/router_queue.py,sha256=wnApY1iXa_3L0-HO6t42Nbsh68zyEw3FWN3oI6PmoXM,1049
295
+ wandb/sdk/interface/router_relay.py,sha256=tOBQYEX7PGw9-qE91CdNhLEQb9KbJiQnlHg7XBOsijA,1390
296
+ wandb/sdk/interface/router_sock.py,sha256=S15CRwMoDFYKnEzwqAjvPQC6BlKtJpxE_rZHeaxM2V4,1002
303
297
  wandb/sdk/interface/summary_record.py,sha256=-wDv_zLYueeUY8IzyF9NPYnYwF3iBpUWbrsGcHAd2YM,1677
304
298
  wandb/sdk/internal/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
305
299
  wandb/sdk/internal/context.py,sha256=dnmsKEoQ2xOtmXM5OBVHdyGO3XdTQrqjxfb1lRADSTc,2518
@@ -309,19 +303,26 @@ wandb/sdk/internal/file_stream.py,sha256=Dnqayrbqa6L2sG03rX2oIFhWT-IME5vl5JTykyL
309
303
  wandb/sdk/internal/flow_control.py,sha256=3LJ-KatyPPH18G7TfSMLDk-BE5tankB4JRhQqLoUOWM,8585
310
304
  wandb/sdk/internal/handler.py,sha256=zquR4k6Ah5ohg9n5XJC_bKAulz3aSUuMjUlHqiIOfFE,34280
311
305
  wandb/sdk/internal/internal.py,sha256=o9_-6C99LMOx_YB5xuZ1I1KgpeHk3jUgll63sEe9gCY,12182
312
- wandb/sdk/internal/internal_api.py,sha256=R9UF5K2xrSGEQ-46Ch_Ab0CW6EHouOHCk7Z1GVyyRIM,160084
306
+ wandb/sdk/internal/internal_api.py,sha256=HXXCtbJ7-N-w7889zwKGRKgzeN-dDSsdxpdLtzPnSZU,163705
313
307
  wandb/sdk/internal/internal_util.py,sha256=dKawQKw4UdZUE6Bmme5baT0FGDC0IUgoD2zM6ey4oBU,2555
314
308
  wandb/sdk/internal/job_builder.py,sha256=_L-DfGFEzDl0JLXPo535KPK3Bk0vqXsTQVF7buXeaiY,22962
315
309
  wandb/sdk/internal/profiler.py,sha256=C8-mPrAhPS6_l-Fj_zCOwmXaEqhjI9Wd0EAPXS9GAEI,2348
316
310
  wandb/sdk/internal/progress.py,sha256=9nNAErIXzJoapctsfqWYhESNPzFkXAu5TVAnaKsIa4k,2339
317
311
  wandb/sdk/internal/run.py,sha256=8OhVy2vfgPC8pNFq0tJ4CkQHETOBfQsFDghw50ccSXc,682
318
312
  wandb/sdk/internal/sample.py,sha256=USAWLhEeP83J13BVOSIy1Rb3kDDKTK9kzR88SlhO7dw,2470
319
- wandb/sdk/internal/sender.py,sha256=QWrtB2AymPyMqQ7EAjdXvOVhwG6p0BjyR_xmP_Q-n3w,64472
320
- wandb/sdk/internal/sender_config.py,sha256=qEuXwOskca3sYyDIRsswlXmj9StCCS0WKQ1qrBXbIjw,6767
321
- wandb/sdk/internal/settings_static.py,sha256=1Dc2MoVCIskHo7rKafjtrVxY9sQWZMNeee39LYBe9t0,3668
313
+ wandb/sdk/internal/sender.py,sha256=kU1EkHkX0S-xKRDjbk7Cj3K2ivVvscrNPsn8-RpLQnk,64588
314
+ wandb/sdk/internal/sender_config.py,sha256=3g0eAgmYJtv7tPXlqf7qayqKj-JOclvVRvzbVi8wfR4,6711
315
+ wandb/sdk/internal/settings_static.py,sha256=-3pJ5RCWxGbNVVE4RvGudTurpn8ObN8wTEx_4dkrS3o,4284
322
316
  wandb/sdk/internal/tb_watcher.py,sha256=3AvQGlZLGkr8POhaSGzbeyIcX0YWkLkblJ0bksAB8U8,18738
323
317
  wandb/sdk/internal/thread_local_settings.py,sha256=UqD6kfjsy6mvxIWcjhd-vJWkNRCeU1whuRe_-VGIklQ,527
324
318
  wandb/sdk/internal/writer.py,sha256=jo_Ex7ik-0_rIahYDHIWcQjm_uxsUNVn92__bI8TloE,7267
319
+ wandb/sdk/internal/_generated/__init__.py,sha256=DRBzTksuUuVY9oRJQsLtRTMMItykywWYvDxktHbCfk8,381
320
+ wandb/sdk/internal/_generated/base.py,sha256=yA7q36_FISIjDI9S_1AYPr3HyijzbITxaWBLhbz4w6I,7441
321
+ wandb/sdk/internal/_generated/enums.py,sha256=vPRi0TPBkZS4D4to-WzhesolsPj3a2B1Okr1xmjRy5k,124
322
+ wandb/sdk/internal/_generated/input_types.py,sha256=vPRi0TPBkZS4D4to-WzhesolsPj3a2B1Okr1xmjRy5k,124
323
+ wandb/sdk/internal/_generated/operations.py,sha256=e_jeUmwP9mnaFe2WUssdmKw2AeN-Iuf3j3FkG9fDnUY,245
324
+ wandb/sdk/internal/_generated/server_features_query.py,sha256=WGjefMl-SDR7o7OVtJk7xKEYICBFA2wceGMXbuyj5QU,647
325
+ wandb/sdk/internal/_generated/typing_compat.py,sha256=eLkBJ3Q5MqL3jn-EhIpGm_9F5By84KEJ4fvIpYL3Qhg,302
325
326
  wandb/sdk/internal/system/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
326
327
  wandb/sdk/internal/system/env_probe_helpers.py,sha256=jIy6gbiaq37SzgcBGe6GepJho1VS5qNQThqOmu-g0OA,397
327
328
  wandb/sdk/internal/system/system_info.py,sha256=51hOq_WBLdDDTX6aFUkcEnRCvOSNZcKGuwx7-ODC4iY,10072
@@ -392,10 +393,11 @@ wandb/sdk/launch/sweeps/scheduler.py,sha256=yPHVT5GD1my_2tlMqGsLOBL3Q8pAxCcxLoFf
392
393
  wandb/sdk/launch/sweeps/scheduler_sweep.py,sha256=beM0go0u9xRcXW20ED8MyTqq4ol_O5_4PODbA7FpN88,2987
393
394
  wandb/sdk/launch/sweeps/utils.py,sha256=TbsAvn3eQnp5js0SvMmPVoP5vaT5bXMPL-ISWyZvVlQ,9837
394
395
  wandb/sdk/lib/__init__.py,sha256=53BA5lIOtUXciXZcPpNsFbp-iUPzI5gQFxplTykNmOE,183
395
- wandb/sdk/lib/apikey.py,sha256=a3m7IaNyx7_yP-rB_PI-DiVJgtv27gqapvYYHgogx_A,10508
396
+ wandb/sdk/lib/apikey.py,sha256=ygJ6VfgSP5ZQNEw8_NmEN4S5j3VfREyr7iMiqi4OKzg,10444
396
397
  wandb/sdk/lib/asyncio_compat.py,sha256=AokSd8OXVYF1ALggThRr4RrgD4UiUtDjP3LRE4HLEm4,6584
397
398
  wandb/sdk/lib/capped_dict.py,sha256=T2CNkmzfWeFgXOsFmx5IDPzgYmP1jxaVAb-nn3-qBBE,820
398
399
  wandb/sdk/lib/config_util.py,sha256=YdYvk-KbTdTa-84XegpvbqMuQfdlOREFiVR7m3q6e3Q,2917
400
+ wandb/sdk/lib/console_capture.py,sha256=eC1bThCHa9DiQTk21ALdK8w1ETeydBpiTOYKN31EjVs,4923
399
401
  wandb/sdk/lib/credentials.py,sha256=WmVdzL1rFy7S0WIHf1ZYd98_eaHTxPKUobReRSPQgBM,4737
400
402
  wandb/sdk/lib/deprecate.py,sha256=r8kT4UY2DSG7V5N78MG46kQOzycBp3_8lH49KnDXxXM,1462
401
403
  wandb/sdk/lib/disabled.py,sha256=zIuxC4uddcjAGCU-PEQGgoe7FzxJTHho3llW0R1uBX4,884
@@ -419,29 +421,31 @@ wandb/sdk/lib/preinit.py,sha256=11QkGmBpoGazNaUGvyDIzBJA4QTggj7fGQdugpCvOiw,1450
419
421
  wandb/sdk/lib/printer.py,sha256=v3T2wqere8HQIVba9LYd8xCIUtV8e8l21Fw4XUEPbXM,15289
420
422
  wandb/sdk/lib/progress.py,sha256=GakG9R_mJxVx8zG8atVHKydDZt0uRFEPwls7b3hUpsc,10652
421
423
  wandb/sdk/lib/proto_util.py,sha256=JxDldi8j6dfF_Lx9zOwqYL6LQZhYYGgKt_AfZtYHAW0,2894
422
- wandb/sdk/lib/redirect.py,sha256=eTNvQp4SFLgN5Kxwmh1BP0du8_7YZxcgoXsmE_2IHSM,26132
424
+ wandb/sdk/lib/redirect.py,sha256=-7iM7Bg12LzsLivkkVL3wO1pT4ofKeShz1L6NTHoSVc,27249
423
425
  wandb/sdk/lib/retry.py,sha256=BAgPgJb1SxRguor2KTfOFnQpTlyNE4M_X8VcEc7YfMs,10173
424
- wandb/sdk/lib/run_moment.py,sha256=HMx8jYIaWwmSPNViP54ZYMbxLKBn2m_ezYCa67UC6H8,2464
426
+ wandb/sdk/lib/run_moment.py,sha256=jMkfNAHOrYHlb6_nisqjfMu2Onm1bDy6lIubOJ2WxCM,2421
425
427
  wandb/sdk/lib/runid.py,sha256=Qa-5ft4B85YUazkV_18OYwf9JhMaAVp0JAInZzxzX5o,392
426
428
  wandb/sdk/lib/server.py,sha256=dn0RXjAnpmigAULLyq75XHYyHxw17MLCJ9fhPq2v22I,1623
427
- wandb/sdk/lib/service_connection.py,sha256=maeOfllw8Bas3Msa3OGl0g-zq_3rIv300di9SkiMonk,6500
429
+ wandb/sdk/lib/service_connection.py,sha256=pf1T0isbDwTkv9MvuIpuk1MHTVkCebK4N8n9pXF4SSg,7455
428
430
  wandb/sdk/lib/service_token.py,sha256=c4olk_g4Suo0zTTDaoR78oHLwpt3NgDzQeF94raigME,2471
429
- wandb/sdk/lib/sock_client.py,sha256=znIOP7ymfMF_UDBJR1FALLQCZYeNbydlOl3jUOlL9Xc,10316
431
+ wandb/sdk/lib/sock_client.py,sha256=YzR4uFMgFBA95ySl9h2QwIqXKMEAyb36Y09vpf70IPU,8383
430
432
  wandb/sdk/lib/sparkline.py,sha256=9xQkuZ0z1DM7rHE2jFNDy5vBdPirnurd__A_bC6-Bgc,1363
431
433
  wandb/sdk/lib/telemetry.py,sha256=Q7rh7qyeicy9KdgQffVK6ZI8vHUdOtwFvxEeeJtOWcI,2754
432
434
  wandb/sdk/lib/timed_input.py,sha256=D7vx3N3m4LlxwkQpzS7lDi7IOnccqZ3WfLlDYRs8Oes,3229
433
435
  wandb/sdk/lib/timer.py,sha256=VZUyl7fmQAB75l9QXw9Nkz9SxyYj-CvLPawAk866YKM,440
434
- wandb/sdk/mailbox/__init__.py,sha256=3hKYgxzS3Fz7ECXbWxeku52pvtQX_NMD_iE38x9RVPg,849
435
- wandb/sdk/mailbox/handles.py,sha256=M1fgYl82u9078hO8daURkM3xAuSHt2UES5CWrGQf5aA,6223
436
- wandb/sdk/mailbox/mailbox.py,sha256=4wqfZSbFRO6j5Nr6JoSmtEnm9xOLXCVcoR59rxP5nB4,3641
437
- wandb/sdk/mailbox/wait_with_progress.py,sha256=i1wvRq_Lyvhb8FCn1HP8AxXnBQ8kg6rlutz6kwPQaEw,4198
436
+ wandb/sdk/lib/wb_logging.py,sha256=9km7TAnJRSwBTQaFSYA4BmQZV3_Gb0y1PBlzqEOdUHA,4839
437
+ wandb/sdk/mailbox/__init__.py,sha256=C4x3ym_hymymKFZiI1Q0HPlGx5x5CzeKPVF229Z2V8w,856
438
+ wandb/sdk/mailbox/mailbox.py,sha256=jUEP2Ltgi9rCZmiDtfb5ZESaqwB2wH-sy72OLRvATZk,4314
439
+ wandb/sdk/mailbox/mailbox_handle.py,sha256=YLKNMkytIAoIpMuV5Nu_g04JdTGzm8P9d9iw2EPJhV4,3743
440
+ wandb/sdk/mailbox/response_handle.py,sha256=kZWVvVbpc92u3u23FTwAod-QCPFENUYREpjXm-wkh6Q,5031
441
+ wandb/sdk/mailbox/wait_with_progress.py,sha256=O3yGgUC3ztpfmdLk_CyirJVeXiCzF8nuDI2CqNiak90,4126
438
442
  wandb/sdk/service/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
439
443
  wandb/sdk/service/_startup_debug.py,sha256=A6P5iCYRoNGcVulziUc7Mw_Y2MCG8PAIAUVjzzMUTPs,602
440
444
  wandb/sdk/service/port_file.py,sha256=_VeftD2LjNBeQH6nfzd9Dyq2u4fT2YRLE7D1ikIhYmU,1546
441
445
  wandb/sdk/service/server.py,sha256=_0MAn4Y1hs2AnapyQb5heYiixKQNxFENfkxIGoiiIB8,3460
442
- wandb/sdk/service/server_sock.py,sha256=RpCQ6MwK_N7e7Nbf7sUeLL0NQa8UPLSyTUxKgglYz50,9705
446
+ wandb/sdk/service/server_sock.py,sha256=HiuncxAaNHUpii7CCIvYr1XVirBHNE7d3WBK2tVVsxY,9695
443
447
  wandb/sdk/service/service.py,sha256=tfw6zDHLv9q7E_W25g5eddC6nuyvFPLZvpsTZv1SBM0,8614
444
- wandb/sdk/service/streams.py,sha256=yniOVA0jCXCKaDl3omK9y4Gan80YT8d7az9vv67LHyk,14528
448
+ wandb/sdk/service/streams.py,sha256=lNtn6aujS5GlhzsG1Uw3JpF7GL5tWewN1dW39uP5hFo,14801
445
449
  wandb/sdk/verify/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
446
450
  wandb/sdk/verify/verify.py,sha256=Bz7DKW7zmhNAPJQNAAQszAmjBilxs2WJJ9rVHRml6LM,18299
447
451
  wandb/sync/__init__.py,sha256=BzzKmWKjsrCTBAj24d-lJOEPkFUmTfvaK2MeXplSwcc,93
@@ -830,8 +834,8 @@ wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/importlib2.py,sha256=cJIaJ2EQso
830
834
  wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/platform.py,sha256=UORYTNVcUSE2NpFcq9UVLIS-tsS0TS_Qw8akhKxn2eY,1506
831
835
  wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/unicode_paths.py,sha256=UWX8DB97ygkEeSxWQUYCHR4MahNilux7vl5TCTQtPPk,2190
832
836
  wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/win32stat.py,sha256=ZOevOTbSo8NRiIxkuBVGaG4yigWnPoO0goxAi-jsBkM,3828
833
- wandb-0.19.7.dist-info/METADATA,sha256=D0M_YGEl-LeOGCauFiR9UIHLGAR2dFrdaA-OTS4USSI,10289
834
- wandb-0.19.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
835
- wandb-0.19.7.dist-info/entry_points.txt,sha256=v4FCOZ9gW7Pc6KLsmgQqpCiKTrA1wh2XHmNf-NUP1-I,67
836
- wandb-0.19.7.dist-info/licenses/LICENSE,sha256=izOKRJpGOx1PrJiGOKR0HsNdlB5JdH2d0Z4P7a7ssTc,1081
837
- wandb-0.19.7.dist-info/RECORD,,
837
+ wandb-0.19.9.dist-info/METADATA,sha256=hQPy6-U-nKSNrjYz0d2NB0dr01ietF6Rh61lQAiLSps,10258
838
+ wandb-0.19.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
839
+ wandb-0.19.9.dist-info/entry_points.txt,sha256=v4FCOZ9gW7Pc6KLsmgQqpCiKTrA1wh2XHmNf-NUP1-I,67
840
+ wandb-0.19.9.dist-info/licenses/LICENSE,sha256=izOKRJpGOx1PrJiGOKR0HsNdlB5JdH2d0Z4P7a7ssTc,1081
841
+ wandb-0.19.9.dist-info/RECORD,,
@@ -1,128 +0,0 @@
1
- # Generated by ariadne-codegen
2
-
3
- from __future__ import annotations
4
-
5
- from typing import Any, Literal, TypeVar
6
-
7
- from pydantic import BaseModel, ConfigDict, Field, Json, ValidationError, WrapValidator
8
- from pydantic.alias_generators import to_camel
9
- from pydantic.main import IncEx
10
- from pydantic_core import to_json
11
- from pydantic_core.core_schema import ValidatorFunctionWrapHandler
12
-
13
- from .typing_compat import Annotated, override
14
-
15
-
16
- # Base class for all automation classes/types.
17
- # Omitted from docstring to avoid inclusion in generated docs.
18
- class Base(BaseModel):
19
- model_config = ConfigDict(
20
- populate_by_name=True,
21
- validate_assignment=True,
22
- validate_default=True,
23
- extra="forbid",
24
- alias_generator=to_camel,
25
- use_attribute_docstrings=True,
26
- from_attributes=True,
27
- revalidate_instances="always",
28
- )
29
-
30
- @override
31
- def model_dump(
32
- self,
33
- *,
34
- mode: Literal["json", "python"] | str = "json", # NOTE: changed default
35
- include: IncEx | None = None,
36
- exclude: IncEx | None = None,
37
- context: dict[str, Any] | None = None,
38
- by_alias: bool = True, # NOTE: changed default
39
- exclude_unset: bool = False,
40
- exclude_defaults: bool = False,
41
- exclude_none: bool = False,
42
- round_trip: bool = True, # NOTE: changed default
43
- warnings: bool | Literal["none", "warn", "error"] = True,
44
- serialize_as_any: bool = False,
45
- ) -> dict[str, Any]:
46
- return super().model_dump(
47
- mode=mode,
48
- include=include,
49
- exclude=exclude,
50
- context=context,
51
- by_alias=by_alias,
52
- exclude_unset=exclude_unset,
53
- exclude_defaults=exclude_defaults,
54
- exclude_none=exclude_none,
55
- round_trip=round_trip,
56
- warnings=warnings,
57
- serialize_as_any=serialize_as_any,
58
- )
59
-
60
- @override
61
- def model_dump_json(
62
- self,
63
- *,
64
- indent: int | None = None,
65
- include: IncEx | None = None,
66
- exclude: IncEx | None = None,
67
- context: dict[str, Any] | None = None,
68
- by_alias: bool = True, # NOTE: changed default
69
- exclude_unset: bool = False,
70
- exclude_defaults: bool = False,
71
- exclude_none: bool = False,
72
- round_trip: bool = True, # NOTE: changed default
73
- warnings: bool | Literal["none", "warn", "error"] = True,
74
- serialize_as_any: bool = False,
75
- ) -> str:
76
- return super().model_dump_json(
77
- indent=indent,
78
- include=include,
79
- exclude=exclude,
80
- context=context,
81
- by_alias=by_alias,
82
- exclude_unset=exclude_unset,
83
- exclude_defaults=exclude_defaults,
84
- exclude_none=exclude_none,
85
- round_trip=round_trip,
86
- warnings=warnings,
87
- serialize_as_any=serialize_as_any,
88
- )
89
-
90
-
91
- # Base class with extra customization for GQL generated types.
92
- # Omitted from docstring to avoid inclusion in generated docs.
93
- class GQLBase(Base):
94
- model_config = ConfigDict(
95
- extra="ignore",
96
- protected_namespaces=(),
97
- )
98
-
99
-
100
- # ------------------------------------------------------------------------------
101
- # Reusable annotations for field types
102
- T = TypeVar("T")
103
-
104
- GQLId = Annotated[
105
- str,
106
- Field(repr=False, strict=True, frozen=True),
107
- ]
108
-
109
- Typename = Annotated[
110
- T,
111
- Field(repr=False, alias="__typename", frozen=True),
112
- ]
113
-
114
-
115
- def validate_maybe_json(v: Any, handler: ValidatorFunctionWrapHandler) -> Any:
116
- """Wraps default Json[...] field validator to allow instantiation with an already-decoded value."""
117
- try:
118
- return handler(v)
119
- except ValidationError:
120
- # Try revalidating after properly jsonifying the value
121
- return handler(to_json(v, by_alias=True, round_trip=True))
122
-
123
-
124
- SerializedToJson = Annotated[
125
- Json[T],
126
- # Allow lenient instantiation/validation: incoming data may already be deserialized.
127
- WrapValidator(validate_maybe_json),
128
- ]
@@ -1,27 +0,0 @@
1
- """MessageFuture - represents a message result of an asynchronous operation.
2
-
3
- Base class MessageFuture for MessageFutureObject and MessageFuturePoll
4
-
5
- """
6
-
7
- import threading
8
- from abc import abstractmethod
9
- from typing import Optional
10
-
11
- from wandb.proto import wandb_internal_pb2 as pb
12
-
13
-
14
- class MessageFuture:
15
- _object: Optional[pb.Result]
16
-
17
- def __init__(self) -> None:
18
- self._object = None
19
- self._object_ready = threading.Event()
20
-
21
- def _set_object(self, obj: pb.Result) -> None:
22
- self._object = obj
23
- self._object_ready.set()
24
-
25
- @abstractmethod
26
- def get(self, timeout: Optional[int] = None) -> Optional[pb.Result]:
27
- raise NotImplementedError
@@ -1,50 +0,0 @@
1
- """MessageFuturePoll - Derived from MessageFuture but implementing polling loop.
2
-
3
- MessageFuture represents a message result of an asynchronous operation.
4
-
5
- MessageFuturePoll implements a polling loop to periodically query for a
6
- completed async operation.
7
-
8
- """
9
-
10
- import time
11
- from typing import Any, Optional
12
-
13
- from wandb.proto import wandb_internal_pb2 as pb
14
-
15
- from .message_future import MessageFuture
16
-
17
-
18
- class MessageFuturePoll(MessageFuture):
19
- _fn: Any
20
- _xid: str
21
-
22
- def __init__(self, fn: Any, xid: str) -> None:
23
- super().__init__()
24
- self._fn = fn
25
- self._xid = xid
26
-
27
- def get(self, timeout: Optional[int] = None) -> Optional[pb.Result]:
28
- self._poll(timeout=timeout)
29
- if self._object_ready.is_set():
30
- return self._object
31
- return None
32
-
33
- def _poll(self, timeout: Optional[int] = None) -> None:
34
- if self._object_ready.is_set():
35
- return
36
- done = False
37
- start_time = time.time()
38
- sleep_time = 0.5
39
- while not done:
40
- result = self._fn(xid=self._xid)
41
- if result:
42
- self._set_object(result)
43
- done = True
44
- continue
45
- now_time = time.time()
46
- if timeout and start_time - now_time > timeout:
47
- done = True
48
- continue
49
- time.sleep(sleep_time)
50
- sleep_time = min(sleep_time * 2, 5)