wandb 0.16.6__py3-none-any.whl → 0.17.0rc2__py3-none-any.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (151) hide show
  1. package_readme.md +95 -0
  2. wandb/__init__.py +2 -2
  3. wandb/agents/pyagent.py +0 -1
  4. wandb/analytics/sentry.py +2 -1
  5. wandb/apis/importers/internals/protocols.py +30 -56
  6. wandb/apis/importers/mlflow.py +13 -26
  7. wandb/apis/importers/wandb.py +8 -14
  8. wandb/apis/public/api.py +1 -0
  9. wandb/apis/public/artifacts.py +1 -0
  10. wandb/apis/public/files.py +1 -0
  11. wandb/apis/public/history.py +1 -0
  12. wandb/apis/public/jobs.py +1 -0
  13. wandb/apis/public/projects.py +1 -0
  14. wandb/apis/public/reports.py +1 -0
  15. wandb/apis/public/runs.py +1 -0
  16. wandb/apis/public/sweeps.py +1 -0
  17. wandb/apis/public/teams.py +1 -0
  18. wandb/apis/public/users.py +1 -0
  19. wandb/apis/reports/v1/_blocks.py +3 -7
  20. wandb/apis/reports/v2/gql.py +1 -0
  21. wandb/apis/reports/v2/interface.py +3 -4
  22. wandb/apis/reports/v2/internal.py +5 -8
  23. wandb/cli/cli.py +2 -2
  24. wandb/data_types.py +9 -6
  25. wandb/docker/__init__.py +1 -1
  26. wandb/env.py +38 -8
  27. wandb/errors/__init__.py +5 -0
  28. wandb/integration/catboost/catboost.py +1 -1
  29. wandb/integration/fastai/__init__.py +1 -0
  30. wandb/integration/huggingface/resolver.py +2 -2
  31. wandb/integration/keras/__init__.py +1 -0
  32. wandb/integration/keras/callbacks/metrics_logger.py +1 -1
  33. wandb/integration/keras/keras.py +7 -7
  34. wandb/integration/langchain/wandb_tracer.py +1 -0
  35. wandb/integration/lightning/fabric/logger.py +1 -3
  36. wandb/integration/metaflow/metaflow.py +41 -6
  37. wandb/integration/openai/fine_tuning.py +3 -3
  38. wandb/keras/__init__.py +1 -0
  39. wandb/old/summary.py +1 -1
  40. wandb/plot/confusion_matrix.py +1 -1
  41. wandb/plots/precision_recall.py +1 -1
  42. wandb/plots/roc.py +1 -1
  43. wandb/proto/v3/wandb_internal_pb2.py +364 -332
  44. wandb/proto/v3/wandb_settings_pb2.py +1 -1
  45. wandb/proto/v4/wandb_internal_pb2.py +322 -316
  46. wandb/proto/v4/wandb_settings_pb2.py +1 -1
  47. wandb/proto/wandb_internal_codegen.py +0 -25
  48. wandb/sdk/artifacts/artifact.py +16 -4
  49. wandb/sdk/artifacts/artifact_download_logger.py +1 -0
  50. wandb/sdk/artifacts/artifact_file_cache.py +18 -4
  51. wandb/sdk/artifacts/artifact_instance_cache.py +1 -0
  52. wandb/sdk/artifacts/artifact_manifest.py +1 -0
  53. wandb/sdk/artifacts/artifact_manifest_entry.py +1 -0
  54. wandb/sdk/artifacts/artifact_manifests/artifact_manifest_v1.py +1 -0
  55. wandb/sdk/artifacts/artifact_saver.py +5 -2
  56. wandb/sdk/artifacts/artifact_state.py +1 -0
  57. wandb/sdk/artifacts/artifact_ttl.py +1 -0
  58. wandb/sdk/artifacts/exceptions.py +1 -0
  59. wandb/sdk/artifacts/storage_handlers/azure_handler.py +1 -0
  60. wandb/sdk/artifacts/storage_handlers/gcs_handler.py +13 -18
  61. wandb/sdk/artifacts/storage_handlers/http_handler.py +1 -0
  62. wandb/sdk/artifacts/storage_handlers/local_file_handler.py +1 -0
  63. wandb/sdk/artifacts/storage_handlers/multi_handler.py +1 -0
  64. wandb/sdk/artifacts/storage_handlers/s3_handler.py +5 -3
  65. wandb/sdk/artifacts/storage_handlers/tracking_handler.py +1 -0
  66. wandb/sdk/artifacts/storage_handlers/wb_artifact_handler.py +1 -0
  67. wandb/sdk/artifacts/storage_handlers/wb_local_artifact_handler.py +1 -0
  68. wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py +1 -0
  69. wandb/sdk/artifacts/storage_policy.py +1 -0
  70. wandb/sdk/data_types/_dtypes.py +8 -8
  71. wandb/sdk/data_types/base_types/media.py +3 -6
  72. wandb/sdk/data_types/helper_types/bounding_boxes_2d.py +3 -1
  73. wandb/sdk/data_types/image.py +1 -1
  74. wandb/sdk/data_types/video.py +1 -1
  75. wandb/sdk/integration_utils/auto_logging.py +5 -6
  76. wandb/sdk/integration_utils/data_logging.py +10 -6
  77. wandb/sdk/interface/interface.py +55 -32
  78. wandb/sdk/interface/interface_shared.py +7 -13
  79. wandb/sdk/internal/datastore.py +1 -1
  80. wandb/sdk/internal/handler.py +18 -2
  81. wandb/sdk/internal/internal.py +0 -1
  82. wandb/sdk/internal/internal_util.py +0 -1
  83. wandb/sdk/internal/job_builder.py +5 -4
  84. wandb/sdk/internal/profiler.py +1 -0
  85. wandb/sdk/internal/run.py +1 -0
  86. wandb/sdk/internal/sender.py +1 -1
  87. wandb/sdk/internal/system/assets/gpu_amd.py +44 -44
  88. wandb/sdk/internal/system/assets/gpu_apple.py +56 -11
  89. wandb/sdk/internal/system/assets/interfaces.py +6 -8
  90. wandb/sdk/internal/system/assets/open_metrics.py +2 -2
  91. wandb/sdk/internal/system/assets/trainium.py +1 -3
  92. wandb/sdk/launch/_project_spec.py +8 -4
  93. wandb/sdk/launch/agent/agent.py +2 -1
  94. wandb/sdk/launch/agent/config.py +72 -11
  95. wandb/sdk/launch/builder/abstract.py +2 -1
  96. wandb/sdk/launch/builder/build.py +29 -2
  97. wandb/sdk/launch/builder/docker_builder.py +1 -0
  98. wandb/sdk/launch/builder/kaniko_builder.py +2 -2
  99. wandb/sdk/launch/builder/noop.py +1 -0
  100. wandb/sdk/launch/create_job.py +18 -0
  101. wandb/sdk/launch/environment/abstract.py +1 -0
  102. wandb/sdk/launch/environment/gcp_environment.py +1 -0
  103. wandb/sdk/launch/environment/local_environment.py +1 -0
  104. wandb/sdk/launch/loader.py +1 -0
  105. wandb/sdk/launch/registry/abstract.py +1 -0
  106. wandb/sdk/launch/registry/azure_container_registry.py +1 -0
  107. wandb/sdk/launch/registry/elastic_container_registry.py +1 -0
  108. wandb/sdk/launch/registry/google_artifact_registry.py +2 -1
  109. wandb/sdk/launch/registry/local_registry.py +1 -0
  110. wandb/sdk/launch/runner/abstract.py +1 -0
  111. wandb/sdk/launch/runner/kubernetes_monitor.py +1 -0
  112. wandb/sdk/launch/runner/kubernetes_runner.py +4 -3
  113. wandb/sdk/launch/runner/sagemaker_runner.py +11 -10
  114. wandb/sdk/launch/sweeps/scheduler.py +4 -3
  115. wandb/sdk/launch/sweeps/scheduler_sweep.py +2 -1
  116. wandb/sdk/launch/sweeps/utils.py +3 -3
  117. wandb/sdk/launch/utils.py +3 -3
  118. wandb/sdk/lib/fsm.py +8 -12
  119. wandb/sdk/lib/gitlib.py +4 -4
  120. wandb/sdk/lib/import_hooks.py +1 -1
  121. wandb/sdk/lib/lazyloader.py +0 -1
  122. wandb/sdk/lib/proto_util.py +1 -1
  123. wandb/sdk/lib/redirect.py +19 -14
  124. wandb/sdk/lib/retry.py +3 -2
  125. wandb/sdk/lib/tracelog.py +1 -1
  126. wandb/sdk/service/service.py +17 -15
  127. wandb/sdk/verify/verify.py +2 -1
  128. wandb/sdk/wandb_manager.py +2 -2
  129. wandb/sdk/wandb_require.py +5 -0
  130. wandb/sdk/wandb_run.py +25 -20
  131. wandb/sdk/wandb_settings.py +0 -1
  132. wandb/sdk/wandb_setup.py +1 -1
  133. wandb/sklearn/__init__.py +1 -0
  134. wandb/sklearn/plot/__init__.py +1 -0
  135. wandb/sklearn/plot/classifier.py +7 -6
  136. wandb/sklearn/plot/clusterer.py +2 -1
  137. wandb/sklearn/plot/regressor.py +1 -0
  138. wandb/sklearn/plot/shared.py +1 -0
  139. wandb/sklearn/utils.py +1 -0
  140. wandb/testing/relay.py +4 -4
  141. wandb/trigger.py +1 -0
  142. wandb/util.py +40 -17
  143. wandb/wandb_controller.py +2 -3
  144. wandb/wandb_torch.py +1 -2
  145. {wandb-0.16.6.dist-info → wandb-0.17.0rc2.dist-info}/METADATA +68 -69
  146. {wandb-0.16.6.dist-info → wandb-0.17.0rc2.dist-info}/RECORD +149 -150
  147. {wandb-0.16.6.dist-info → wandb-0.17.0rc2.dist-info}/WHEEL +1 -2
  148. wandb/bin/apple_gpu_stats +0 -0
  149. wandb-0.16.6.dist-info/top_level.txt +0 -1
  150. {wandb-0.16.6.dist-info → wandb-0.17.0rc2.dist-info}/entry_points.txt +0 -0
  151. {wandb-0.16.6.dist-info → wandb-0.17.0rc2.dist-info/licenses}/LICENSE +0 -0
{wandb-0.16.6.dist-info → wandb-0.17.0rc2.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.6
3
+ Version: 0.17.0rc2
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,100 @@ 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"
63
+ Requires-Dist: typing-extensions; python_version < '3.10'
66
64
  Provides-Extra: async
67
- Requires-Dist: httpx >=0.23.0 ; extra == 'async'
65
+ Requires-Dist: httpx>=0.23.0; extra == 'async'
68
66
  Provides-Extra: aws
69
- Requires-Dist: boto3 ; extra == 'aws'
67
+ Requires-Dist: boto3; extra == 'aws'
70
68
  Provides-Extra: azure
71
- Requires-Dist: azure-identity ; extra == 'azure'
72
- Requires-Dist: azure-storage-blob ; extra == 'azure'
69
+ Requires-Dist: azure-identity; extra == 'azure'
70
+ Requires-Dist: azure-storage-blob; extra == 'azure'
73
71
  Provides-Extra: gcp
74
- Requires-Dist: google-cloud-storage ; extra == 'gcp'
72
+ Requires-Dist: google-cloud-storage; extra == 'gcp'
75
73
  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'
74
+ Requires-Dist: filelock; extra == 'importers'
75
+ Requires-Dist: mlflow; extra == 'importers'
76
+ Requires-Dist: polars; extra == 'importers'
77
+ Requires-Dist: rich; extra == 'importers'
78
+ Requires-Dist: tenacity; extra == 'importers'
81
79
  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'
80
+ Requires-Dist: google-cloud-storage; extra == 'kubeflow'
81
+ Requires-Dist: kubernetes; extra == 'kubeflow'
82
+ Requires-Dist: minio; extra == 'kubeflow'
83
+ Requires-Dist: sh; extra == 'kubeflow'
86
84
  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'
85
+ Requires-Dist: awscli; extra == 'launch'
86
+ Requires-Dist: azure-containerregistry; extra == 'launch'
87
+ Requires-Dist: azure-identity; extra == 'launch'
88
+ Requires-Dist: azure-storage-blob; extra == 'launch'
89
+ Requires-Dist: boto3; extra == 'launch'
90
+ Requires-Dist: botocore; extra == 'launch'
91
+ Requires-Dist: chardet; extra == 'launch'
92
+ Requires-Dist: google-auth; extra == 'launch'
93
+ Requires-Dist: google-cloud-aiplatform; extra == 'launch'
94
+ Requires-Dist: google-cloud-artifact-registry; extra == 'launch'
95
+ Requires-Dist: google-cloud-compute; extra == 'launch'
96
+ Requires-Dist: google-cloud-storage; extra == 'launch'
97
+ Requires-Dist: iso8601; extra == 'launch'
98
+ Requires-Dist: kubernetes; extra == 'launch'
99
+ Requires-Dist: kubernetes-asyncio; extra == 'launch'
100
+ Requires-Dist: nbconvert; extra == 'launch'
101
+ Requires-Dist: nbformat; extra == 'launch'
102
+ Requires-Dist: optuna; extra == 'launch'
103
+ Requires-Dist: pydantic; extra == 'launch'
104
+ Requires-Dist: pyyaml>=6.0.0; extra == 'launch'
105
+ Requires-Dist: tomli; extra == 'launch'
106
+ Requires-Dist: typing-extensions; extra == 'launch'
109
107
  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'
108
+ Requires-Dist: bokeh; extra == 'media'
109
+ Requires-Dist: moviepy; extra == 'media'
110
+ Requires-Dist: numpy; extra == 'media'
111
+ Requires-Dist: pillow; extra == 'media'
112
+ Requires-Dist: plotly>=5.18.0; extra == 'media'
113
+ Requires-Dist: rdkit-pypi; extra == 'media'
114
+ Requires-Dist: soundfile; extra == 'media'
117
115
  Provides-Extra: models
118
- Requires-Dist: cloudpickle ; extra == 'models'
116
+ Requires-Dist: cloudpickle; extra == 'models'
119
117
  Provides-Extra: perf
120
- Requires-Dist: orjson ; extra == 'perf'
118
+ Requires-Dist: orjson; extra == 'perf'
121
119
  Provides-Extra: reports
122
- Requires-Dist: pydantic >=2.0.0 ; extra == 'reports'
120
+ Requires-Dist: pydantic>=2.0.0; extra == 'reports'
123
121
  Provides-Extra: sweeps
124
- Requires-Dist: sweeps >=0.2.0 ; extra == 'sweeps'
122
+ Requires-Dist: sweeps>=0.2.0; extra == 'sweeps'
123
+ Description-Content-Type: text/markdown
125
124
 
126
125
  <div align="center">
127
126
  <img src="https://i.imgur.com/RUtiVzH.png" width="600" /><br><br>