wandb 0.17.7__py3-none-macosx_11_0_arm64.whl → 0.17.8__py3-none-macosx_11_0_arm64.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (41) hide show
  1. package_readme.md +47 -53
  2. wandb/__init__.py +8 -4
  3. wandb/__init__.pyi +964 -0
  4. wandb/bin/apple_gpu_stats +0 -0
  5. wandb/bin/wandb-core +0 -0
  6. wandb/data_types.py +1 -0
  7. wandb/env.py +13 -0
  8. wandb/integration/keras/__init__.py +2 -5
  9. wandb/integration/keras/callbacks/metrics_logger.py +10 -4
  10. wandb/integration/keras/callbacks/model_checkpoint.py +0 -5
  11. wandb/integration/keras/keras.py +11 -0
  12. wandb/proto/v3/wandb_internal_pb2.py +24 -24
  13. wandb/proto/v3/wandb_settings_pb2.py +2 -2
  14. wandb/proto/v3/wandb_telemetry_pb2.py +12 -12
  15. wandb/proto/v4/wandb_internal_pb2.py +24 -24
  16. wandb/proto/v4/wandb_settings_pb2.py +2 -2
  17. wandb/proto/v4/wandb_telemetry_pb2.py +12 -12
  18. wandb/proto/v5/wandb_internal_pb2.py +24 -24
  19. wandb/proto/v5/wandb_settings_pb2.py +2 -2
  20. wandb/proto/v5/wandb_telemetry_pb2.py +12 -12
  21. wandb/proto/wandb_deprecated.py +2 -0
  22. wandb/sdk/artifacts/artifact.py +21 -25
  23. wandb/sdk/artifacts/artifact_manifest_entry.py +10 -2
  24. wandb/sdk/artifacts/storage_handlers/gcs_handler.py +31 -0
  25. wandb/sdk/data_types/video.py +2 -2
  26. wandb/sdk/internal/handler.py +5 -1
  27. wandb/sdk/lib/_settings_toposort_generated.py +1 -0
  28. wandb/sdk/service/service.py +7 -2
  29. wandb/sdk/wandb_config.py +3 -0
  30. wandb/sdk/wandb_init.py +4 -1
  31. wandb/sdk/wandb_manager.py +0 -3
  32. wandb/sdk/wandb_require.py +22 -1
  33. wandb/sdk/wandb_run.py +28 -24
  34. wandb/sdk/wandb_settings.py +25 -1
  35. wandb/sdk/wandb_setup.py +3 -0
  36. wandb/testing/relay.py +7 -1
  37. {wandb-0.17.7.dist-info → wandb-0.17.8.dist-info}/METADATA +48 -54
  38. {wandb-0.17.7.dist-info → wandb-0.17.8.dist-info}/RECORD +41 -40
  39. {wandb-0.17.7.dist-info → wandb-0.17.8.dist-info}/WHEEL +0 -0
  40. {wandb-0.17.7.dist-info → wandb-0.17.8.dist-info}/entry_points.txt +0 -0
  41. {wandb-0.17.7.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 [![PyPI](https://img.shields.io/pypi/v/wandb)](https://pypi.python.org/pypi/wandb) [![Conda (channel only)](https://img.shields.io/conda/vn/conda-forge/wandb)](https://anaconda.org/conda-forge/wandb) [![CircleCI](https://img.shields.io/circleci/build/github/wandb/wandb/main)](https://circleci.com/gh/wandb/wandb) [![Codecov](https://img.shields.io/codecov/c/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
- ## Features
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
- - Store hyper-parameters used in a training run
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
- [Documentation →](https://docs.wandb.com)
15
+ # Quickstart
27
16
 
28
- ## Quickstart
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
- In your training script:
27
+ 3. Third, log into W&B:
35
28
 
36
29
  ```python
37
- import wandb
30
+ wandb.login()
31
+ ```
38
32
 
39
- # Your custom arguments defined here
40
- args = ...
33
+ 4. Use the example code snippet below as a template to integrate W&B to your Python script:
41
34
 
42
- wandb.init(config=args, project="my-project")
43
- wandb.config["more"] = "custom"
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
- def training_loop():
47
- while True:
48
- # Do some machine learning
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
- If you're already using Tensorboard or [TensorboardX](https://github.com/lanpa/tensorboardX), you can integrate with one line:
45
+ # Model training code here ...
55
46
 
56
- ```python
57
- wandb.init(sync_tensorboard=True)
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
- ## Running your script
55
+ For example, if the preceding code was stored in a script called train.py:
61
56
 
62
- Run `wandb login` from your terminal to signup or authenticate your machine (we store your api key in ~/.netrc). You can also set the `WANDB_API_KEY` environment variable with a key from your [settings](https://app.wandb.ai/settings).
57
+ ```shell
58
+ python train.py
59
+ ```
63
60
 
64
- Run your script with `python my_script.py` and all metadata will be synced to the cloud. 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. If you want to test your script without syncing to the cloud you can set the environment variable `WANDB_MODE=dryrun`.
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
- If you are using [docker](https://docker.com) to run your code, we provide a wrapper command `wandb docker` that mounts your current directory, sets environment variables, and ensures the wandb library is installed. Training your models in docker gives you the ability to restore the exact code and environment with the `wandb restore` command.
63
+  
67
64
 
68
- ## Web Interface
65
+ # Integrations
69
66
 
70
- [Sign up for a free account ](https://wandb.com)
71
- [![Watch the video](https://i.imgur.com/PW0Ejlc.png)](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
- ## Detailed Usage
69
+  
75
70
 
76
- Framework specific and detailed usage can be found in our [documentation](http://docs.wandb.com/).
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
- ## Testing
74
+  
79
75
 
80
- To run basic test use `make test`. More detailed information can be found at CONTRIBUTING.md.
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
- We use [circleci](https://circleci.com) for CI.
79
+  
83
80
 
84
- # Academic Researchers
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
- We make it easy to cite W&B in your published paper. [Learn more ](https://www.wandb.com/academic)
88
- [![](https://i.imgur.com/loKLiez.png)](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
- ## Community
91
- Got questions, feedback or want to join a community of ML engineers working on exciting projects?
85
+  
92
86
 
93
- <a href="https://bit.ly/wb-slack"><img src="https://svgshare.com/i/M93.svg" alt="slack" width="55"/></a> Join our [slack](https://bit.ly/wb-slack) community.
87
+ # License
94
88
 
95
- [![Twitter](https://img.shields.io/twitter/follow/weights_biases?style=social)](https://twitter.com/weights_biases) Follow us on [Twitter](https://twitter.com/weights_biases).
89
+ [MIT License](https://github.com/wandb/wandb/blob/main/LICENSE)
wandb/__init__.py CHANGED
@@ -8,7 +8,7 @@ For scripts and interactive notebooks, see https://github.com/wandb/examples.
8
8
 
9
9
  For reference documentation, see https://docs.wandb.com/ref/python.
10
10
  """
11
- __version__ = "0.17.7"
11
+ __version__ = "0.17.8"
12
12
 
13
13
  from typing import Optional
14
14
 
@@ -199,14 +199,18 @@ if wandb_sdk.lib.ipython.in_notebook():
199
199
  from .analytics import Sentry as _Sentry
200
200
 
201
201
  if "dev" in __version__:
202
+ import wandb.env
202
203
  import os
203
204
 
204
205
  # disable error reporting in dev versions for the python client
205
- os.environ["WANDB_ERROR_REPORTING"] = os.environ.get(
206
- "WANDB_ERROR_REPORTING", "false"
206
+ os.environ[wandb.env.ERROR_REPORTING] = os.environ.get(
207
+ wandb.env.ERROR_REPORTING, "false"
207
208
  )
209
+
208
210
  # turn on wandb-core for dev versions
209
- os.environ["WANDB__REQUIRE_CORE"] = os.environ.get("WANDB__REQUIRE_CORE", "true")
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")
210
214
 
211
215
  _sentry = _Sentry()
212
216
  _sentry.setup()