wandb 0.16.5__py3-none-any.whl → 0.17.0__py3-none-any.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (194) hide show
  1. package_readme.md +95 -0
  2. wandb/__init__.py +2 -3
  3. wandb/agents/pyagent.py +0 -1
  4. wandb/analytics/sentry.py +2 -1
  5. wandb/apis/importers/internals/internal.py +0 -1
  6. wandb/apis/importers/internals/protocols.py +30 -56
  7. wandb/apis/importers/mlflow.py +13 -26
  8. wandb/apis/importers/wandb.py +8 -14
  9. wandb/apis/internal.py +0 -3
  10. wandb/apis/public/api.py +55 -3
  11. wandb/apis/public/artifacts.py +1 -0
  12. wandb/apis/public/files.py +1 -0
  13. wandb/apis/public/history.py +1 -0
  14. wandb/apis/public/jobs.py +17 -4
  15. wandb/apis/public/projects.py +1 -0
  16. wandb/apis/public/reports.py +1 -0
  17. wandb/apis/public/runs.py +15 -17
  18. wandb/apis/public/sweeps.py +1 -0
  19. wandb/apis/public/teams.py +1 -0
  20. wandb/apis/public/users.py +1 -0
  21. wandb/apis/reports/v1/_blocks.py +3 -7
  22. wandb/apis/reports/v2/gql.py +1 -0
  23. wandb/apis/reports/v2/interface.py +3 -4
  24. wandb/apis/reports/v2/internal.py +5 -8
  25. wandb/cli/cli.py +95 -22
  26. wandb/data_types.py +9 -6
  27. wandb/docker/__init__.py +1 -1
  28. wandb/env.py +38 -8
  29. wandb/errors/__init__.py +5 -0
  30. wandb/errors/term.py +10 -2
  31. wandb/filesync/step_checksum.py +1 -4
  32. wandb/filesync/step_prepare.py +4 -24
  33. wandb/filesync/step_upload.py +4 -106
  34. wandb/filesync/upload_job.py +0 -76
  35. wandb/integration/catboost/catboost.py +1 -1
  36. wandb/integration/fastai/__init__.py +1 -0
  37. wandb/integration/huggingface/resolver.py +2 -2
  38. wandb/integration/keras/__init__.py +1 -0
  39. wandb/integration/keras/callbacks/metrics_logger.py +1 -1
  40. wandb/integration/keras/keras.py +7 -7
  41. wandb/integration/langchain/wandb_tracer.py +1 -0
  42. wandb/integration/lightning/fabric/logger.py +1 -3
  43. wandb/integration/metaflow/metaflow.py +41 -6
  44. wandb/integration/openai/fine_tuning.py +77 -40
  45. wandb/integration/prodigy/prodigy.py +1 -1
  46. wandb/old/summary.py +1 -1
  47. wandb/plot/confusion_matrix.py +1 -1
  48. wandb/plot/pr_curve.py +2 -1
  49. wandb/plot/roc_curve.py +2 -1
  50. wandb/{plots → plot}/utils.py +13 -25
  51. wandb/proto/v3/wandb_internal_pb2.py +364 -332
  52. wandb/proto/v3/wandb_settings_pb2.py +2 -2
  53. wandb/proto/v3/wandb_telemetry_pb2.py +10 -10
  54. wandb/proto/v4/wandb_internal_pb2.py +322 -316
  55. wandb/proto/v4/wandb_settings_pb2.py +2 -2
  56. wandb/proto/v4/wandb_telemetry_pb2.py +10 -10
  57. wandb/proto/wandb_deprecated.py +7 -1
  58. wandb/proto/wandb_internal_codegen.py +3 -29
  59. wandb/sdk/artifacts/artifact.py +51 -20
  60. wandb/sdk/artifacts/artifact_download_logger.py +1 -0
  61. wandb/sdk/artifacts/artifact_file_cache.py +18 -4
  62. wandb/sdk/artifacts/artifact_instance_cache.py +1 -0
  63. wandb/sdk/artifacts/artifact_manifest.py +1 -0
  64. wandb/sdk/artifacts/artifact_manifest_entry.py +7 -3
  65. wandb/sdk/artifacts/artifact_manifests/artifact_manifest_v1.py +1 -0
  66. wandb/sdk/artifacts/artifact_saver.py +18 -27
  67. wandb/sdk/artifacts/artifact_state.py +1 -0
  68. wandb/sdk/artifacts/artifact_ttl.py +1 -0
  69. wandb/sdk/artifacts/exceptions.py +1 -0
  70. wandb/sdk/artifacts/storage_handlers/azure_handler.py +1 -0
  71. wandb/sdk/artifacts/storage_handlers/gcs_handler.py +13 -18
  72. wandb/sdk/artifacts/storage_handlers/http_handler.py +1 -0
  73. wandb/sdk/artifacts/storage_handlers/local_file_handler.py +1 -0
  74. wandb/sdk/artifacts/storage_handlers/multi_handler.py +1 -0
  75. wandb/sdk/artifacts/storage_handlers/s3_handler.py +5 -3
  76. wandb/sdk/artifacts/storage_handlers/tracking_handler.py +1 -0
  77. wandb/sdk/artifacts/storage_handlers/wb_artifact_handler.py +1 -0
  78. wandb/sdk/artifacts/storage_handlers/wb_local_artifact_handler.py +1 -0
  79. wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py +3 -42
  80. wandb/sdk/artifacts/storage_policy.py +2 -12
  81. wandb/sdk/data_types/_dtypes.py +8 -8
  82. wandb/sdk/data_types/base_types/media.py +3 -6
  83. wandb/sdk/data_types/helper_types/bounding_boxes_2d.py +3 -1
  84. wandb/sdk/data_types/image.py +1 -1
  85. wandb/sdk/data_types/video.py +1 -1
  86. wandb/sdk/integration_utils/auto_logging.py +5 -6
  87. wandb/sdk/integration_utils/data_logging.py +10 -6
  88. wandb/sdk/interface/interface.py +86 -38
  89. wandb/sdk/interface/interface_shared.py +7 -13
  90. wandb/sdk/internal/datastore.py +1 -1
  91. wandb/sdk/internal/file_pusher.py +2 -5
  92. wandb/sdk/internal/file_stream.py +5 -18
  93. wandb/sdk/internal/handler.py +18 -2
  94. wandb/sdk/internal/internal.py +0 -1
  95. wandb/sdk/internal/internal_api.py +1 -129
  96. wandb/sdk/internal/internal_util.py +0 -1
  97. wandb/sdk/internal/job_builder.py +159 -45
  98. wandb/sdk/internal/profiler.py +1 -0
  99. wandb/sdk/internal/progress.py +0 -28
  100. wandb/sdk/internal/run.py +1 -0
  101. wandb/sdk/internal/sender.py +1 -2
  102. wandb/sdk/internal/system/assets/gpu_amd.py +44 -44
  103. wandb/sdk/internal/system/assets/gpu_apple.py +56 -11
  104. wandb/sdk/internal/system/assets/interfaces.py +6 -8
  105. wandb/sdk/internal/system/assets/open_metrics.py +2 -2
  106. wandb/sdk/internal/system/assets/trainium.py +1 -3
  107. wandb/sdk/launch/__init__.py +9 -1
  108. wandb/sdk/launch/_launch.py +9 -24
  109. wandb/sdk/launch/_launch_add.py +1 -3
  110. wandb/sdk/launch/_project_spec.py +188 -241
  111. wandb/sdk/launch/agent/agent.py +115 -48
  112. wandb/sdk/launch/agent/config.py +80 -14
  113. wandb/sdk/launch/builder/abstract.py +69 -1
  114. wandb/sdk/launch/builder/build.py +156 -555
  115. wandb/sdk/launch/builder/context_manager.py +235 -0
  116. wandb/sdk/launch/builder/docker_builder.py +8 -23
  117. wandb/sdk/launch/builder/kaniko_builder.py +161 -159
  118. wandb/sdk/launch/builder/noop.py +1 -0
  119. wandb/sdk/launch/builder/templates/dockerfile.py +92 -0
  120. wandb/sdk/launch/create_job.py +68 -63
  121. wandb/sdk/launch/environment/abstract.py +1 -0
  122. wandb/sdk/launch/environment/gcp_environment.py +1 -0
  123. wandb/sdk/launch/environment/local_environment.py +1 -0
  124. wandb/sdk/launch/inputs/files.py +148 -0
  125. wandb/sdk/launch/inputs/internal.py +217 -0
  126. wandb/sdk/launch/inputs/manage.py +95 -0
  127. wandb/sdk/launch/loader.py +1 -0
  128. wandb/sdk/launch/registry/abstract.py +1 -0
  129. wandb/sdk/launch/registry/azure_container_registry.py +1 -0
  130. wandb/sdk/launch/registry/elastic_container_registry.py +1 -0
  131. wandb/sdk/launch/registry/google_artifact_registry.py +2 -1
  132. wandb/sdk/launch/registry/local_registry.py +1 -0
  133. wandb/sdk/launch/runner/abstract.py +1 -0
  134. wandb/sdk/launch/runner/kubernetes_monitor.py +4 -1
  135. wandb/sdk/launch/runner/kubernetes_runner.py +9 -10
  136. wandb/sdk/launch/runner/local_container.py +2 -3
  137. wandb/sdk/launch/runner/local_process.py +8 -29
  138. wandb/sdk/launch/runner/sagemaker_runner.py +21 -20
  139. wandb/sdk/launch/runner/vertex_runner.py +8 -7
  140. wandb/sdk/launch/sweeps/scheduler.py +7 -4
  141. wandb/sdk/launch/sweeps/scheduler_sweep.py +2 -1
  142. wandb/sdk/launch/sweeps/utils.py +3 -3
  143. wandb/sdk/launch/utils.py +33 -140
  144. wandb/sdk/lib/_settings_toposort_generated.py +1 -5
  145. wandb/sdk/lib/fsm.py +8 -12
  146. wandb/sdk/lib/gitlib.py +4 -4
  147. wandb/sdk/lib/import_hooks.py +1 -1
  148. wandb/sdk/lib/lazyloader.py +0 -1
  149. wandb/sdk/lib/proto_util.py +23 -2
  150. wandb/sdk/lib/redirect.py +19 -14
  151. wandb/sdk/lib/retry.py +3 -2
  152. wandb/sdk/lib/run_moment.py +7 -1
  153. wandb/sdk/lib/tracelog.py +1 -1
  154. wandb/sdk/service/service.py +19 -16
  155. wandb/sdk/verify/verify.py +2 -1
  156. wandb/sdk/wandb_init.py +16 -63
  157. wandb/sdk/wandb_manager.py +2 -2
  158. wandb/sdk/wandb_require.py +5 -0
  159. wandb/sdk/wandb_run.py +164 -90
  160. wandb/sdk/wandb_settings.py +2 -48
  161. wandb/sdk/wandb_setup.py +1 -1
  162. wandb/sklearn/__init__.py +1 -0
  163. wandb/sklearn/plot/__init__.py +1 -0
  164. wandb/sklearn/plot/classifier.py +11 -12
  165. wandb/sklearn/plot/clusterer.py +2 -1
  166. wandb/sklearn/plot/regressor.py +1 -0
  167. wandb/sklearn/plot/shared.py +1 -0
  168. wandb/sklearn/utils.py +1 -0
  169. wandb/testing/relay.py +4 -4
  170. wandb/trigger.py +1 -0
  171. wandb/util.py +67 -54
  172. wandb/wandb_controller.py +2 -3
  173. wandb/wandb_torch.py +1 -2
  174. {wandb-0.16.5.dist-info → wandb-0.17.0.dist-info}/METADATA +67 -70
  175. {wandb-0.16.5.dist-info → wandb-0.17.0.dist-info}/RECORD +178 -188
  176. {wandb-0.16.5.dist-info → wandb-0.17.0.dist-info}/WHEEL +1 -2
  177. wandb/bin/apple_gpu_stats +0 -0
  178. wandb/catboost/__init__.py +0 -9
  179. wandb/fastai/__init__.py +0 -9
  180. wandb/keras/__init__.py +0 -18
  181. wandb/lightgbm/__init__.py +0 -9
  182. wandb/plots/__init__.py +0 -6
  183. wandb/plots/explain_text.py +0 -36
  184. wandb/plots/heatmap.py +0 -81
  185. wandb/plots/named_entity.py +0 -43
  186. wandb/plots/part_of_speech.py +0 -50
  187. wandb/plots/plot_definitions.py +0 -768
  188. wandb/plots/precision_recall.py +0 -121
  189. wandb/plots/roc.py +0 -103
  190. wandb/sacred/__init__.py +0 -3
  191. wandb/xgboost/__init__.py +0 -9
  192. wandb-0.16.5.dist-info/top_level.txt +0 -1
  193. {wandb-0.16.5.dist-info → wandb-0.17.0.dist-info}/entry_points.txt +0 -0
  194. {wandb-0.16.5.dist-info → wandb-0.17.0.dist-info/licenses}/LICENSE +0 -0
{wandb-0.16.5.dist-info → wandb-0.17.0.dist-info}/METADATA RENAMED
@@ -1,7 +1,10 @@
1
- Metadata-Version: 2.1
1
+ Metadata-Version: 2.3
2
2
  Name: wandb
3
- Version: 0.16.5
3
+ Version: 0.17.0
4
4
  Summary: A CLI and library for interacting with the Weights & Biases API.
5
+ Project-URL: Source, https://github.com/wandb/wandb
6
+ Project-URL: Bug Reports, https://github.com/wandb/wandb/issues
7
+ Project-URL: Documentation, https://docs.wandb.ai/
5
8
  Author-email: Weights & Biases <support@wandb.com>
6
9
  License: MIT License
7
10
 
@@ -24,104 +27,98 @@ License: MIT License
24
27
  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
25
28
  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
26
29
  SOFTWARE.
27
-
28
- Project-URL: Source, https://github.com/wandb/wandb
29
- Project-URL: Bug Reports, https://github.com/wandb/wandb/issues
30
- Project-URL: Documentation, https://docs.wandb.ai/
30
+ License-File: LICENSE
31
31
  Classifier: Development Status :: 5 - Production/Stable
32
32
  Classifier: Intended Audience :: Developers
33
33
  Classifier: Intended Audience :: Science/Research
34
34
  Classifier: License :: OSI Approved :: MIT License
35
35
  Classifier: Natural Language :: English
36
36
  Classifier: Programming Language :: Python :: 3
37
+ Classifier: Programming Language :: Python :: 3 :: Only
37
38
  Classifier: Programming Language :: Python :: 3.7
38
39
  Classifier: Programming Language :: Python :: 3.8
39
40
  Classifier: Programming Language :: Python :: 3.9
40
41
  Classifier: Programming Language :: Python :: 3.10
41
42
  Classifier: Programming Language :: Python :: 3.11
42
43
  Classifier: Programming Language :: Python :: 3.12
43
- Classifier: Programming Language :: Python :: 3 :: Only
44
44
  Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
45
45
  Classifier: Topic :: Software Development :: Libraries :: Python Modules
46
46
  Classifier: Topic :: System :: Logging
47
47
  Classifier: Topic :: System :: Monitoring
48
48
  Requires-Python: >=3.7
49
- Description-Content-Type: text/markdown
50
- License-File: LICENSE
51
- Requires-Dist: Click !=8.0.0,>=7.1
52
- Requires-Dist: GitPython !=3.1.29,>=1.0.0
53
- Requires-Dist: requests <3,>=2.0.0
54
- Requires-Dist: psutil >=5.0.0
55
- Requires-Dist: sentry-sdk >=1.0.0
56
- Requires-Dist: docker-pycreds >=0.4.0
57
- Requires-Dist: PyYAML
49
+ Requires-Dist: click!=8.0.0,>=7.1
50
+ Requires-Dist: docker-pycreds>=0.4.0
51
+ Requires-Dist: gitpython!=3.1.29,>=1.0.0
52
+ Requires-Dist: platformdirs
53
+ Requires-Dist: protobuf!=4.21.0,<5,>=3.12.0; python_version < '3.9' and sys_platform == 'linux'
54
+ Requires-Dist: protobuf!=4.21.0,<5,>=3.15.0; python_version == '3.9' and sys_platform == 'linux'
55
+ Requires-Dist: protobuf!=4.21.0,<5,>=3.19.0; python_version > '3.9' and sys_platform == 'linux'
56
+ Requires-Dist: protobuf!=4.21.0,<5,>=3.19.0; sys_platform != 'linux'
57
+ Requires-Dist: psutil>=5.0.0
58
+ Requires-Dist: pyyaml
59
+ Requires-Dist: requests<3,>=2.0.0
60
+ Requires-Dist: sentry-sdk>=1.0.0
58
61
  Requires-Dist: setproctitle
59
62
  Requires-Dist: setuptools
60
- Requires-Dist: appdirs >=1.4.3
61
- Requires-Dist: typing-extensions ; python_version < "3.10"
62
- Requires-Dist: protobuf !=4.21.0,<5,>=3.12.0 ; python_version < "3.9" and sys_platform == "linux"
63
- Requires-Dist: protobuf !=4.21.0,<5,>=3.15.0 ; python_version == "3.9" and sys_platform == "linux"
64
- Requires-Dist: protobuf !=4.21.0,<5,>=3.19.0 ; python_version > "3.9" and sys_platform == "linux"
65
- Requires-Dist: protobuf !=4.21.0,<5,>=3.19.0 ; sys_platform != "linux"
66
- Provides-Extra: async
67
- Requires-Dist: httpx >=0.23.0 ; extra == 'async'
63
+ Requires-Dist: typing-extensions; python_version < '3.10'
68
64
  Provides-Extra: aws
69
- Requires-Dist: boto3 ; extra == 'aws'
65
+ Requires-Dist: boto3; extra == 'aws'
70
66
  Provides-Extra: azure
71
- Requires-Dist: azure-identity ; extra == 'azure'
72
- Requires-Dist: azure-storage-blob ; extra == 'azure'
67
+ Requires-Dist: azure-identity; extra == 'azure'
68
+ Requires-Dist: azure-storage-blob; extra == 'azure'
73
69
  Provides-Extra: gcp
74
- Requires-Dist: google-cloud-storage ; extra == 'gcp'
70
+ Requires-Dist: google-cloud-storage; extra == 'gcp'
75
71
  Provides-Extra: importers
76
- Requires-Dist: polars ; extra == 'importers'
77
- Requires-Dist: rich ; extra == 'importers'
78
- Requires-Dist: filelock ; extra == 'importers'
79
- Requires-Dist: mlflow ; extra == 'importers'
80
- Requires-Dist: tenacity ; extra == 'importers'
72
+ Requires-Dist: filelock; extra == 'importers'
73
+ Requires-Dist: mlflow; extra == 'importers'
74
+ Requires-Dist: polars; extra == 'importers'
75
+ Requires-Dist: rich; extra == 'importers'
76
+ Requires-Dist: tenacity; extra == 'importers'
81
77
  Provides-Extra: kubeflow
82
- Requires-Dist: kubernetes ; extra == 'kubeflow'
83
- Requires-Dist: minio ; extra == 'kubeflow'
84
- Requires-Dist: google-cloud-storage ; extra == 'kubeflow'
85
- Requires-Dist: sh ; extra == 'kubeflow'
78
+ Requires-Dist: google-cloud-storage; extra == 'kubeflow'
79
+ Requires-Dist: kubernetes; extra == 'kubeflow'
80
+ Requires-Dist: minio; extra == 'kubeflow'
81
+ Requires-Dist: sh; extra == 'kubeflow'
86
82
  Provides-Extra: launch
87
- Requires-Dist: awscli ; extra == 'launch'
88
- Requires-Dist: azure-identity ; extra == 'launch'
89
- Requires-Dist: azure-containerregistry ; extra == 'launch'
90
- Requires-Dist: azure-storage-blob ; extra == 'launch'
91
- Requires-Dist: boto3 ; extra == 'launch'
92
- Requires-Dist: botocore ; extra == 'launch'
93
- Requires-Dist: chardet ; extra == 'launch'
94
- Requires-Dist: google-auth ; extra == 'launch'
95
- Requires-Dist: google-cloud-aiplatform ; extra == 'launch'
96
- Requires-Dist: google-cloud-artifact-registry ; extra == 'launch'
97
- Requires-Dist: google-cloud-compute ; extra == 'launch'
98
- Requires-Dist: google-cloud-storage ; extra == 'launch'
99
- Requires-Dist: iso8601 ; extra == 'launch'
100
- Requires-Dist: kubernetes ; extra == 'launch'
101
- Requires-Dist: kubernetes-asyncio ; extra == 'launch'
102
- Requires-Dist: optuna ; extra == 'launch'
103
- Requires-Dist: nbconvert ; extra == 'launch'
104
- Requires-Dist: nbformat ; extra == 'launch'
105
- Requires-Dist: pydantic ; extra == 'launch'
106
- Requires-Dist: typing-extensions ; extra == 'launch'
107
- Requires-Dist: tomli ; extra == 'launch'
108
- Requires-Dist: PyYAML >=6.0.0 ; extra == 'launch'
83
+ Requires-Dist: awscli; extra == 'launch'
84
+ Requires-Dist: azure-containerregistry; extra == 'launch'
85
+ Requires-Dist: azure-identity; extra == 'launch'
86
+ Requires-Dist: azure-storage-blob; extra == 'launch'
87
+ Requires-Dist: boto3; extra == 'launch'
88
+ Requires-Dist: botocore; extra == 'launch'
89
+ Requires-Dist: chardet; extra == 'launch'
90
+ Requires-Dist: google-auth; extra == 'launch'
91
+ Requires-Dist: google-cloud-aiplatform; extra == 'launch'
92
+ Requires-Dist: google-cloud-artifact-registry; extra == 'launch'
93
+ Requires-Dist: google-cloud-compute; extra == 'launch'
94
+ Requires-Dist: google-cloud-storage; extra == 'launch'
95
+ Requires-Dist: iso8601; extra == 'launch'
96
+ Requires-Dist: kubernetes; extra == 'launch'
97
+ Requires-Dist: kubernetes-asyncio; extra == 'launch'
98
+ Requires-Dist: nbconvert; extra == 'launch'
99
+ Requires-Dist: nbformat; extra == 'launch'
100
+ Requires-Dist: optuna; extra == 'launch'
101
+ Requires-Dist: pydantic; extra == 'launch'
102
+ Requires-Dist: pyyaml>=6.0.0; extra == 'launch'
103
+ Requires-Dist: tomli; extra == 'launch'
104
+ Requires-Dist: typing-extensions; extra == 'launch'
109
105
  Provides-Extra: media
110
- Requires-Dist: numpy ; extra == 'media'
111
- Requires-Dist: moviepy ; extra == 'media'
112
- Requires-Dist: pillow ; extra == 'media'
113
- Requires-Dist: bokeh ; extra == 'media'
114
- Requires-Dist: soundfile ; extra == 'media'
115
- Requires-Dist: plotly >=5.18.0 ; extra == 'media'
116
- Requires-Dist: rdkit-pypi ; extra == 'media'
106
+ Requires-Dist: bokeh; extra == 'media'
107
+ Requires-Dist: moviepy; extra == 'media'
108
+ Requires-Dist: numpy; extra == 'media'
109
+ Requires-Dist: pillow; extra == 'media'
110
+ Requires-Dist: plotly>=5.18.0; extra == 'media'
111
+ Requires-Dist: rdkit-pypi; extra == 'media'
112
+ Requires-Dist: soundfile; extra == 'media'
117
113
  Provides-Extra: models
118
- Requires-Dist: cloudpickle ; extra == 'models'
114
+ Requires-Dist: cloudpickle; extra == 'models'
119
115
  Provides-Extra: perf
120
- Requires-Dist: orjson ; extra == 'perf'
116
+ Requires-Dist: orjson; extra == 'perf'
121
117
  Provides-Extra: reports
122
- Requires-Dist: pydantic >=2.0.0 ; extra == 'reports'
118
+ Requires-Dist: pydantic>=2.0.0; extra == 'reports'
123
119
  Provides-Extra: sweeps
124
- Requires-Dist: sweeps >=0.2.0 ; extra == 'sweeps'
120
+ Requires-Dist: sweeps>=0.2.0; extra == 'sweeps'
121
+ Description-Content-Type: text/markdown
125
122
 
126
123
  <div align="center">
127
124
  <img src="https://i.imgur.com/RUtiVzH.png" width="600" /><br><br>