wandb 0.21.1__py3-none-win_amd64.whl → 0.21.2__py3-none-win_amd64.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 +1 -1
  2. wandb/__init__.pyi +1 -1
  3. wandb/apis/public/api.py +1 -2
  4. wandb/apis/public/artifacts.py +3 -5
  5. wandb/apis/public/registries/_utils.py +14 -16
  6. wandb/apis/public/registries/registries_search.py +176 -289
  7. wandb/apis/public/reports.py +13 -10
  8. wandb/automations/_generated/delete_automation.py +1 -3
  9. wandb/automations/_generated/enums.py +13 -11
  10. wandb/bin/gpu_stats.exe +0 -0
  11. wandb/bin/wandb-core +0 -0
  12. wandb/cli/cli.py +47 -2
  13. wandb/integration/metaflow/data_pandas.py +2 -2
  14. wandb/integration/metaflow/data_pytorch.py +75 -0
  15. wandb/integration/metaflow/data_sklearn.py +76 -0
  16. wandb/integration/metaflow/metaflow.py +16 -87
  17. wandb/integration/weave/__init__.py +6 -0
  18. wandb/integration/weave/interface.py +49 -0
  19. wandb/integration/weave/weave.py +63 -0
  20. wandb/proto/v3/wandb_internal_pb2.py +3 -2
  21. wandb/proto/v4/wandb_internal_pb2.py +2 -2
  22. wandb/proto/v5/wandb_internal_pb2.py +2 -2
  23. wandb/proto/v6/wandb_internal_pb2.py +2 -2
  24. wandb/sdk/artifacts/_factories.py +17 -0
  25. wandb/sdk/artifacts/_generated/__init__.py +221 -13
  26. wandb/sdk/artifacts/_generated/artifact_by_id.py +17 -0
  27. wandb/sdk/artifacts/_generated/artifact_by_name.py +22 -0
  28. wandb/sdk/artifacts/_generated/artifact_collection_membership_file_urls.py +43 -0
  29. wandb/sdk/artifacts/_generated/artifact_created_by.py +47 -0
  30. wandb/sdk/artifacts/_generated/artifact_file_urls.py +22 -0
  31. wandb/sdk/artifacts/_generated/artifact_type.py +31 -0
  32. wandb/sdk/artifacts/_generated/artifact_used_by.py +43 -0
  33. wandb/sdk/artifacts/_generated/artifact_via_membership_by_name.py +26 -0
  34. wandb/sdk/artifacts/_generated/delete_artifact.py +28 -0
  35. wandb/sdk/artifacts/_generated/enums.py +5 -0
  36. wandb/sdk/artifacts/_generated/fetch_artifact_manifest.py +38 -0
  37. wandb/sdk/artifacts/_generated/fetch_registries.py +32 -0
  38. wandb/sdk/artifacts/_generated/fragments.py +279 -41
  39. wandb/sdk/artifacts/_generated/link_artifact.py +6 -0
  40. wandb/sdk/artifacts/_generated/operations.py +654 -51
  41. wandb/sdk/artifacts/_generated/registry_collections.py +34 -0
  42. wandb/sdk/artifacts/_generated/registry_versions.py +34 -0
  43. wandb/sdk/artifacts/_generated/unlink_artifact.py +25 -0
  44. wandb/sdk/artifacts/_graphql_fragments.py +3 -86
  45. wandb/sdk/artifacts/_validators.py +6 -4
  46. wandb/sdk/artifacts/artifact.py +406 -543
  47. wandb/sdk/artifacts/artifact_file_cache.py +10 -6
  48. wandb/sdk/artifacts/artifact_manifest.py +10 -9
  49. wandb/sdk/artifacts/artifact_manifest_entry.py +9 -10
  50. wandb/sdk/artifacts/artifact_manifests/artifact_manifest_v1.py +5 -3
  51. wandb/sdk/artifacts/storage_handlers/http_handler.py +1 -1
  52. wandb/sdk/artifacts/storage_handlers/s3_handler.py +1 -1
  53. wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py +1 -1
  54. wandb/sdk/data_types/video.py +2 -2
  55. wandb/sdk/interface/interface_queue.py +1 -4
  56. wandb/sdk/interface/interface_shared.py +26 -37
  57. wandb/sdk/interface/interface_sock.py +24 -14
  58. wandb/sdk/internal/settings_static.py +2 -3
  59. wandb/sdk/launch/create_job.py +12 -1
  60. wandb/sdk/launch/runner/kubernetes_runner.py +24 -29
  61. wandb/sdk/lib/asyncio_compat.py +16 -16
  62. wandb/sdk/lib/asyncio_manager.py +252 -0
  63. wandb/sdk/lib/hashutil.py +13 -4
  64. wandb/sdk/lib/printer.py +2 -2
  65. wandb/sdk/lib/printer_asyncio.py +3 -1
  66. wandb/sdk/lib/retry.py +185 -78
  67. wandb/sdk/lib/service/service_client.py +106 -0
  68. wandb/sdk/lib/service/service_connection.py +20 -26
  69. wandb/sdk/lib/service/service_token.py +30 -13
  70. wandb/sdk/mailbox/mailbox.py +13 -5
  71. wandb/sdk/mailbox/mailbox_handle.py +22 -13
  72. wandb/sdk/mailbox/response_handle.py +42 -106
  73. wandb/sdk/mailbox/wait_with_progress.py +7 -42
  74. wandb/sdk/wandb_init.py +11 -25
  75. wandb/sdk/wandb_login.py +1 -1
  76. wandb/sdk/wandb_run.py +91 -55
  77. wandb/sdk/wandb_settings.py +45 -32
  78. wandb/sdk/wandb_setup.py +176 -96
  79. wandb/util.py +1 -1
  80. {wandb-0.21.1.dist-info → wandb-0.21.2.dist-info}/METADATA +1 -1
  81. {wandb-0.21.1.dist-info → wandb-0.21.2.dist-info}/RECORD +84 -68
  82. wandb/sdk/interface/interface_relay.py +0 -38
  83. wandb/sdk/interface/router.py +0 -89
  84. wandb/sdk/interface/router_queue.py +0 -43
  85. wandb/sdk/interface/router_relay.py +0 -50
  86. wandb/sdk/interface/router_sock.py +0 -32
  87. wandb/sdk/lib/sock_client.py +0 -232
  88. {wandb-0.21.1.dist-info → wandb-0.21.2.dist-info}/WHEEL +0 -0
  89. {wandb-0.21.1.dist-info → wandb-0.21.2.dist-info}/entry_points.txt +0 -0
  90. {wandb-0.21.1.dist-info → wandb-0.21.2.dist-info}/licenses/LICENSE +0 -0
@@ -1,6 +1,6 @@
1
1
  package_readme.md,sha256=XGlaq8rMFcoBb21rCr2d5qeSM79ZI4WslLmXqRimTGQ,4395
2
- wandb/__init__.py,sha256=q56WIJIb9u4guY044HErNWFK457wKs95mFXgjVXI4jU,7019
3
- wandb/__init__.pyi,sha256=kZum0k3QMs02JXvkW5w5w7FHPZMlMPluLBu_bU561Gs,47627
2
+ wandb/__init__.py,sha256=_nTQiJkk-zl4kRquQFftsa_Px2ldqHwq3RoBA1D9Ldc,7019
3
+ wandb/__init__.pyi,sha256=5LoC6ETFSwoljpNw7SuNhV7E949iIb5ARJ5z-xJ97tE,47627
4
4
  wandb/__main__.py,sha256=uHY6OxHT6RtTH34zC8_UC1GsCTkndgbdsHXv-t7dOMI,67
5
5
  wandb/_iterutils.py,sha256=_ZUvfdQ8YeKrT_0HKaAKjkVyLHQdTYRcCh0l-zGxTMM,2465
6
6
  wandb/data_types.py,sha256=DdCkf7Dh_j86Q74FWzh3M20EW_hzKpNagexjo03qv-A,2349
@@ -9,7 +9,7 @@ wandb/jupyter.py,sha256=lh9J-oZLRB_INd5FHrk3XTbK5TJ1b8-BTZ3Uh9piZAs,17861
9
9
  wandb/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
10
10
  wandb/sklearn.py,sha256=8wlo8mLyekq6yRZBgluzPj4J-RP3wkLaEWm0yh7gvWw,838
11
11
  wandb/trigger.py,sha256=d9TizmMPsvr0k9_3SBK2nq-Mb95bTzf9DZ1aE_F0ASo,644
12
- wandb/util.py,sha256=mjnS-Fq92e7FiRcSIUg3pGl2mwxB0ZvKDcDkhH5px0c,67402
12
+ wandb/util.py,sha256=t5H7IRtpAmNdMGqE8bU2w8qdlNLr6w_cSn4GaUYmf-s,67421
13
13
  wandb/wandb_agent.py,sha256=-X7d5b8wihbHMwTmuLTZir620nWhTc345xp6lkSyF1o,21448
14
14
  wandb/wandb_controller.py,sha256=od0i1iO61iHMmiE_W-onoIwUAFeoym4RsUAw_0mJ11E,25655
15
15
  wandb/wandb_run.py,sha256=yZosTmFJIHHEsL9uGtFMRIlkdxaxjsG9z-Rfn1NLEbY,136
@@ -34,8 +34,8 @@ wandb/apis/importers/internals/internal.py,sha256=ljRzt3FqYc4zRpCsCl9PUvXFxYAqiP
34
34
  wandb/apis/importers/internals/protocols.py,sha256=PRnN66EtZPDtCiZomB_CtTtE0tMcebeG9aAK5OyDSGk,2988
35
35
  wandb/apis/importers/internals/util.py,sha256=MuclUljXdA__0ULaqcBM6ux4pK6YUW5i6ezri5X7_7Q,2159
36
36
  wandb/apis/public/__init__.py,sha256=SCRGdvgyjdTDlbm79g-fPoz7Bo02y017Xl_z2c8g01Q,1249
37
- wandb/apis/public/api.py,sha256=NBfhAO-OZ_uBEHyH6_TqYqSI9v_oQ6hsPSqA1hdHFEo,90669
38
- wandb/apis/public/artifacts.py,sha256=GEpMpORU3rPvuDQpmUPy1GdjOzfD66qAXHt3sk88u1s,36101
37
+ wandb/apis/public/api.py,sha256=4-IiR36fPRGkrLoI7nybjMiQ22o9uMgAuAnUAvoEJtk,90671
38
+ wandb/apis/public/artifacts.py,sha256=Dpc9msTknNL-IgY-MaE8HFp4sZSeoVLdX3itRJoFfoY,36014
39
39
  wandb/apis/public/automations.py,sha256=dNlgj9xDDLmcEP0Ex0oiIWLY2-QLGxCD15Sv6nsdlv0,2700
40
40
  wandb/apis/public/const.py,sha256=aK9Fcp1clmTHWj0C24fTRU3ecP5u91dPmp298kLiBdM,125
41
41
  wandb/apis/public/files.py,sha256=o-OblPIuVpQ5ukWp8EtLQHsgTrWtD2SQOdT1oW5BFA8,13503
@@ -44,7 +44,7 @@ wandb/apis/public/integrations.py,sha256=U6A81bWisSgyuj4_F9D6X_ZjAuRw317L4oLetbx
44
44
  wandb/apis/public/jobs.py,sha256=5APZKQN6g7ZyvjjTCR6-scCc_QsnOo-ceIbN8ituB9k,27044
45
45
  wandb/apis/public/projects.py,sha256=KFrUZ2PnUz07hWOSGhWGk6rLYOxxVwwwzXoe52Jt9hc,8294
46
46
  wandb/apis/public/query_generator.py,sha256=3rQkvnFYdV8c1WfRjwLkZWROy_LUXb4T7OoDDsjojrY,6703
47
- wandb/apis/public/reports.py,sha256=4RIx_KUMFoxkX2mGA4JGKfwefyQt4i5CbS-EmvASJXA,20173
47
+ wandb/apis/public/reports.py,sha256=3zqxkIIj6tN0GHHdBsn72r1GRkm-r2fy_vAUyB2-mvo,20414
48
48
  wandb/apis/public/runs.py,sha256=9po-3-LUnILn-vOtVSDbz178T3wxRwQO1VQA0dhOBo0,43207
49
49
  wandb/apis/public/sweeps.py,sha256=wkmSTxvsKUE2GJY5vnEN_otb0VoOpDKb6gGJBq6ibtM,13386
50
50
  wandb/apis/public/teams.py,sha256=jLu8LrYwLLCNpAOzrsBiHmqQIgGyfEti-7E5eNj77VA,6850
@@ -52,8 +52,8 @@ wandb/apis/public/users.py,sha256=Ns2k1nsq5MxSYliIHiN2ZRLaMHAGssBKro1Ht8tlKCU,55
52
52
  wandb/apis/public/utils.py,sha256=Xj3eQfhh142L7Le2XuOzDi88pZqaYXOdzL9CNU5WDTs,8103
53
53
  wandb/apis/public/registries/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
54
54
  wandb/apis/public/registries/_freezable_list.py,sha256=98MhXhU6meYVjyG5uCskaLaB0bjL4Qw9fm0hKGsDgmM,6547
55
- wandb/apis/public/registries/_utils.py,sha256=AwRdWZCWDleS2Dh-AcetItkRKr44OIDXeWLrNt9czjw,4417
56
- wandb/apis/public/registries/registries_search.py,sha256=iGS_abq1O0HbYV6CkbXbzRiFeamXYBoKfKHGX_3lJNU,15544
55
+ wandb/apis/public/registries/_utils.py,sha256=t3WjmTbf42mug8t9HA6UVzS63UhkBh4JGyL-HjX1tO4,4429
56
+ wandb/apis/public/registries/registries_search.py,sha256=7xvEa8I5hkBeF2GQyhCVyDyvjuCYxb9obLXtT_t93Hg,11705
57
57
  wandb/apis/public/registries/registry.py,sha256=cFEelF_TcMNepNzV0LBNO-xAQKfHytSKOiKfKc6ypKw,14083
58
58
  wandb/apis/reports/__init__.py,sha256=pKAM02nyHJV6DwGQuzAhSlqTOsCHKah1FlJIDERyY5U,33
59
59
  wandb/apis/reports/v1/__init__.py,sha256=nxs3gJlbvVc0b_pV5DUypk1amMkRSq_M-xUw7qPTfwI,271
@@ -74,8 +74,8 @@ wandb/automations/_filters/run_metrics.py,sha256=eS_Ut5qHOS7H1vqMd1ObKhYiNJ-oBet
74
74
  wandb/automations/_generated/__init__.py,sha256=9h4EKABBVPwyWtb7NrZvz2dezRuc6hNZl1K3w_TUetI,7001
75
75
  wandb/automations/_generated/create_automation.py,sha256=8v-XMu54bRFtgjlImElzJT8qpUP9RKrJ86JCdbTu7jA,357
76
76
  wandb/automations/_generated/create_generic_webhook_integration.py,sha256=p2f685wcqlzepdL7_CRpnLKjvtkDBK1IGbxqfhLnaQY,1463
77
- wandb/automations/_generated/delete_automation.py,sha256=64lLU11WH5e0EY1u5DBqFvZqRreHp4FOzThj9jPLpqE,357
78
- wandb/automations/_generated/enums.py,sha256=s1pCCCaUnqr8XKh3O3P2JXOC7q5XpjVpyipB-LvBpZM,832
77
+ wandb/automations/_generated/delete_automation.py,sha256=bdSzqEA_zYzRzyPb8NC2TSzSDokBhMYPIvhDV7w7GXI,316
78
+ wandb/automations/_generated/enums.py,sha256=eVflwepEDpN3OcaBDjwOsUpA9XBeGzasm02e4EH9UrQ,904
79
79
  wandb/automations/_generated/fragments.py,sha256=z9A4aWg9Z0ogjmn53zpiuHBEq1OdB6Dl_sv13jNd6w4,12375
80
80
  wandb/automations/_generated/generic_webhook_integrations_by_entity.py,sha256=GKT8Cy_7dl-C4FqYOGwLF_I1_y9OnQWT9TgryYusPiI,621
81
81
  wandb/automations/_generated/get_automations.py,sha256=GdDO-evZ9c2mBzbuaFaJJu18x9oq21heZc6k4ZlvSSY,561
@@ -86,11 +86,11 @@ wandb/automations/_generated/operations.py,sha256=V0n3PZ_e754Ym8IraYjFzLWg0wqnA_
86
86
  wandb/automations/_generated/slack_integrations_by_entity.py,sha256=95llPwkGxKrwX4R6ze9IxzmMuyLpIWoperLUy-hKsE0,558
87
87
  wandb/automations/_generated/update_automation.py,sha256=9SvGNXYK7fEZdhztonzRcMiJ8Yqps-42hKB16CUOFYI,357
88
88
  wandb/beta/workflows.py,sha256=8uFVT9y-9CotD7OLkSQ4mRd_tsrSRsC1q0tAmJ489qI,11825
89
- wandb/bin/gpu_stats.exe,sha256=vvKnulkiezEohKZA-HvOdgSAEeBG2SU0LkmSd9yJzZ4,8121344
90
- wandb/bin/wandb-core,sha256=q4CA5CuSwF8vGoinsOuYBUMhOcy_7_qrW0a01xRyg6g,55567872
89
+ wandb/bin/gpu_stats.exe,sha256=7bYm2DxBLyrqNJrMkiGINpu-qaIym2Cb6raNzSaDKnU,8110592
90
+ wandb/bin/wandb-core,sha256=xe8XmdWcg_Upa46IAvV1zwAuxRHI7YrHhgr9MaQ7RSs,43168768
91
91
  wandb/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
92
92
  wandb/cli/beta.py,sha256=NBgJ67mwdd3k7-PEfiNTvE8dh7Ku4hv4FMZMJ9pN0Is,5362
93
- wandb/cli/cli.py,sha256=IseWEzm7uC40LZEHDbrEb_IZjHl6IOTp9ROKRLqFk_U,96315
93
+ wandb/cli/cli.py,sha256=lya6C7k2IBaNFhuloH2FMWbBaYeINvYXr-2xoj--Vnw,97908
94
94
  wandb/docker/__init__.py,sha256=ySgObcuW3AU-CcHIy2-hk2OCEMPHJmoP679eIpWz_kc,8953
95
95
  wandb/docker/names.py,sha256=E0v_-WInxWgg3vK14Tpj0F4zuwaBl8nR7OTQCYbcod4,1351
96
96
  wandb/docker/wandb-entrypoint.sh,sha256=ksJ_wObRwZxZtdu1Ahc1X8VNB1U68a3nleioDDBO-jU,1021
@@ -141,9 +141,11 @@ wandb/integration/lightning/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMp
141
141
  wandb/integration/lightning/fabric/__init__.py,sha256=_KF_WmU4ynBy9WT8EVXZw0cwDtF2t1TQ_GzUDQLXeAg,97
142
142
  wandb/integration/lightning/fabric/logger.py,sha256=ewGUt_MixsVWXFbSI58HWY5m0DekvZwSBRxzGE0ohyk,27996
143
143
  wandb/integration/metaflow/__init__.py,sha256=EVDYZF58325MGWEqnFOGWlcYLB3KilfTXZMA5sKvPFg,272
144
- wandb/integration/metaflow/data_pandas.py,sha256=dCcLXvMdGpK1opky3rO5dT7Y7-iwu4-D-GiQx5BJldc,1793
144
+ wandb/integration/metaflow/data_pandas.py,sha256=LUWwh7zGchKG2rDPcgZYvlqT93kRTbfpcu4ObVpxiaE,1787
145
+ wandb/integration/metaflow/data_pytorch.py,sha256=GMojrXibwNQvcp3u3BdBxP6ZpH5QBCIFZEoy30JOVZM,1787
146
+ wandb/integration/metaflow/data_sklearn.py,sha256=XxxtPtGZQi3TXX7oPwieF9Qew3jXxGkjF7ZnTt0YfSA,1829
145
147
  wandb/integration/metaflow/errors.py,sha256=Vn0mwpdWtkPlK2QWIKBt0y85053bdOJnYAaeemo7IRM,360
146
- wandb/integration/metaflow/metaflow.py,sha256=LrcqGDU-j0NkMwqzm2hBAPBP9RWCdAA_BoSb0MQ52WM,11648
148
+ wandb/integration/metaflow/metaflow.py,sha256=8gfcxNiX0ufIKNkH26M6-feYzcR4Ty31MLfKr2ZJ7wc,9738
147
149
  wandb/integration/openai/__init__.py,sha256=negbYxbmhe4hA7Zl41FA5FnZ1JP6G6IYUTK4lS4YSqs,69
148
150
  wandb/integration/openai/fine_tuning.py,sha256=U-e1ZPAcMW60EvSmTywWWBPHJkMgIR_gr0WvPeHcFqA,18989
149
151
  wandb/integration/openai/openai.py,sha256=43Lm8sk_VksyRf8g6TJqbNlTlwA3ujaZxF5wfhqtjQk,518
@@ -190,6 +192,9 @@ wandb/integration/ultralytics/callback.py,sha256=5eQKj7AI_y6O18ONmhl5tKhCo-5xEvU
190
192
  wandb/integration/ultralytics/classification_utils.py,sha256=b_EGdLZkf1sjJuJwmp7segl_-47T-99d0Yyrn57j3xM,3251
191
193
  wandb/integration/ultralytics/mask_utils.py,sha256=8Bue8DMcR8J1sXrZaz-6u58GOZMuwj5c0V75KrV3S1A,7282
192
194
  wandb/integration/ultralytics/pose_utils.py,sha256=8rqsRiTAT-mQ3LgARAKQXdgfBTOh0j4q9li8RJjP7LY,3812
195
+ wandb/integration/weave/__init__.py,sha256=6KlykgTB8o_q6E9qWRozArOQ6sxbb4MjCGLaWTI_-WA,164
196
+ wandb/integration/weave/interface.py,sha256=rvX4bfcUw6BJqxJ2S7fCcreMH0prUQbsd5vhxXuGuAk,1214
197
+ wandb/integration/weave/weave.py,sha256=cFj0RF7ZN6cqrtS75fcIwQSHEtR5x70TVgrMk1wiH0E,2001
193
198
  wandb/integration/xgboost/__init__.py,sha256=WGqILO3kWero1jaSRwLkEJ4JDOeVw8LQT0xI8142mwA,354
194
199
  wandb/integration/xgboost/xgboost.py,sha256=ARaZjA_GFL2GX5itRJmVf8pw4YbSPVETVVMM2UourB0,6684
195
200
  wandb/integration/yolov8/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -223,23 +228,23 @@ wandb/proto/wandb_settings_pb2.py,sha256=vXdUZkLz1x7hw662A1GmOvedOWB551q4dmmw15n
223
228
  wandb/proto/wandb_telemetry_pb2.py,sha256=-vMGhrHljykiYoRmS8Y9jN8Or4g02VzHIrG4j2b2MOM,417
224
229
  wandb/proto/v3/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
225
230
  wandb/proto/v3/wandb_base_pb2.py,sha256=zwma_gb3IOSfBJ1tvMIdmQtQQZLe29upb8Mqr4m9No4,2410
226
- wandb/proto/v3/wandb_internal_pb2.py,sha256=gRZ9iWHrSkgisexyhTl6JEkL5OnAMtPkhjuW1s0UNHs,118705
231
+ wandb/proto/v3/wandb_internal_pb2.py,sha256=UwkQkdB8oUpeGmQHkBYxwgy86LVLxslkPDW6joUzOVM,118824
227
232
  wandb/proto/v3/wandb_server_pb2.py,sha256=gjIGUr71bFP25WBV6UTtGQdS7jk_8HEasE1kw1anlyk,15742
228
233
  wandb/proto/v3/wandb_settings_pb2.py,sha256=StpCiKAm5QlFDWlIncW9GW-hAx3H05H4LdL2ecGsy2U,21968
229
234
  wandb/proto/v3/wandb_telemetry_pb2.py,sha256=pMtYjhfvkvGt3WKFRRbqasaiiwdf1k_-Xbzm4hxBelg,14624
230
235
  wandb/proto/v4/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
231
236
  wandb/proto/v4/wandb_base_pb2.py,sha256=tl7f-74ItLSWCP_GDfAWm02sTEgUpWZGoP_vqEpvRE8,1452
232
- wandb/proto/v4/wandb_internal_pb2.py,sha256=oJ7zwuoa8vgDndFK_fTDNSLGlhZGa78jEpAwEIyKeLU,54135
237
+ wandb/proto/v4/wandb_internal_pb2.py,sha256=_mydxuEoPF6DmRYETxOfGrYuWOFWQdY9wdgPMN-pwJU,54202
233
238
  wandb/proto/v4/wandb_server_pb2.py,sha256=Wo4FbkBik5cftjZDtlO80LIMRfJetOGOAxj5zy9jXdQ,7044
234
239
  wandb/proto/v4/wandb_settings_pb2.py,sha256=jD91g1lN4Xqcr74XQoYT6vyvqfFOxbxrGlzv4Wve4FM,18268
235
240
  wandb/proto/v4/wandb_telemetry_pb2.py,sha256=84ARYGIGmo5LsWQPs8uzyBMFW-HrUC4SCwQHflpDap0,12006
236
241
  wandb/proto/v5/wandb_base_pb2.py,sha256=ES3U80f2YCt-fwiqaIrz7BGHVywwx6ibEDAnlWpohig,1603
237
- wandb/proto/v5/wandb_internal_pb2.py,sha256=QPFVFjGQYuJTY60qA7VbMsuKvcP7fsqQGCWXQcu4wJw,58483
242
+ wandb/proto/v5/wandb_internal_pb2.py,sha256=qy53aiey0CaEW6y83iQJJyG0s5iWPTvr9TdxHJI2F_Y,58550
238
243
  wandb/proto/v5/wandb_server_pb2.py,sha256=Zkw2UopD20-C514TgqZ1P3NZTSLPTGjSVqihxcMpVoM,7603
239
244
  wandb/proto/v5/wandb_settings_pb2.py,sha256=VxT4LysYbRGPmBdB0AAtX3FWZ4BZBlNdtQuQhfTRxgs,18625
240
245
  wandb/proto/v5/wandb_telemetry_pb2.py,sha256=SxwhZBlF8wMtNOVXRkvSKTo6dmLgtZL-DoUTK_ihKBk,12277
241
246
  wandb/proto/v6/wandb_base_pb2.py,sha256=pKoyhXXI2gap0mWgjXy3AspHkf-vwUcSp1cuz3VO4O4,1862
242
- wandb/proto/v6/wandb_internal_pb2.py,sha256=mEq8DNxpCSp9-KMT5tvrOssh-R9fQrb5eD7adeFl9j0,58746
247
+ wandb/proto/v6/wandb_internal_pb2.py,sha256=G8xGzEjRXVRbJATJ2RuaPXzHZTRcyfCtjuvh6FK_A_Q,58813
243
248
  wandb/proto/v6/wandb_server_pb2.py,sha256=jqUQpTgP0sY4Yd6l58N7MEMWyBkYH6Rb8XvmJn2JY0M,7864
244
249
  wandb/proto/v6/wandb_settings_pb2.py,sha256=DoGCT7goUXJFEhf239GmIV6iMp1BhdOdVPUCdBODpzM,18888
245
250
  wandb/proto/v6/wandb_telemetry_pb2.py,sha256=QnqVX1Lr8N9-xizv4LTxyuJ8MAAkXminnQ4hG5y-Fw8,12541
@@ -247,28 +252,29 @@ wandb/sdk/__init__.py,sha256=6lzqckLZUs7GpFZIwpgxGJwJDvhuyo-XCQnSrtZqE1c,850
247
252
  wandb/sdk/wandb_alerts.py,sha256=f6ygzuXTDT0IvMLcKlgatmXKx5HMPsm8sYwvPocl0Js,205
248
253
  wandb/sdk/wandb_config.py,sha256=yVtkXWoDbVmgUD8UVV-f4y4r_A1atVSvxfCk5eN7Hp4,11015
249
254
  wandb/sdk/wandb_helper.py,sha256=kc5Ib648to7cEGEwAuJus07rsHudL1Ux7FWPPSRnKy8,1878
250
- wandb/sdk/wandb_init.py,sha256=6f8vNylK2wDEJjIgQyY20Sl6k2ikOnPmWMWFEbhPZdM,62944
251
- wandb/sdk/wandb_login.py,sha256=rfGSE9AvA1RTWyXTHaG-JmccjYGAYFy9_ywtiMKE4T0,11658
255
+ wandb/sdk/wandb_init.py,sha256=9T31ThA9DP1_6F0rWoFU_bAW6fgY27iQ1-0LNrfirDk,62471
256
+ wandb/sdk/wandb_login.py,sha256=H_T0DThPMk93rVTdet22i7qzRJ1kmzVFpkhlXkpRk1Q,11657
252
257
  wandb/sdk/wandb_metric.py,sha256=JsJP2UZCliJ8OdC_uvvd_XF5k4FIAt3YWlmN1O9jLmc,3458
253
258
  wandb/sdk/wandb_require.py,sha256=bUS9nZXUVjmFcc6EMW6ZSzXtQf1xg0bX98ljVxFdVX0,2801
254
259
  wandb/sdk/wandb_require_helpers.py,sha256=4PUXmVw86_XaKj3rn20s5DAjBMO8L0m26KqnTLaQJNc,1375
255
- wandb/sdk/wandb_run.py,sha256=wEeQym1vQEqy0Exk5LCOd8BCY8PvKUGjAdYiG9fAsR8,149829
256
- wandb/sdk/wandb_settings.py,sha256=9P7LWvyNXMcOa8cGcAYlZBzVRgTEnmOHANBJj0tFh0A,74288
257
- wandb/sdk/wandb_setup.py,sha256=TR1iUAkhHw9xQS9lJfqJgoYPY9cPQNiCojk3uD6FOmY,16631
260
+ wandb/sdk/wandb_run.py,sha256=qW5soKEh9Bf1xyccOETIpFFDrtgXOToaehRG-k3GAac,150836
261
+ wandb/sdk/wandb_settings.py,sha256=8NSnyAgIpgNhUyCISfvLoUM5Cu2ldNk8_KzBtsHv9go,75033
262
+ wandb/sdk/wandb_setup.py,sha256=Z-tmy9pypXF1cwEYpoqqVo-yOsrSbCFYilZPKHKgcSw,19676
258
263
  wandb/sdk/wandb_summary.py,sha256=eEV3hvHhbc1XQus0MUqFmvhXCzd3SPjvVVVg_fVZ1QM,4686
259
264
  wandb/sdk/wandb_sweep.py,sha256=Q7IKd0wXBRxoZLvJVq2knGzAe0W1tVfQjZfFm39D7qs,4102
260
265
  wandb/sdk/wandb_sync.py,sha256=Bk1X5a2RgpdB24trwwW1fLC1sEKET87ySCH84T-kgf4,2177
261
266
  wandb/sdk/wandb_watch.py,sha256=BT3_kwr7xm_z12cYzB0Tm_cwuLtpVWt2dALKj4piDx4,4892
262
267
  wandb/sdk/artifacts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
263
- wandb/sdk/artifacts/_graphql_fragments.py,sha256=_2b71Lx0tsaM6TB51x3pvkbtnydGG6l8FtQI7V7uVXg,2704
268
+ wandb/sdk/artifacts/_factories.py,sha256=u4i4U86Q98__9u1MsbvFKS3LHfCeym14yivOA6vu9DM,560
269
+ wandb/sdk/artifacts/_graphql_fragments.py,sha256=JTPeETiDEXyBxOM_3YYlEGfOI8-AhK8vE3a2dfSB5N8,600
264
270
  wandb/sdk/artifacts/_internal_artifact.py,sha256=ZWu3XwH0zF8Z5LdfuGSmR-kMcVZW6B-_dCvHjZsUIX0,2029
265
- wandb/sdk/artifacts/_validators.py,sha256=Fu4XLexZHsnBtSj4tljF2t60UEmAP4c8dPNpYfhgKbI,11159
266
- wandb/sdk/artifacts/artifact.py,sha256=thsyidFAXLcRpTNOcZhzPOZdGsQbT4T-1H3uwOOM278,109640
271
+ wandb/sdk/artifacts/_validators.py,sha256=gTHRgftmFRq9DqqW1vSTKT_lW4Az5ve5zMmUIhk_zyk,11257
272
+ wandb/sdk/artifacts/artifact.py,sha256=G3_rModzh6WZEUdegOL-wME_eNcdJvbGe0Gut9og5cM,105943
267
273
  wandb/sdk/artifacts/artifact_download_logger.py,sha256=CPle_AgeO53rr9FwQTuB5FV4bRKWdtoElJUDMQf9I7A,1582
268
- wandb/sdk/artifacts/artifact_file_cache.py,sha256=MMSG6OTOSB_HAeLesXfgLIWpdYJVUW4vAKacZjLIEmY,9924
274
+ wandb/sdk/artifacts/artifact_file_cache.py,sha256=yVc5Po8oU9M1DKglS01Ts6sd9gXvL7-Gc8IDMLlD5bc,10130
269
275
  wandb/sdk/artifacts/artifact_instance_cache.py,sha256=Y86c2ph4Fz1p5mfTpWMEPh1VhRzi-OyLGswa-NQDuUw,518
270
- wandb/sdk/artifacts/artifact_manifest.py,sha256=2Ocn_6ANCt8pokJQ838q3l3EEl6hBOtxEILRNsxUxAo,2620
271
- wandb/sdk/artifacts/artifact_manifest_entry.py,sha256=lXQqGwETDj7fPYABONMixiO4IjxbShJPDX4Zk8rfU9A,8902
276
+ wandb/sdk/artifacts/artifact_manifest.py,sha256=V-xjNzEhPoTV0PmO8Vhpbu0Ivg4QbaLWid1GLdaxfJc,2610
277
+ wandb/sdk/artifacts/artifact_manifest_entry.py,sha256=nes1jOCl2WmBudNc4SHvM88YQh9tkWPp_76T4Qrd7Og,8980
272
278
  wandb/sdk/artifacts/artifact_saver.py,sha256=0O4M7nkLdr0M0afStwcqcoRKFmrFK09Uu8ztfP0bTLg,9931
273
279
  wandb/sdk/artifacts/artifact_state.py,sha256=7bdU6ZIt-nx9aSO-aaRdZjj64MKD1ue6MhS6umlD1_U,285
274
280
  wandb/sdk/artifacts/artifact_ttl.py,sha256=L4gGRTKjQAu3hKjiko4TbOAQwVBsZWjQe7esFN7d7rY,131
@@ -277,47 +283,61 @@ wandb/sdk/artifacts/staging.py,sha256=mkW7Ct_fGm-V7DhVmK_aesO6li8zadWWB1NiKd04ZB
277
283
  wandb/sdk/artifacts/storage_handler.py,sha256=-dhnO1V6HvqiJ7bFwF_hRjxp72uRZIGLvcLpjLCNhzY,1896
278
284
  wandb/sdk/artifacts/storage_layout.py,sha256=C83BTO0Z6Bbc74p3IZ3N_yPtn0AkgVjmDwcJSanKe5M,117
279
285
  wandb/sdk/artifacts/storage_policy.py,sha256=g9ICIXrzbQ-DLBGNL5w_wd9TZf5oGLgtKFLRj0ATrIA,2291
280
- wandb/sdk/artifacts/_generated/__init__.py,sha256=ekArg-UFRQ_iol5KWb983P5ieGBUfiSBLW5nh_OAot0,12594
286
+ wandb/sdk/artifacts/_generated/__init__.py,sha256=_yrDX4ccLUMpea03gPBR63Bl6hZYrMyjgpRvOoP6l6Y,20611
281
287
  wandb/sdk/artifacts/_generated/add_aliases.py,sha256=K8HqNr3Ymd79shMYo3PRAof4DhZ6rEwEBTOiOvRip_w,416
288
+ wandb/sdk/artifacts/_generated/artifact_by_id.py,sha256=Xlq1FLE72MKOuEjOGsSxI9BButMwyW1PuSE5Ff1t5aE,337
289
+ wandb/sdk/artifacts/_generated/artifact_by_name.py,sha256=4TokH0DwaMHEAE8rKiuF_x-4G4ZJGRNInel_03tfi1Q,469
290
+ wandb/sdk/artifacts/_generated/artifact_collection_membership_file_urls.py,sha256=PpsEIXKVg0plyEpEBrsxl5xqN8RNCb3Xrb3iB051QMQ,1425
282
291
  wandb/sdk/artifacts/_generated/artifact_collection_membership_files.py,sha256=sGekmY1FfI8GK6xyYUHW2ncRP3MXBTbSQMyyAfaOns8,1386
292
+ wandb/sdk/artifacts/_generated/artifact_created_by.py,sha256=zqTFc2VaVnE2oEdvpMxYDfKrsL9Uq43IQRODcdL83uw,1263
293
+ wandb/sdk/artifacts/_generated/artifact_file_urls.py,sha256=q7xyTzV3UJyvp1otJ0oTF4DO1g8utqLYAZWuIZf5i0s,480
294
+ wandb/sdk/artifacts/_generated/artifact_type.py,sha256=_Gcb83PAdl2cXduBaK83UxAdwXS3G_agIu4CHWEZ0HQ,717
295
+ wandb/sdk/artifacts/_generated/artifact_used_by.py,sha256=1TujQDAAxMVAltVDP_t9dYu8EY2l3NrgEALKN8fzlE4,1123
283
296
  wandb/sdk/artifacts/_generated/artifact_version_files.py,sha256=olzXNAKeIxeR5kw0ZrOKdphGKkF1eIBHYF5v7e3VbKc,961
297
+ wandb/sdk/artifacts/_generated/artifact_via_membership_by_name.py,sha256=nQw7i9tpe-EgYEPaD2CMNRVlXeCgvofQ7cbNppI_7vQ,660
284
298
  wandb/sdk/artifacts/_generated/create_artifact_collection_tag_assignments.py,sha256=WhyErkmRPgrbh20m980pPuJz6BO2PSv1hSWO1T_tEfw,1070
285
299
  wandb/sdk/artifacts/_generated/delete_aliases.py,sha256=pyQ73fLsygF8VVknlgmugAlfwSdNqOA0eD8YG5CaU68,440
300
+ wandb/sdk/artifacts/_generated/delete_artifact.py,sha256=lN9HuzZOOF63Ex140yhp0HOJLWlS9J2WJRxJd1Nt6EI,623
286
301
  wandb/sdk/artifacts/_generated/delete_artifact_collection_tag_assignments.py,sha256=X3i8E3yOgUVAf1MnRmnrmWW2aIJQgrxc45XZgQGHRbY,667
287
302
  wandb/sdk/artifacts/_generated/delete_artifact_portfolio.py,sha256=wf_A0_ba2lVUnwgPO2EyCS-0CiEQ3iN0Vgku65cMLt0,1026
288
303
  wandb/sdk/artifacts/_generated/delete_artifact_sequence.py,sha256=k9KuV_u1QQwwFuc7l1zBeKpNiW5IJE6xmMb1sDkcLe4,1014
289
- wandb/sdk/artifacts/_generated/enums.py,sha256=inwefx49oJ8Z19smv0o49vG7eepnNFgzaQGQipRZ7EA,362
304
+ wandb/sdk/artifacts/_generated/enums.py,sha256=TR33jhsWq0RATqx1OmGQ94HhBdCrmx98hHwHXOMLZjo,464
305
+ wandb/sdk/artifacts/_generated/fetch_artifact_manifest.py,sha256=LhFBrc_gBAhKST7CtP-ep6ZziyPv76ytjRS-KyhEkVQ,1093
290
306
  wandb/sdk/artifacts/_generated/fetch_linked_artifacts.py,sha256=i6fV1sLTYjhjR3EsMmlOQ6bqGLsPn7Hq_ilvs2UbHmM,2215
291
- wandb/sdk/artifacts/_generated/fragments.py,sha256=O3nz4g0HOkmXj6XIwxNMXfHaGmzQFl-gWQ8LVkdbiZw,6912
307
+ wandb/sdk/artifacts/_generated/fetch_registries.py,sha256=UOYnAi9Xg1eh8m6TX2fKmztspOvjrA7yFZN7ytpVirY,754
308
+ wandb/sdk/artifacts/_generated/fragments.py,sha256=a8g-DXyFpl9OBynzVwcCy11kQ4MLgHDuz2kposHMz1M,14920
292
309
  wandb/sdk/artifacts/_generated/input_types.py,sha256=okRnv1R4lNnXzDPBGtatjcDqPoLxuJdb6EUKfrCgGw4,1560
293
- wandb/sdk/artifacts/_generated/link_artifact.py,sha256=x1ICISGCXjQqZTM9dJqxYZfxzopIB8Pi9dnBuKqJy3E,477
310
+ wandb/sdk/artifacts/_generated/link_artifact.py,sha256=PLI1mC_yoQuhLXLzRqpcUdhKcniS3rlXFe3Qf6WoBZs,693
294
311
  wandb/sdk/artifacts/_generated/move_artifact_collection.py,sha256=NFdT8EYWz_yPquFRtqiX-ljaBbmVx5Iav0loricc1vM,997
295
- wandb/sdk/artifacts/_generated/operations.py,sha256=CtAN9UjJCcl4_Cz0Evg9_mF5Kl8MaMwuMPcnxvABjhk,13655
312
+ wandb/sdk/artifacts/_generated/operations.py,sha256=i6_8BSqvbwbIGvC_W7znpT_iyq89A5BFfVFsecLnufs,24982
296
313
  wandb/sdk/artifacts/_generated/project_artifact_collection.py,sha256=zWxM_YdZI9oMuZKZW-XqAV_HhsGwiUUYQwOIagS9HoU,3494
297
314
  wandb/sdk/artifacts/_generated/project_artifact_collections.py,sha256=wVTGJpaoT5TPbXb2qgFELRZ7DaHv4CuXy0BMDrWgKHE,898
298
315
  wandb/sdk/artifacts/_generated/project_artifact_type.py,sha256=5hUK0eMCwjAoAz6HpG9KfPtzZdIexZjmpXXw4A8BQpI,567
299
316
  wandb/sdk/artifacts/_generated/project_artifact_types.py,sha256=WSyev_foF5C7noNc-bbygsHzLd4nDpqjhxzwHEMQdO0,566
300
317
  wandb/sdk/artifacts/_generated/project_artifacts.py,sha256=cKdM9B35nOAKiS4A4AdhIpBJ8NQaZfo8vwgNl_EZlVU,1170
318
+ wandb/sdk/artifacts/_generated/registry_collections.py,sha256=boJA9IqLTdd9YFP18eqZLaveM5FUKZMMytCcMdtHBOE,869
319
+ wandb/sdk/artifacts/_generated/registry_versions.py,sha256=ztU-QwGLv72--EG7TTyDVM6LskZEENoaE8_Mq6GYrEE,839
301
320
  wandb/sdk/artifacts/_generated/run_input_artifacts.py,sha256=42doBI4GjApQZT8dBr9T7s3zGYCZAfJYMVAo0_UM_dc,1457
302
321
  wandb/sdk/artifacts/_generated/run_output_artifacts.py,sha256=0Icucf5mPEVQvAsqPyb4MljNp7g58Urj8aRx16fEM04,1483
322
+ wandb/sdk/artifacts/_generated/unlink_artifact.py,sha256=_qudNG3GOmTqxVkCYup3HTScWIIGNezkH1brThIo76M,596
303
323
  wandb/sdk/artifacts/_generated/update_artifact.py,sha256=0nDR_DR09kJam3J3vhHRuvt1qnzEWCbW8rG9awpSydA,566
304
324
  wandb/sdk/artifacts/_generated/update_artifact_portfolio.py,sha256=qASaZLHbYR_i3b46pb_LwQGMj2WS6vqaw5qysnIL2Io,1013
305
325
  wandb/sdk/artifacts/_generated/update_artifact_sequence.py,sha256=HlR4Q6911IaOzQSdJG4ldfaY-WnMPHY7Fo284mgNrOU,1001
306
326
  wandb/sdk/artifacts/artifact_manifests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
307
- wandb/sdk/artifacts/artifact_manifests/artifact_manifest_v1.py,sha256=YQ1SdU7e5Cl-9Ga9cC25XMOWK6rzLZ8eZq-mjNr-3lE,3535
327
+ wandb/sdk/artifacts/artifact_manifests/artifact_manifest_v1.py,sha256=r0K0ZPe2-c7g6v3XnIwbcywIIUpCxmIBLsWyrtOg_7Q,3591
308
328
  wandb/sdk/artifacts/storage_handlers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
309
329
  wandb/sdk/artifacts/storage_handlers/azure_handler.py,sha256=D3YwnAvY94GRNCNFtxuZSUSSrV8gFCAwWjFG5XosamI,8547
310
330
  wandb/sdk/artifacts/storage_handlers/gcs_handler.py,sha256=itARuzT6VhIfjP9SLT4_HS-oBesl5JkvFWdaEWXabeQ,8738
311
- wandb/sdk/artifacts/storage_handlers/http_handler.py,sha256=sRdLyqO24kDaM9ccD3fFfy1UQHKIo72XNp3jeQhrD4s,4184
331
+ wandb/sdk/artifacts/storage_handlers/http_handler.py,sha256=KiyCD4W_2EB-ILWpjrnJBJ2FjRWKYalpR2HKfO2Rn6k,4147
312
332
  wandb/sdk/artifacts/storage_handlers/local_file_handler.py,sha256=XAz2Q8Kc06aWA89cikAyjI8qkZ-rDh7HQz5FCB0vSy0,5612
313
333
  wandb/sdk/artifacts/storage_handlers/multi_handler.py,sha256=m5t_fxXy8Gsk7wnZnGpU-D3gvweHERcNRIZPK-kL380,1898
314
- wandb/sdk/artifacts/storage_handlers/s3_handler.py,sha256=tE8ClD5DclrHOefpIykeFj5u6EucPdAi3xyXncDcjbg,13237
334
+ wandb/sdk/artifacts/storage_handlers/s3_handler.py,sha256=MubZIGrRChOPy4v9KJgMuGPy_Sy59Asx3uMchTz55m8,13200
315
335
  wandb/sdk/artifacts/storage_handlers/tracking_handler.py,sha256=u9F5fegy-a3O1gwntsRWEX_Ah09T5GCzP5Igw2JILGk,2555
316
336
  wandb/sdk/artifacts/storage_handlers/wb_artifact_handler.py,sha256=XgYdqZMrMk3atIDc0wOgtoCdEijXUm8KjcFjS3YYvB0,4895
317
337
  wandb/sdk/artifacts/storage_handlers/wb_local_artifact_handler.py,sha256=WP_7fqMSHWJoLTG5g042rLrbdfYn3F3VkV3rsaV_hMA,2636
318
338
  wandb/sdk/artifacts/storage_policies/__init__.py,sha256=G8quZY8-eynVVXmNBbiLGfUoI2P1rOE-LOmpzOwNJe0,230
319
339
  wandb/sdk/artifacts/storage_policies/register.py,sha256=azfof-H42vIuvndo9hvN4cZ3UXWG-nZcrFQ1QFL9oIc,50
320
- wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py,sha256=atd1-Z64jMb0fv1toj3ijfZRvdaAsL0a1ueXWLQFNpc,22754
340
+ wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py,sha256=Wv8uHIpydOVFNszC8neWStqd6e2pLQ2KouzoZ7O4364,22717
321
341
  wandb/sdk/backend/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
322
342
  wandb/sdk/backend/backend.py,sha256=hir04WylvBJptctooIpNeZIbZzHoNFBkLFLXrbUxhPY,1367
323
343
  wandb/sdk/data_types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -337,7 +357,7 @@ wandb/sdk/data_types/table.py,sha256=tQ9hBXkHFfoFTaRMej05FCkLJEmAlcO7Hr1NRWNb6fo
337
357
  wandb/sdk/data_types/table_decorators.py,sha256=Cv8iM4Z_IIsUFltrAK_pwFEFaBXY1cilCNrfHnMP8IY,3835
338
358
  wandb/sdk/data_types/trace_tree.py,sha256=AySSd4CGi99K_8MflBFZDYidCL7AurxTk4fWrG9c8JI,15278
339
359
  wandb/sdk/data_types/utils.py,sha256=Hb4J2RDXPvVQcjs5LHGjudW082oSU7rBOqthK1zS-rM,9222
340
- wandb/sdk/data_types/video.py,sha256=6kH4bhSxylgZUnp3V92VJMDaFHStFPvX4oAw41c_zjs,10920
360
+ wandb/sdk/data_types/video.py,sha256=2VEquCa7rtmU_GcXhuiK-axnJuXFPUa5pQ-z6u4ZHUs,10908
341
361
  wandb/sdk/data_types/base_types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
342
362
  wandb/sdk/data_types/base_types/json_metadata.py,sha256=idI3dNB1e348AefsQC1GNpVqYl2raWzJmDfl2ITL-_w,1608
343
363
  wandb/sdk/data_types/base_types/media.py,sha256=x2y7-zZYTAa5bL5EcfYtX78y5fj0betjHbYZWSVbNjY,12907
@@ -352,14 +372,9 @@ wandb/sdk/integration_utils/data_logging.py,sha256=DtSEZB-TzxQKhjm9IXNxDeOAUZyDX
352
372
  wandb/sdk/interface/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
353
373
  wandb/sdk/interface/constants.py,sha256=GKZ2hATTBCt7yU8kloafZR34yLChiI36O_VtxZZbQyg,64
354
374
  wandb/sdk/interface/interface.py,sha256=IXw6f43drG1XviWnZNBHNJgI4SV4uhglAhlhqkhlwK0,38944
355
- wandb/sdk/interface/interface_queue.py,sha256=tgK-OtoX2JMi0hvKwH0fd6lQkiYd3565RSalkWHmY6k,1300
356
- wandb/sdk/interface/interface_relay.py,sha256=OxzeDHMrC4qC6WNUHWGJi1QOpuHAIw-Yyom3tbGO91w,905
357
- wandb/sdk/interface/interface_shared.py,sha256=o79nFBM1vvn1-N4R-TH6IjtBO0XkM-2P_H36wWvQ4mA,18716
358
- wandb/sdk/interface/interface_sock.py,sha256=olNlc54pCCIYn6jHKO_YY8qCVoLASnyQ3qDsJ2-VRfg,1082
359
- wandb/sdk/interface/router.py,sha256=RvHVVFB3NGpzMiJrz1q1iDLtARAhKBTp7qIC_9SA_rY,2719
360
- wandb/sdk/interface/router_queue.py,sha256=DNrkDO73rR5ThyXVa03gzfP94QIjBrWhy3rKPXHqj-0,1092
361
- wandb/sdk/interface/router_relay.py,sha256=kaHeBTFvPFr2rln3XrT2XpnVfPpPQQfyiRBXyKUy18w,1440
362
- wandb/sdk/interface/router_sock.py,sha256=0lZp7_Ksle1wZFAWUp49Yr6hLyUBNh9_xLh5ObiwN2Q,1034
375
+ wandb/sdk/interface/interface_queue.py,sha256=SQtGYBbSIH2jt34dzZsl4cv5mQ320kO-zOv6QHB1SIE,1200
376
+ wandb/sdk/interface/interface_shared.py,sha256=01UjDHXexNYM4rfWAtCxO7FdOCQ4xidV4MPGFqHoe6w,18209
377
+ wandb/sdk/interface/interface_sock.py,sha256=hx0UW7D3WX0YSNcM_Mc58lM9_0Fw56VA8oPgIGRw3KI,1426
363
378
  wandb/sdk/interface/summary_record.py,sha256=NZOachyUirH7MATnxNUu5PVfVvDHNUFbEkCvRyoEFUo,1744
364
379
  wandb/sdk/internal/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
365
380
  wandb/sdk/internal/context.py,sha256=HNcOpyptz_03YGlZV3BVzPrBXDoMhtLBxvM2-Z_5xXs,2607
@@ -376,7 +391,7 @@ wandb/sdk/internal/run.py,sha256=gY_cGt1_xkTBfynDSz1dMJrPKicXlEGQce4MhfmOBxI,651
376
391
  wandb/sdk/internal/sample.py,sha256=tVNzrLatHr8P1kbVzw8bWhLpqZxx7zdm4fgv7rv2hO0,2540
377
392
  wandb/sdk/internal/sender.py,sha256=xSB4b3NmsA6PEb3xD1RnmrweW6NAja1id_pT14Bq5Rg,66872
378
393
  wandb/sdk/internal/sender_config.py,sha256=rDsETNuJzIDg6PyI1tjpJ6ZtsrWtGRxbJGIagZiCO-8,7336
379
- wandb/sdk/internal/settings_static.py,sha256=EtAN0OjTHu8wCM-wWaI1X6v0wDTW66PZEAmCPSzggqo,4405
394
+ wandb/sdk/internal/settings_static.py,sha256=A8PYbBOMWbYgNv6yES60Akm93rLSpTlt-Sryl5hkH8A,4367
380
395
  wandb/sdk/internal/tb_watcher.py,sha256=yfZBz_6qCVB93fVOZMgvcS2nZ2z0RuLP0SOnPV95xZc,19195
381
396
  wandb/sdk/internal/thread_local_settings.py,sha256=f6uzjTa0d6twZ9aFHzdUlUqv3gy6AnbOZoWYm95DxKA,545
382
397
  wandb/sdk/internal/_generated/__init__.py,sha256=w4dmU4U0gMfBLGu8o8UPylYZyNxr3py7ELlJP5iHvPA,396
@@ -388,7 +403,7 @@ wandb/sdk/launch/__init__.py,sha256=70GMH3jQPioNeidxTZUuuT8_8Gxjhwnw9cNCSTqvFj0,
388
403
  wandb/sdk/launch/_launch.py,sha256=acZiTr8l4D3-kGW2R5zdqmk12wMX4EcZo9Prxe2MP5E,12146
389
404
  wandb/sdk/launch/_launch_add.py,sha256=xeXunxOqeggIrOKd9U415SBaGeYwMP1_osEymWtSBkw,9050
390
405
  wandb/sdk/launch/_project_spec.py,sha256=UAtAg2dKGI-FujAgSl6NcijRkxo4upzjWAxah0x-LXQ,22196
391
- wandb/sdk/launch/create_job.py,sha256=ehQtIdDZ8B0gxyrPhdS-09UdfvaXnYv4AVlQxYiRe8U,17987
406
+ wandb/sdk/launch/create_job.py,sha256=p4bifnzmnZWymw5bPZqfmlttKtpQuhspH9xSn0VWJu8,18295
392
407
  wandb/sdk/launch/errors.py,sha256=qE6PTcZilrIMBLOy5v3I5xp4Ex-Nf0HxsiVrnJ99AlE,288
393
408
  wandb/sdk/launch/git_reference.py,sha256=5pswecUCOOo2UUrfA5I9q6zrFe80M5IGODLNzXxDlgo,3867
394
409
  wandb/sdk/launch/loader.py,sha256=gka4OPM9Co3xyjNXFkrHW2IgRHrAMZqqqkiLx4E-YpE,9176
@@ -426,7 +441,7 @@ wandb/sdk/launch/registry/local_registry.py,sha256=TVgbQMGhIDVQdoefmLWH3kIOCAxQ5
426
441
  wandb/sdk/launch/runner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
427
442
  wandb/sdk/launch/runner/abstract.py,sha256=h-WO2p2yOeiMHNAvyq-Hh9_1jZE1WnTOmE9HebKcGLo,5804
428
443
  wandb/sdk/launch/runner/kubernetes_monitor.py,sha256=pLVdUSaS4pZcwcUJTzKcFe9nvpeGuOC2-Gz2JKeAaCU,18298
429
- wandb/sdk/launch/runner/kubernetes_runner.py,sha256=qp3h7Lz52gyakCHSyUHeZgJ_lLq1w8OnwALURTAXuVk,49385
444
+ wandb/sdk/launch/runner/kubernetes_runner.py,sha256=0N5nwwftBmR3q3A8ZUH7GtljXKhhYe_rRyRrHhH8zjA,49009
430
445
  wandb/sdk/launch/runner/local_container.py,sha256=FNkALxDBhX-wKoxp2VpMaNyxbfnlrmGSOjnPQHdJfTg,10659
431
446
  wandb/sdk/launch/runner/local_process.py,sha256=sGd4wdHX9hANoOUi4KToWL8zJbbozRtaNx_4hBHgl8g,2743
432
447
  wandb/sdk/launch/runner/sagemaker_runner.py,sha256=Nxu6_pX8HhCqqvQQIERj7HqaNKKJrs5MwontCzePNJ0,15692
@@ -437,7 +452,8 @@ wandb/sdk/launch/sweeps/scheduler_sweep.py,sha256=0iZPUWcChhtXC2IQWxUVAAlE6eT2og
437
452
  wandb/sdk/launch/sweeps/utils.py,sha256=ORipWOX7Pg6u4saJ_WG1ipr8lSiuMh2D0Tz1WCyrM6c,10553
438
453
  wandb/sdk/lib/__init__.py,sha256=_sOt85qPxtPyM_LaN0IE6dO1CImzwXJXzVHC7R24VBE,188
439
454
  wandb/sdk/lib/apikey.py,sha256=a-ikKRlXqOPSBisn46WnjbmSfvPCxaHb1eZ35agV8X4,11431
440
- wandb/sdk/lib/asyncio_compat.py,sha256=lpAtYKk4xgEpLB3RMJz4WcHdOeoBRm2a5a7HRTTIYNA,7020
455
+ wandb/sdk/lib/asyncio_compat.py,sha256=B4srpIKFLSPQiVzNpjwPUjSBUZQtV-9ifx_akmZUR-c,7206
456
+ wandb/sdk/lib/asyncio_manager.py,sha256=CJd-Ef_0kF8k_Kv42RZGP3ftczDEDHTEYhSdnRx9K0o,9012
441
457
  wandb/sdk/lib/capped_dict.py,sha256=HuFhPHl0e_pK6ETDxYh5RIPO-46I3EjXgzFdqbJTXDs,846
442
458
  wandb/sdk/lib/config_util.py,sha256=vPvxiRBdBT_AT9PS0ZimbvqXr8hPjk58fhrn11lDv40,2994
443
459
  wandb/sdk/lib/console_capture.py,sha256=QJT90F8b33NcqgE-GXM_11f1JrzbNsrQWOCKEgO-dv0,6841
@@ -452,7 +468,7 @@ wandb/sdk/lib/fsm.py,sha256=nhcbQ4lO2iGWuaYkxYvl3Z0OdKIksn6Ahzj_o82ZYpM,5209
452
468
  wandb/sdk/lib/gitlib.py,sha256=HRsKW3zvlSsDRvmrWJqtuKmf2QGqV0Y8_pQeFC8hzPQ,8072
453
469
  wandb/sdk/lib/gql_request.py,sha256=41aK9xpHxvcO3Sy1cPFm0Py-fWqD3BQQ_4MEppxBYG0,2464
454
470
  wandb/sdk/lib/handler_util.py,sha256=FizfOj-9I-sY92H84uwGInUe7wE8vlU_uQl7-SGI01Y,610
455
- wandb/sdk/lib/hashutil.py,sha256=yIFwl-Jv0CwDxt-MhYYfpbjntLTv3GO0MxMbmZn4uHk,3226
471
+ wandb/sdk/lib/hashutil.py,sha256=MerxylcZ9nbFCpQ66T2pZKzA5nBRiYuf-lj6q2j7rzg,3698
456
472
  wandb/sdk/lib/import_hooks.py,sha256=oJBolf3pWRGdzpeG9VErDYbhicHsTDWdOgNxauvuCqw,10546
457
473
  wandb/sdk/lib/interrupt.py,sha256=_m7yu7q-qJD_MmqNb5BiYapP5h6LFnKRWxCJWePBz9U,1390
458
474
  wandb/sdk/lib/ipython.py,sha256=xQ-B7P14XqiSBZLBd-2tmvPvU_-iUw_uxIkvGFK7qYM,3936
@@ -461,31 +477,31 @@ wandb/sdk/lib/lazyloader.py,sha256=MoMgx_tBjA__yFKcYzhimWiug_TSQeRUr71sPNUkTsk,1
461
477
  wandb/sdk/lib/module.py,sha256=xfs4GCwUjr6CV1bz6BRw9wulcn0aGrG1nSMMajEYwmQ,1988
462
478
  wandb/sdk/lib/paths.py,sha256=Knkww9wdMK4wqsj5B9aMEJDvIFoCf80xXl28It3FIG4,4635
463
479
  wandb/sdk/lib/preinit.py,sha256=IDK_WXbcrfzXUNWZur505lHIY_cYs1IEWp26HMpIf74,1492
464
- wandb/sdk/lib/printer.py,sha256=YCkBdW4scPPoJ_Z017B9VTTWBpK0gC0jq0U4Lg8pF28,16828
465
- wandb/sdk/lib/printer_asyncio.py,sha256=fAQILyGBUzZYxUNPQ9_UQkmy5ujbU8zunZFhtLtRT-M,1470
480
+ wandb/sdk/lib/printer.py,sha256=aArbmWSLVz6CXo4JcnuEkbNjekqVj9NVaobajhJt954,16813
481
+ wandb/sdk/lib/printer_asyncio.py,sha256=kmknEO3UDvceZvCq18Y2pH_6K79kJ7XFV9pTTnNfTwI,1545
466
482
  wandb/sdk/lib/progress.py,sha256=NbQ3NTy80Ekyhe_lAL7-uT_6O_kX4YniakK5kWaDA3Q,10341
467
483
  wandb/sdk/lib/proto_util.py,sha256=YaGg9FoKtWmgQD8SkkKN630gyG93WoYY5JHqwdWaQKg,2984
468
484
  wandb/sdk/lib/redirect.py,sha256=kribtu1VYPwIfdTbSwP7roFyrfd6CBDnNROyRUnAkLY,28249
469
- wandb/sdk/lib/retry.py,sha256=NYLAkjBudM6UlnjViTBe10lR9onTXFCfSvix5SCEtn0,10303
485
+ wandb/sdk/lib/retry.py,sha256=A9UhcdkSvvCxECcjswS1xRb4EkDtbvPeGY-rLMxeH78,13474
470
486
  wandb/sdk/lib/run_moment.py,sha256=wXlPOeRDlypFwwy_uG-07qe8s9pYC84x374hTR4zs0Q,2501
471
487
  wandb/sdk/lib/runid.py,sha256=rHYRTjJu8gTZ6Aoo0WZ5jQfAtNLXQo6aY6PD-i3Fh6I,404
472
488
  wandb/sdk/lib/server.py,sha256=f8idM8TiJKS1nYTjijhVkzOTp8e2flNpLUWcZ2K08f0,1681
473
- wandb/sdk/lib/sock_client.py,sha256=5jPF3wcDtNzsZS4v3pMa5y5uJY3ggS_ASGDM5ln1Tto,8344
474
489
  wandb/sdk/lib/sparkline.py,sha256=CivfHHGPrbnnacpfjsoYUrCtX6Xz7AHoybEeOuWeEI0,1407
475
490
  wandb/sdk/lib/telemetry.py,sha256=25ZdppATjPlRR2uTh_lLUgXVCAFS49hT8ArSYAWCN3g,2854
476
491
  wandb/sdk/lib/timed_input.py,sha256=XF03SXTQj0AysHiIV-LKtbwxtSUx0E7xts7zDPs9kJQ,3362
477
492
  wandb/sdk/lib/timer.py,sha256=Ar1t8f3OFAA4PB2fB2MT9D41y3g2Or56wSAYezvdXqo,459
478
493
  wandb/sdk/lib/wb_logging.py,sha256=riPp_MSnb57YCcdCGoroCpyC1olAkIGxZge0llyVJOA,5000
479
494
  wandb/sdk/lib/service/ipc_support.py,sha256=lEhZ0XNBf1avUf5PDXEIRpz9JwrPiJ9cijrXDhvuO0g,354
480
- wandb/sdk/lib/service/service_connection.py,sha256=3LIakvkrAEk4uM_Q2DN76__m0gtolsuMWPv_9Fg4i8o,6632
495
+ wandb/sdk/lib/service/service_client.py,sha256=mbYgWTMi2bmes6BC8Gpl3NmqWnjjbDpOuZp5dxG9i8U,3324
496
+ wandb/sdk/lib/service/service_connection.py,sha256=Dk7JO39V6j1c1eTmu7iVP7RORT4ua0Yvryu5k6_QEwg,6259
481
497
  wandb/sdk/lib/service/service_port_file.py,sha256=HAWdSBEIGtvJ8ZQJczSD7wrcHZQXZQ6MnVG73hPAyXs,3048
482
498
  wandb/sdk/lib/service/service_process.py,sha256=XKNRLBSNN2R8iI12imYDIkC5x-ekndaAKD1n80zjxtQ,3211
483
- wandb/sdk/lib/service/service_token.py,sha256=2HIYE4re66QybikT3fSKqD80w1DmdUCLi8j6mgbZtm0,4840
499
+ wandb/sdk/lib/service/service_token.py,sha256=bDkHGJvpBprW1Ie_IkzXROu9eJkMd5k1cikdlmTUfdo,5288
484
500
  wandb/sdk/mailbox/__init__.py,sha256=0gYfvSzPYluoQOQzT3j2AQoE4R4gPU2FCXBYcEMJOP8,879
485
- wandb/sdk/mailbox/mailbox.py,sha256=94heh_4IaPRYviea26hT-NFk7vhFOhZITG1stTe2LRs,4449
486
- wandb/sdk/mailbox/mailbox_handle.py,sha256=ckbl3G2SaufhwtnJUbjCqqIBtpgN4nr_JG70ih7KYO4,3781
487
- wandb/sdk/mailbox/response_handle.py,sha256=pKWqtDCj5BNnSzS0Ni-UHObn6hmxjT2D2pRHDWJ3pN4,5109
488
- wandb/sdk/mailbox/wait_with_progress.py,sha256=BfkNHbF8kkwLnyRhrcVMeoFTaotJijkmuZQpSnAvffI,4261
501
+ wandb/sdk/mailbox/mailbox.py,sha256=A44lHvE4GTBrFPJzeu44yxVpuUwqfpf1cUoCxltFvp0,4949
502
+ wandb/sdk/mailbox/mailbox_handle.py,sha256=c49ZvyOUSXn6Nj_vGkZ3DyRSL09oKRLDE_1NEZ1EubY,4143
503
+ wandb/sdk/mailbox/response_handle.py,sha256=4y6CI1b-miL_PZ-lQh3R4UhlwArEEZC4fifRC9xU8z0,2915
504
+ wandb/sdk/mailbox/wait_with_progress.py,sha256=D5mb5afZkvFgEmTFR_u7s0Y_m3NIBzHRqnK_6SPqdJI,3190
489
505
  wandb/sdk/projects/_generated/__init__.py,sha256=YlxTcR5iXbhPQT6eKPLxm_KMo8ZptY3mflKfBkEOBZg,1350
490
506
  wandb/sdk/projects/_generated/delete_project.py,sha256=aUZhcp8bvoNlUmWqbHT2PsxZTMR85J7GdbAVwLKI10Q,517
491
507
  wandb/sdk/projects/_generated/enums.py,sha256=n4NAp3Y6Cb0bOtRvpJ-v8tl37-pg5CBEZ7m8Wzg7THY,128
@@ -881,8 +897,8 @@ wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/importlib2.py,sha256=kX0rdVmTDL
881
897
  wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/platform.py,sha256=7fpTDfxSYvSRtHvyog-plRdLR5A6k1QVY_AL0gVhhPM,1563
882
898
  wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/unicode_paths.py,sha256=xzyQmuba2gns1s3Qemu9SXaKV5zeTL3TP9--xOi541g,2254
883
899
  wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/win32stat.py,sha256=R48kuuEIi7XzCJBJ6Xo7v6DJIbOP5EwcsWaPf5Axn_g,3951
884
- wandb-0.21.1.dist-info/METADATA,sha256=-Qnky98Yncsm3DQyzFMySPaegJlaln53YmjSofAYTN8,10250
885
- wandb-0.21.1.dist-info/WHEEL,sha256=vGlXFq5Cg2SEc12yCQt0M53oxbuIdJrfMMMiwCzLXhI,93
886
- wandb-0.21.1.dist-info/entry_points.txt,sha256=v4FCOZ9gW7Pc6KLsmgQqpCiKTrA1wh2XHmNf-NUP1-I,67
887
- wandb-0.21.1.dist-info/licenses/LICENSE,sha256=rJ7p1acqNi17WFOAJ9WqsImXZtKZDA3i_gzdDVGRuFQ,1102
888
- wandb-0.21.1.dist-info/RECORD,,
900
+ wandb-0.21.2.dist-info/METADATA,sha256=Em3fwcRKRLmCCxH81amjaAmKAwnzo4C1mTyF4kYM1KM,10250
901
+ wandb-0.21.2.dist-info/WHEEL,sha256=vGlXFq5Cg2SEc12yCQt0M53oxbuIdJrfMMMiwCzLXhI,93
902
+ wandb-0.21.2.dist-info/entry_points.txt,sha256=v4FCOZ9gW7Pc6KLsmgQqpCiKTrA1wh2XHmNf-NUP1-I,67
903
+ wandb-0.21.2.dist-info/licenses/LICENSE,sha256=rJ7p1acqNi17WFOAJ9WqsImXZtKZDA3i_gzdDVGRuFQ,1102
904
+ wandb-0.21.2.dist-info/RECORD,,
@@ -1,38 +0,0 @@
1
- """InterfaceRelay - Derived from InterfaceQueue using RelayRouter to preserve uuid req/resp.
2
-
3
- See interface.py for how interface classes relate to each other.
4
- """
5
-
6
- from __future__ import annotations
7
-
8
- import logging
9
- from typing import TYPE_CHECKING
10
-
11
- from wandb.proto import wandb_internal_pb2 as pb
12
- from wandb.sdk.mailbox import Mailbox
13
-
14
- from .interface_queue import InterfaceQueue
15
-
16
- if TYPE_CHECKING:
17
- from queue import Queue
18
-
19
-
20
- logger = logging.getLogger("wandb")
21
-
22
-
23
- class InterfaceRelay(InterfaceQueue):
24
- _mailbox: Mailbox
25
-
26
- def __init__(
27
- self,
28
- mailbox: Mailbox,
29
- record_q: Queue[pb.Record],
30
- result_q: Queue[pb.Result],
31
- relay_q: Queue[pb.Result],
32
- ) -> None:
33
- self.relay_q = relay_q
34
- super().__init__(
35
- record_q=record_q,
36
- result_q=result_q,
37
- mailbox=mailbox,
38
- )
@@ -1,89 +0,0 @@
1
- """Router - handle message router (base class).
2
-
3
- Router to manage responses.
4
-
5
- """
6
-
7
- from __future__ import annotations
8
-
9
- import logging
10
- import threading
11
- from abc import abstractmethod
12
- from typing import TYPE_CHECKING
13
-
14
- from wandb.proto import wandb_internal_pb2 as pb
15
- from wandb.proto import wandb_server_pb2 as spb
16
- from wandb.sdk import mailbox
17
-
18
- if TYPE_CHECKING:
19
- from queue import Queue
20
-
21
-
22
- logger = logging.getLogger("wandb")
23
-
24
-
25
- class MessageRouterClosedError(Exception):
26
- """Router has been closed."""
27
-
28
-
29
- class MessageRouter:
30
- _request_queue: Queue[pb.Record]
31
- _response_queue: Queue[pb.Result]
32
- _mailbox: mailbox.Mailbox | None
33
-
34
- def __init__(self, mailbox: mailbox.Mailbox | None = None) -> None:
35
- self._mailbox = mailbox
36
- self._lock = threading.Lock()
37
-
38
- self._join_event = threading.Event()
39
- self._thread = threading.Thread(target=self.message_loop)
40
- self._thread.name = "MsgRouterThr"
41
- self._thread.daemon = True
42
- self._thread.start()
43
-
44
- @abstractmethod
45
- def _read_message(self) -> pb.Result | spb.ServerResponse | None:
46
- raise NotImplementedError
47
-
48
- @abstractmethod
49
- def _send_message(self, record: pb.Record) -> None:
50
- raise NotImplementedError
51
-
52
- def message_loop(self) -> None:
53
- try:
54
- while not self._join_event.is_set():
55
- try:
56
- msg = self._read_message()
57
- except EOFError:
58
- # On abnormal shutdown the queue will be destroyed underneath
59
- # resulting in EOFError. message_loop needs to exit..
60
- logger.warning("EOFError seen in message_loop")
61
- break
62
- except MessageRouterClosedError as e:
63
- logger.warning("message_loop has been closed", exc_info=e)
64
- break
65
- if not msg:
66
- continue
67
- self._handle_msg_rcv(msg)
68
-
69
- finally:
70
- if self._mailbox:
71
- self._mailbox.close()
72
-
73
- def join(self) -> None:
74
- self._join_event.set()
75
- self._thread.join()
76
-
77
- def _handle_msg_rcv(self, msg: pb.Result | spb.ServerResponse) -> None:
78
- if not self._mailbox:
79
- return
80
-
81
- if isinstance(msg, pb.Result) and msg.control.mailbox_slot:
82
- self._mailbox.deliver(
83
- spb.ServerResponse(
84
- request_id=msg.control.mailbox_slot,
85
- result_communicate=msg,
86
- )
87
- )
88
- elif isinstance(msg, spb.ServerResponse) and msg.request_id:
89
- self._mailbox.deliver(msg)
@@ -1,43 +0,0 @@
1
- """Router - handle message router (queue).
2
-
3
- Router to manage responses from a queue.
4
-
5
- """
6
-
7
- from __future__ import annotations
8
-
9
- import queue
10
- from typing import TYPE_CHECKING
11
-
12
- from wandb.proto import wandb_internal_pb2 as pb
13
- from wandb.sdk.mailbox import Mailbox
14
-
15
- from .router import MessageRouter
16
-
17
- if TYPE_CHECKING:
18
- from queue import Queue
19
-
20
-
21
- class MessageQueueRouter(MessageRouter):
22
- _request_queue: Queue[pb.Record]
23
- _response_queue: Queue[pb.Result]
24
-
25
- def __init__(
26
- self,
27
- request_queue: Queue[pb.Record],
28
- response_queue: Queue[pb.Result],
29
- mailbox: Mailbox | None = None,
30
- ) -> None:
31
- self._request_queue = request_queue
32
- self._response_queue = response_queue
33
- super().__init__(mailbox=mailbox)
34
-
35
- def _read_message(self) -> pb.Result | None:
36
- try:
37
- msg = self._response_queue.get(timeout=1)
38
- except queue.Empty:
39
- return None
40
- return msg
41
-
42
- def _send_message(self, record: pb.Record) -> None:
43
- self._request_queue.put(record)
@@ -1,50 +0,0 @@
1
- """Router - handle message router (relay).
2
-
3
- Router to manage responses from a queue with relay.
4
-
5
- """
6
-
7
- from __future__ import annotations
8
-
9
- from typing import TYPE_CHECKING
10
-
11
- from wandb.proto import wandb_internal_pb2 as pb
12
- from wandb.proto import wandb_server_pb2 as spb
13
- from wandb.sdk.mailbox import Mailbox
14
-
15
- from .router_queue import MessageQueueRouter
16
-
17
- if TYPE_CHECKING:
18
- from queue import Queue
19
-
20
-
21
- class MessageRelayRouter(MessageQueueRouter):
22
- _relay_queue: Queue[pb.Result]
23
-
24
- def __init__(
25
- self,
26
- request_queue: Queue[pb.Record],
27
- response_queue: Queue[pb.Result],
28
- relay_queue: Queue[pb.Result],
29
- mailbox: Mailbox,
30
- ) -> None:
31
- self._relay_queue = relay_queue
32
- super().__init__(
33
- request_queue=request_queue,
34
- response_queue=response_queue,
35
- mailbox=mailbox,
36
- )
37
-
38
- def _handle_msg_rcv(self, msg: pb.Result | spb.ServerResponse) -> None:
39
- if isinstance(msg, pb.Result):
40
- relay_msg = msg
41
- else:
42
- relay_msg = msg.result_communicate
43
-
44
- # This is legacy-service logic for returning responses to the client.
45
- # A different thread reads the "relay queue" and writes responses on
46
- # the socket.
47
- if relay_msg.control.relay_id:
48
- self._relay_queue.put(relay_msg)
49
- else:
50
- super()._handle_msg_rcv(msg)
@@ -1,32 +0,0 @@
1
- """Router - handle message router (sock).
2
-
3
- Router to manage responses from a socket client.
4
-
5
- """
6
-
7
- from __future__ import annotations
8
-
9
- from wandb.proto import wandb_internal_pb2 as pb
10
- from wandb.proto import wandb_server_pb2 as spb
11
- from wandb.sdk.lib.sock_client import SockClient, SockClientClosedError
12
- from wandb.sdk.mailbox import Mailbox
13
-
14
- from .router import MessageRouter, MessageRouterClosedError
15
-
16
-
17
- class MessageSockRouter(MessageRouter):
18
- _sock_client: SockClient
19
- _mailbox: Mailbox
20
-
21
- def __init__(self, sock_client: SockClient, mailbox: Mailbox) -> None:
22
- self._sock_client = sock_client
23
- super().__init__(mailbox=mailbox)
24
-
25
- def _read_message(self) -> spb.ServerResponse | None:
26
- try:
27
- return self._sock_client.read_server_response(timeout=1)
28
- except SockClientClosedError as e:
29
- raise MessageRouterClosedError from e
30
-
31
- def _send_message(self, record: pb.Record) -> None:
32
- self._sock_client.send_record_communicate(record)