wandb 0.17.6__py3-none-win32.whl → 0.17.8__py3-none-win32.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.
- package_readme.md +47 -53
- wandb/__init__.py +10 -19
- wandb/__init__.pyi +964 -0
- wandb/agents/pyagent.py +1 -2
- wandb/bin/wandb-core +0 -0
- wandb/cli/cli.py +21 -0
- wandb/data_types.py +4 -3
- wandb/env.py +13 -0
- wandb/integration/keras/__init__.py +2 -5
- wandb/integration/keras/callbacks/metrics_logger.py +10 -4
- wandb/integration/keras/callbacks/model_checkpoint.py +0 -5
- wandb/integration/keras/keras.py +11 -0
- wandb/integration/kfp/wandb_logging.py +1 -1
- wandb/integration/lightning/fabric/logger.py +1 -1
- wandb/integration/openai/fine_tuning.py +13 -5
- wandb/integration/ultralytics/pose_utils.py +0 -1
- wandb/proto/v3/wandb_internal_pb2.py +24 -24
- wandb/proto/v3/wandb_settings_pb2.py +2 -2
- wandb/proto/v3/wandb_telemetry_pb2.py +12 -12
- wandb/proto/v4/wandb_internal_pb2.py +24 -24
- wandb/proto/v4/wandb_settings_pb2.py +2 -2
- wandb/proto/v4/wandb_telemetry_pb2.py +12 -12
- wandb/proto/v5/wandb_internal_pb2.py +24 -24
- wandb/proto/v5/wandb_settings_pb2.py +2 -2
- wandb/proto/v5/wandb_telemetry_pb2.py +12 -12
- wandb/proto/wandb_deprecated.py +2 -0
- wandb/sdk/artifacts/artifact.py +22 -26
- wandb/sdk/artifacts/artifact_manifest_entry.py +10 -2
- wandb/sdk/artifacts/storage_handlers/gcs_handler.py +31 -0
- wandb/sdk/data_types/_dtypes.py +5 -5
- wandb/sdk/data_types/base_types/media.py +3 -1
- wandb/sdk/data_types/helper_types/bounding_boxes_2d.py +3 -1
- wandb/sdk/data_types/helper_types/image_mask.py +3 -1
- wandb/sdk/data_types/image.py +3 -1
- wandb/sdk/data_types/saved_model.py +3 -1
- wandb/sdk/data_types/video.py +2 -2
- wandb/sdk/interface/interface.py +17 -16
- wandb/sdk/interface/interface_shared.py +6 -9
- wandb/sdk/internal/datastore.py +1 -1
- wandb/sdk/internal/handler.py +5 -3
- wandb/sdk/internal/internal.py +1 -1
- wandb/sdk/internal/job_builder.py +5 -2
- wandb/sdk/internal/tb_watcher.py +2 -2
- wandb/sdk/internal/update.py +2 -2
- wandb/sdk/launch/builder/kaniko_builder.py +13 -5
- wandb/sdk/launch/create_job.py +2 -0
- wandb/sdk/lib/_settings_toposort_generated.py +1 -0
- wandb/sdk/lib/apikey.py +1 -1
- wandb/sdk/service/service.py +7 -2
- wandb/sdk/service/streams.py +2 -4
- wandb/sdk/wandb_config.py +4 -1
- wandb/sdk/wandb_init.py +59 -8
- wandb/sdk/wandb_manager.py +0 -3
- wandb/sdk/wandb_require.py +22 -1
- wandb/sdk/wandb_run.py +137 -92
- wandb/sdk/wandb_settings.py +26 -2
- wandb/sdk/wandb_setup.py +69 -3
- wandb/sdk/wandb_sweep.py +5 -2
- wandb/testing/relay.py +7 -1
- {wandb-0.17.6.dist-info → wandb-0.17.8.dist-info}/METADATA +48 -54
- {wandb-0.17.6.dist-info → wandb-0.17.8.dist-info}/RECORD +64 -63
- {wandb-0.17.6.dist-info → wandb-0.17.8.dist-info}/WHEEL +0 -0
- {wandb-0.17.6.dist-info → wandb-0.17.8.dist-info}/entry_points.txt +0 -0
- {wandb-0.17.6.dist-info → wandb-0.17.8.dist-info}/licenses/LICENSE +0 -0
package_readme.md
CHANGED
@@ -4,92 +4,86 @@
|
|
4
4
|
|
5
5
|
# Weights and Biases [](https://pypi.python.org/pypi/wandb) [](https://anaconda.org/conda-forge/wandb) [](https://circleci.com/gh/wandb/wandb) [](https://codecov.io/gh/wandb/wandb)
|
6
6
|
|
7
|
-
Use W&B to build better models faster. Track and visualize all the pieces of your machine learning pipeline, from datasets to production models.
|
7
|
+
Use W&B to build better models faster. Track and visualize all the pieces of your machine learning pipeline, from datasets to production machine learning models. Get started with W&B today, [sign up for an account!](https://wandb.com?utm_source=github&utm_medium=code&utm_campaign=wandb&utm_content=readme)
|
8
8
|
|
9
|
-
- Quickly identify model regressions. Use W&B to visualize results in real time, all in a central dashboard.
|
10
|
-
- Focus on the interesting ML. Spend less time manually tracking results in spreadsheets and text files.
|
11
|
-
- Capture dataset versions with W&B Artifacts to identify how changing data affects your resulting models.
|
12
|
-
- Reproduce any model, with saved code, hyperparameters, launch commands, input data, and resulting model weights.
|
13
9
|
|
14
|
-
[Sign up for a free account →](https://wandb.ai/login?signup=true)
|
15
10
|
|
16
|
-
|
11
|
+
See the [W&B Developer Guide](https://docs.wandb.ai/?utm_source=github&utm_medium=code&utm_campaign=wandb&utm_content=documentation) and [API Reference Guide](https://docs.wandb.ai/ref?utm_source=github&utm_medium=code&utm_campaign=wandb&utm_content=documentation) for a full technical description of the W&B platform.
|
17
12
|
|
18
|
-
|
19
|
-
- Search, compare, and visualize training runs
|
20
|
-
- Analyze system usage metrics alongside runs
|
21
|
-
- Collaborate with team members
|
22
|
-
- Replicate historic results
|
23
|
-
- Run parameter sweeps
|
24
|
-
- Keep records of experiments available forever
|
13
|
+
|
25
14
|
|
26
|
-
|
15
|
+
# Quickstart
|
27
16
|
|
28
|
-
|
17
|
+
Get started with W&B in four steps:
|
18
|
+
|
19
|
+
1. First, sign up for a [W&B account](https://wandb.ai/login?utm_source=github&utm_medium=code&utm_campaign=wandb&utm_content=quickstart).
|
20
|
+
|
21
|
+
2. Second, install the W&B SDK with [pip](https://pip.pypa.io/en/stable/). Navigate to your terminal and type the following command:
|
29
22
|
|
30
23
|
```shell
|
31
24
|
pip install wandb
|
32
25
|
```
|
33
26
|
|
34
|
-
|
27
|
+
3. Third, log into W&B:
|
35
28
|
|
36
29
|
```python
|
37
|
-
|
30
|
+
wandb.login()
|
31
|
+
```
|
38
32
|
|
39
|
-
|
40
|
-
args = ...
|
33
|
+
4. Use the example code snippet below as a template to integrate W&B to your Python script:
|
41
34
|
|
42
|
-
|
43
|
-
wandb
|
35
|
+
```python
|
36
|
+
import wandb
|
44
37
|
|
38
|
+
# Start a W&B Run with wandb.init
|
39
|
+
run = wandb.init(project="my_first_project")
|
45
40
|
|
46
|
-
|
47
|
-
|
48
|
-
|
49
|
-
epoch, loss, val_loss = ...
|
50
|
-
# Framework agnostic / custom metrics
|
51
|
-
wandb.log({"epoch": epoch, "loss": loss, "val_loss": val_loss})
|
52
|
-
```
|
41
|
+
# Save model inputs and hyperparameters in a wandb.config object
|
42
|
+
config = run.config
|
43
|
+
config.learning_rate = 0.01
|
53
44
|
|
54
|
-
|
45
|
+
# Model training code here ...
|
55
46
|
|
56
|
-
|
57
|
-
|
47
|
+
# Log metrics over time to visualize performance with wandb.log
|
48
|
+
for i in range(10):
|
49
|
+
run.log({"loss": ...})
|
50
|
+
|
51
|
+
# Mark the run as finished, and finish uploading all data
|
52
|
+
run.finish()
|
58
53
|
```
|
59
54
|
|
60
|
-
|
55
|
+
For example, if the preceding code was stored in a script called train.py:
|
61
56
|
|
62
|
-
|
57
|
+
```shell
|
58
|
+
python train.py
|
59
|
+
```
|
63
60
|
|
64
|
-
|
61
|
+
You will see a URL in your terminal logs when your script starts and finishes. Data is staged locally in a directory named _wandb_ relative to your script. Navigate to the W&B App to view a dashboard of your first W&B Experiment. Use the W&B App to compare multiple experiments in a unified place, dive into the results of a single run, and much more!
|
65
62
|
|
66
|
-
|
63
|
+
|
67
64
|
|
68
|
-
|
65
|
+
# Integrations
|
69
66
|
|
70
|
-
|
71
|
-
[](https://youtu.be/EeqhOSvNX-A)
|
72
|
-
[Introduction video →](https://youtu.be/EeqhOSvNX-A)
|
67
|
+
Use your favorite framework with W&B. W&B integrations make it fast and easy to set up experiment tracking and data versioning inside existing projects. For more information on how to integrate W&B with the framework of your choice, see [W&B Integrations](https://docs.wandb.ai/guides/integrations) in the W&B Developer Guide.
|
73
68
|
|
74
|
-
|
69
|
+
|
75
70
|
|
76
|
-
|
71
|
+
# Contribution guidelines
|
72
|
+
Weights & Biases ❤️ open source, and we welcome contributions from the community! See the [Contribution guide](https://github.com/wandb/wandb/blob/main/CONTRIBUTING.md) for more information on the development workflow and the internals of the wandb library. For wandb bugs and feature requests, visit [GitHub Issues](https://github.com/wandb/wandb/issues) or contact support@wandb.com.
|
77
73
|
|
78
|
-
|
74
|
+
|
79
75
|
|
80
|
-
|
76
|
+
# Academic Researchers
|
77
|
+
Reach out to W&B Support at support@wandb.com to get a [free academic license](https://www.wandb.com/academic) for you and your research group.
|
81
78
|
|
82
|
-
|
79
|
+
|
83
80
|
|
84
|
-
#
|
85
|
-
If you'd like a free academic account for your research group, [reach out to us →](https://www.wandb.com/academic)
|
81
|
+
# W&B Community
|
86
82
|
|
87
|
-
|
88
|
-
[](https://www.wandb.com/academic)
|
83
|
+
Be a part of the growing W&B Community and interact with the W&B team in our [Discord](https://wandb.me/discord). Stay connected with the latest ML updates and tutorials with [W&B Fully Connected](https://wandb.ai/fully-connected).
|
89
84
|
|
90
|
-
|
91
|
-
Got questions, feedback or want to join a community of ML engineers working on exciting projects?
|
85
|
+
|
92
86
|
|
93
|
-
|
87
|
+
# License
|
94
88
|
|
95
|
-
[
|
89
|
+
[MIT License](https://github.com/wandb/wandb/blob/main/LICENSE)
|
wandb/__init__.py
CHANGED
@@ -1,9 +1,6 @@
|
|
1
1
|
"""Use wandb to track machine learning work.
|
2
2
|
|
3
|
-
|
4
|
-
- wandb.init — initialize a new run at the top of your training script
|
5
|
-
- wandb.config — track hyperparameters and metadata
|
6
|
-
- wandb.log — log metrics and media over time within your training loop
|
3
|
+
Train and fine-tune models, manage models from experimentation to production.
|
7
4
|
|
8
5
|
For guides and examples, see https://docs.wandb.ai.
|
9
6
|
|
@@ -11,11 +8,7 @@ For scripts and interactive notebooks, see https://github.com/wandb/examples.
|
|
11
8
|
|
12
9
|
For reference documentation, see https://docs.wandb.com/ref/python.
|
13
10
|
"""
|
14
|
-
__version__ = "0.17.
|
15
|
-
|
16
|
-
|
17
|
-
# Used with pypi checks and other messages related to pip
|
18
|
-
_wandb_module = "wandb"
|
11
|
+
__version__ = "0.17.8"
|
19
12
|
|
20
13
|
from typing import Optional
|
21
14
|
|
@@ -118,13 +111,6 @@ def _assert_is_user_process():
|
|
118
111
|
assert not _IS_INTERNAL_PROCESS
|
119
112
|
|
120
113
|
|
121
|
-
# toplevel:
|
122
|
-
# save()
|
123
|
-
# restore()
|
124
|
-
# login()
|
125
|
-
# sweep()
|
126
|
-
# agent()
|
127
|
-
|
128
114
|
# globals
|
129
115
|
Api = PublicApi
|
130
116
|
api = InternalApi()
|
@@ -213,14 +199,18 @@ if wandb_sdk.lib.ipython.in_notebook():
|
|
213
199
|
from .analytics import Sentry as _Sentry
|
214
200
|
|
215
201
|
if "dev" in __version__:
|
202
|
+
import wandb.env
|
216
203
|
import os
|
217
204
|
|
218
205
|
# disable error reporting in dev versions for the python client
|
219
|
-
os.environ[
|
220
|
-
|
206
|
+
os.environ[wandb.env.ERROR_REPORTING] = os.environ.get(
|
207
|
+
wandb.env.ERROR_REPORTING, "false"
|
221
208
|
)
|
209
|
+
|
222
210
|
# turn on wandb-core for dev versions
|
223
|
-
|
211
|
+
if not wandb.env.is_require_legacy_service():
|
212
|
+
os.environ[wandb.env._REQUIRE_CORE] = os.environ.get(
|
213
|
+
wandb.env._REQUIRE_CORE, "true")
|
224
214
|
|
225
215
|
_sentry = _Sentry()
|
226
216
|
_sentry.setup()
|
@@ -254,4 +244,5 @@ __all__ = (
|
|
254
244
|
"log_model",
|
255
245
|
"use_model",
|
256
246
|
"link_model",
|
247
|
+
"define_metric",
|
257
248
|
)
|