wandb 0.17.8rc1__py3-none-macosx_10_14_x86_64.whl → 0.19.7__py3-none-macosx_10_14_x86_64.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 +52 -50
- wandb/__init__.py +19 -17
- wandb/__init__.pyi +530 -276
- wandb/_iterutils.py +46 -0
- wandb/agents/pyagent.py +2 -2
- wandb/analytics/sentry.py +92 -95
- wandb/apis/attrs.py +15 -4
- wandb/apis/importers/internals/internal.py +6 -6
- wandb/apis/importers/internals/protocols.py +11 -7
- wandb/apis/importers/wandb.py +2 -2
- wandb/apis/internal.py +7 -0
- wandb/apis/normalize.py +15 -23
- wandb/apis/public/_generated/__init__.py +21 -0
- wandb/apis/public/_generated/base.py +128 -0
- wandb/apis/public/_generated/enums.py +4 -0
- wandb/apis/public/_generated/input_types.py +4 -0
- wandb/apis/public/_generated/operations.py +15 -0
- wandb/apis/public/_generated/server_features_query.py +27 -0
- wandb/apis/public/_generated/typing_compat.py +14 -0
- wandb/apis/public/api.py +450 -56
- wandb/apis/public/artifacts.py +31 -54
- wandb/apis/public/files.py +82 -14
- wandb/apis/public/jobs.py +12 -10
- wandb/apis/public/query_generator.py +1 -1
- wandb/apis/public/registries.py +573 -0
- wandb/apis/public/reports.py +6 -17
- wandb/apis/public/runs.py +137 -31
- wandb/apis/public/sweeps.py +1 -1
- wandb/apis/public/teams.py +3 -3
- wandb/apis/public/users.py +1 -1
- wandb/apis/public/utils.py +104 -0
- wandb/apis/reports/v1/__init__.py +1 -1
- wandb/apis/reports/v2/__init__.py +1 -1
- wandb/apis/workspaces/__init__.py +1 -1
- wandb/bin/gpu_stats +0 -0
- wandb/bin/wandb-core +0 -0
- wandb/cli/beta.py +181 -0
- wandb/cli/cli.py +50 -246
- wandb/data_types.py +13 -2019
- wandb/docker/__init__.py +6 -5
- wandb/docker/auth.py +2 -3
- wandb/env.py +90 -85
- wandb/errors/__init__.py +11 -40
- wandb/errors/errors.py +37 -0
- wandb/errors/links.py +73 -0
- wandb/errors/term.py +355 -43
- wandb/errors/warnings.py +2 -0
- wandb/filesync/dir_watcher.py +3 -3
- wandb/filesync/step_prepare.py +1 -1
- wandb/filesync/step_upload.py +2 -5
- wandb/filesync/upload_job.py +1 -1
- wandb/integration/catboost/catboost.py +2 -2
- wandb/integration/diffusers/pipeline_resolver.py +1 -1
- wandb/integration/diffusers/resolvers/multimodal.py +6 -6
- wandb/integration/diffusers/resolvers/utils.py +1 -1
- wandb/integration/fastai/__init__.py +8 -14
- wandb/integration/gym/__init__.py +1 -7
- wandb/integration/keras/__init__.py +2 -5
- wandb/integration/keras/callbacks/metrics_logger.py +12 -13
- wandb/integration/keras/callbacks/model_checkpoint.py +2 -14
- wandb/integration/keras/callbacks/tables_builder.py +3 -1
- wandb/integration/keras/keras.py +21 -10
- wandb/integration/kfp/kfp_patch.py +26 -16
- wandb/integration/lightgbm/__init__.py +3 -3
- wandb/integration/lightning/fabric/logger.py +3 -1
- wandb/integration/metaflow/metaflow.py +15 -17
- wandb/integration/openai/fine_tuning.py +5 -5
- wandb/integration/prodigy/prodigy.py +3 -11
- wandb/integration/sacred/__init__.py +1 -1
- wandb/integration/sagemaker/__init__.py +5 -3
- wandb/integration/sagemaker/auth.py +1 -1
- wandb/integration/sagemaker/config.py +17 -8
- wandb/integration/sagemaker/files.py +0 -1
- wandb/integration/sagemaker/resources.py +47 -18
- wandb/integration/sb3/sb3.py +1 -7
- wandb/{sklearn → integration/sklearn}/calculate/calibration_curves.py +7 -7
- wandb/{sklearn → integration/sklearn}/calculate/class_proportions.py +1 -1
- wandb/{sklearn → integration/sklearn}/calculate/confusion_matrix.py +3 -2
- wandb/{sklearn → integration/sklearn}/calculate/elbow_curve.py +6 -6
- wandb/{sklearn → integration/sklearn}/calculate/learning_curve.py +2 -2
- wandb/{sklearn → integration/sklearn}/calculate/outlier_candidates.py +2 -2
- wandb/{sklearn → integration/sklearn}/calculate/residuals.py +8 -8
- wandb/{sklearn → integration/sklearn}/calculate/silhouette.py +2 -2
- wandb/{sklearn → integration/sklearn}/calculate/summary_metrics.py +2 -2
- wandb/{sklearn → integration/sklearn}/plot/classifier.py +12 -12
- wandb/{sklearn → integration/sklearn}/plot/clusterer.py +13 -9
- wandb/{sklearn → integration/sklearn}/plot/regressor.py +8 -8
- wandb/{sklearn → integration/sklearn}/plot/shared.py +5 -5
- wandb/{sklearn → integration/sklearn}/utils.py +9 -9
- wandb/integration/tensorboard/log.py +1 -5
- wandb/integration/tensorboard/monkeypatch.py +3 -2
- wandb/{wandb_torch.py → integration/torch/wandb_torch.py} +38 -34
- wandb/integration/ultralytics/bbox_utils.py +9 -2
- wandb/integration/ultralytics/callback.py +2 -2
- wandb/integration/xgboost/xgboost.py +1 -1
- wandb/jupyter.py +26 -13
- wandb/old/core.py +2 -80
- wandb/plot/__init__.py +17 -8
- wandb/plot/bar.py +54 -26
- wandb/plot/confusion_matrix.py +150 -68
- wandb/plot/custom_chart.py +124 -0
- wandb/plot/histogram.py +48 -19
- wandb/plot/line.py +59 -25
- wandb/plot/line_series.py +148 -60
- wandb/plot/pr_curve.py +93 -44
- wandb/plot/roc_curve.py +82 -37
- wandb/plot/scatter.py +54 -20
- wandb/plot/viz.py +41 -0
- wandb/proto/v3/wandb_base_pb2.py +2 -1
- wandb/proto/v3/wandb_internal_pb2.py +368 -248
- wandb/proto/v3/wandb_server_pb2.py +58 -37
- wandb/proto/v3/wandb_settings_pb2.py +26 -15
- wandb/proto/v3/wandb_telemetry_pb2.py +13 -12
- wandb/proto/v4/wandb_base_pb2.py +2 -1
- wandb/proto/v4/wandb_internal_pb2.py +253 -228
- wandb/proto/v4/wandb_server_pb2.py +42 -37
- wandb/proto/v4/wandb_settings_pb2.py +18 -15
- wandb/proto/v4/wandb_telemetry_pb2.py +13 -12
- wandb/proto/v5/wandb_base_pb2.py +3 -2
- wandb/proto/v5/wandb_internal_pb2.py +254 -229
- wandb/proto/v5/wandb_server_pb2.py +43 -38
- wandb/proto/v5/wandb_settings_pb2.py +19 -16
- wandb/proto/v5/wandb_telemetry_pb2.py +14 -13
- wandb/proto/wandb_deprecated.py +7 -11
- wandb/proto/wandb_generate_deprecated.py +3 -7
- wandb/sdk/__init__.py +4 -4
- wandb/sdk/artifacts/_graphql_fragments.py +126 -0
- wandb/sdk/artifacts/_validators.py +121 -0
- wandb/sdk/artifacts/artifact.py +571 -508
- wandb/sdk/artifacts/artifact_file_cache.py +16 -18
- wandb/sdk/artifacts/artifact_instance_cache.py +4 -2
- wandb/sdk/artifacts/artifact_manifest.py +22 -19
- wandb/sdk/artifacts/artifact_manifest_entry.py +36 -26
- wandb/sdk/artifacts/artifact_manifests/artifact_manifest_v1.py +9 -7
- wandb/sdk/artifacts/artifact_saver.py +33 -29
- wandb/sdk/artifacts/exceptions.py +26 -25
- wandb/sdk/artifacts/storage_handler.py +11 -9
- wandb/sdk/artifacts/storage_handlers/azure_handler.py +25 -18
- wandb/sdk/artifacts/storage_handlers/gcs_handler.py +47 -18
- wandb/sdk/artifacts/storage_handlers/http_handler.py +15 -14
- wandb/sdk/artifacts/storage_handlers/local_file_handler.py +24 -17
- wandb/sdk/artifacts/storage_handlers/multi_handler.py +14 -12
- wandb/sdk/artifacts/storage_handlers/s3_handler.py +22 -24
- wandb/sdk/artifacts/storage_handlers/tracking_handler.py +10 -8
- wandb/sdk/artifacts/storage_handlers/wb_artifact_handler.py +14 -12
- wandb/sdk/artifacts/storage_handlers/wb_local_artifact_handler.py +11 -9
- wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py +31 -29
- wandb/sdk/artifacts/storage_policy.py +20 -20
- wandb/sdk/backend/backend.py +17 -40
- wandb/sdk/data_types/_dtypes.py +7 -3
- wandb/sdk/data_types/audio.py +165 -0
- wandb/sdk/data_types/base_types/media.py +66 -5
- wandb/sdk/data_types/base_types/wb_value.py +22 -14
- wandb/sdk/data_types/bokeh.py +70 -0
- wandb/sdk/data_types/graph.py +405 -0
- wandb/sdk/data_types/helper_types/bounding_boxes_2d.py +29 -11
- wandb/sdk/data_types/helper_types/image_mask.py +24 -12
- wandb/sdk/data_types/histogram.py +5 -7
- wandb/sdk/data_types/html.py +1 -1
- wandb/sdk/data_types/image.py +160 -5
- wandb/sdk/data_types/molecule.py +3 -3
- wandb/sdk/data_types/object_3d.py +6 -10
- wandb/sdk/data_types/plotly.py +1 -1
- wandb/sdk/data_types/saved_model.py +37 -49
- wandb/sdk/data_types/table.py +1204 -0
- wandb/sdk/data_types/trace_tree.py +3 -3
- wandb/sdk/data_types/utils.py +52 -8
- wandb/sdk/data_types/video.py +25 -13
- wandb/sdk/integration_utils/auto_logging.py +1 -8
- wandb/sdk/interface/interface.py +129 -88
- wandb/sdk/interface/interface_queue.py +5 -14
- wandb/sdk/interface/interface_relay.py +1 -6
- wandb/sdk/interface/interface_shared.py +51 -116
- wandb/sdk/interface/interface_sock.py +9 -24
- wandb/sdk/interface/router.py +21 -18
- wandb/sdk/interface/router_queue.py +2 -4
- wandb/sdk/interface/router_relay.py +2 -3
- wandb/sdk/interface/router_sock.py +5 -4
- wandb/sdk/internal/datastore.py +1 -1
- wandb/sdk/internal/file_pusher.py +1 -1
- wandb/sdk/internal/file_stream.py +9 -9
- wandb/sdk/internal/flow_control.py +1 -1
- wandb/sdk/internal/handler.py +33 -20
- wandb/sdk/internal/internal.py +5 -19
- wandb/sdk/internal/internal_api.py +374 -46
- wandb/sdk/internal/internal_util.py +0 -3
- wandb/sdk/internal/job_builder.py +24 -15
- wandb/sdk/internal/progress.py +1 -5
- wandb/sdk/internal/sample.py +2 -2
- wandb/sdk/internal/sender.py +32 -83
- wandb/sdk/internal/settings_static.py +23 -14
- wandb/sdk/internal/system/assets/__init__.py +0 -2
- wandb/sdk/internal/system/assets/cpu.py +3 -3
- wandb/sdk/internal/system/assets/disk.py +7 -7
- wandb/sdk/internal/system/assets/gpu.py +8 -8
- wandb/sdk/internal/system/assets/gpu_amd.py +12 -9
- wandb/sdk/internal/system/assets/interfaces.py +11 -13
- wandb/sdk/internal/system/assets/ipu.py +1 -1
- wandb/sdk/internal/system/assets/memory.py +3 -3
- wandb/sdk/internal/system/assets/open_metrics.py +2 -8
- wandb/sdk/internal/system/assets/trainium.py +5 -10
- wandb/sdk/internal/system/system_info.py +15 -14
- wandb/sdk/internal/system/system_monitor.py +8 -13
- wandb/sdk/internal/tb_watcher.py +16 -13
- wandb/sdk/internal/writer.py +2 -4
- wandb/sdk/launch/__init__.py +2 -1
- wandb/sdk/launch/_launch.py +4 -3
- wandb/sdk/launch/_launch_add.py +2 -2
- wandb/sdk/launch/_project_spec.py +3 -3
- wandb/sdk/launch/agent/run_queue_item_file_saver.py +1 -7
- wandb/sdk/launch/builder/abstract.py +1 -1
- wandb/sdk/launch/builder/kaniko_builder.py +0 -1
- wandb/sdk/launch/create_job.py +2 -2
- wandb/sdk/launch/environment/local_environment.py +0 -1
- wandb/sdk/launch/errors.py +0 -6
- wandb/sdk/launch/inputs/internal.py +2 -2
- wandb/sdk/launch/registry/local_registry.py +0 -2
- wandb/sdk/launch/runner/abstract.py +1 -11
- wandb/sdk/launch/runner/kubernetes_monitor.py +2 -4
- wandb/sdk/launch/sweeps/__init__.py +0 -2
- wandb/sdk/launch/sweeps/scheduler.py +0 -2
- wandb/sdk/launch/sweeps/scheduler_sweep.py +0 -1
- wandb/sdk/launch/utils.py +1 -0
- wandb/sdk/lib/apikey.py +100 -56
- wandb/sdk/lib/asyncio_compat.py +210 -0
- wandb/sdk/lib/file_stream_utils.py +1 -1
- wandb/sdk/lib/filesystem.py +1 -1
- wandb/sdk/lib/fsm.py +13 -7
- wandb/sdk/lib/gitlib.py +1 -1
- wandb/sdk/lib/gql_request.py +1 -1
- wandb/sdk/lib/hashutil.py +34 -12
- wandb/sdk/lib/interrupt.py +37 -0
- wandb/sdk/lib/ipython.py +22 -42
- wandb/sdk/lib/lazyloader.py +1 -1
- wandb/sdk/lib/module.py +12 -3
- wandb/sdk/lib/printer.py +358 -123
- wandb/sdk/lib/progress.py +341 -0
- wandb/sdk/lib/retry.py +7 -4
- wandb/sdk/lib/run_moment.py +19 -13
- wandb/sdk/lib/server.py +30 -24
- wandb/sdk/lib/service_connection.py +216 -0
- wandb/sdk/lib/service_token.py +94 -0
- wandb/sdk/lib/sock_client.py +14 -15
- wandb/sdk/lib/sparkline.py +1 -2
- wandb/sdk/lib/telemetry.py +1 -1
- wandb/sdk/mailbox/__init__.py +23 -0
- wandb/sdk/mailbox/handles.py +199 -0
- wandb/sdk/mailbox/mailbox.py +121 -0
- wandb/sdk/mailbox/wait_with_progress.py +134 -0
- wandb/sdk/service/_startup_debug.py +1 -1
- wandb/sdk/service/server.py +5 -17
- wandb/sdk/service/server_sock.py +9 -6
- wandb/sdk/service/service.py +12 -37
- wandb/sdk/service/streams.py +94 -99
- wandb/sdk/verify/verify.py +67 -15
- wandb/sdk/wandb_config.py +2 -2
- wandb/sdk/wandb_init.py +943 -697
- wandb/sdk/wandb_login.py +103 -133
- wandb/sdk/wandb_metadata.py +607 -0
- wandb/sdk/wandb_require.py +22 -14
- wandb/sdk/wandb_run.py +933 -1152
- wandb/sdk/wandb_settings.py +1331 -1735
- wandb/sdk/wandb_setup.py +170 -165
- wandb/sdk/wandb_sweep.py +2 -2
- wandb/sdk/wandb_sync.py +25 -23
- wandb/sdk/wandb_watch.py +49 -27
- wandb/sklearn.py +35 -0
- wandb/sync/sync.py +11 -12
- wandb/util.py +118 -66
- wandb/wandb_agent.py +14 -19
- wandb/wandb_controller.py +19 -21
- {wandb-0.17.8rc1.dist-info → wandb-0.19.7.dist-info}/METADATA +70 -63
- {wandb-0.17.8rc1.dist-info → wandb-0.19.7.dist-info}/RECORD +281 -266
- {wandb-0.17.8rc1.dist-info → wandb-0.19.7.dist-info}/WHEEL +1 -1
- wandb/bin/apple_gpu_stats +0 -0
- wandb/integration/magic.py +0 -556
- wandb/magic.py +0 -3
- wandb/sdk/internal/system/assets/gpu_apple.py +0 -177
- wandb/sdk/internal/update.py +0 -113
- wandb/sdk/lib/_settings_toposort_generate.py +0 -159
- wandb/sdk/lib/_settings_toposort_generated.py +0 -249
- wandb/sdk/lib/_wburls_generate.py +0 -25
- wandb/sdk/lib/_wburls_generated.py +0 -22
- wandb/sdk/lib/console.py +0 -39
- wandb/sdk/lib/mailbox.py +0 -460
- wandb/sdk/lib/reporting.py +0 -99
- wandb/sdk/lib/tracelog.py +0 -255
- wandb/sdk/lib/wburls.py +0 -46
- wandb/sdk/service/service_base.py +0 -50
- wandb/sdk/service/service_sock.py +0 -70
- wandb/sdk/wandb_manager.py +0 -235
- wandb/testing/relay.py +0 -874
- wandb/viz.py +0 -123
- /wandb/{sklearn → integration/sklearn}/__init__.py +0 -0
- /wandb/{sklearn → integration/sklearn}/calculate/__init__.py +0 -0
- /wandb/{sklearn → integration/sklearn}/calculate/decision_boundaries.py +0 -0
- /wandb/{sklearn → integration/sklearn}/calculate/feature_importances.py +0 -0
- /wandb/{sklearn → integration/sklearn}/plot/__init__.py +0 -0
- {wandb-0.17.8rc1.dist-info → wandb-0.19.7.dist-info}/entry_points.txt +0 -0
- {wandb-0.17.8rc1.dist-info → wandb-0.19.7.dist-info}/licenses/LICENSE +0 -0
package_readme.md
CHANGED
@@ -4,92 +4,94 @@
|
|
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:
|
28
|
+
|
29
|
+
```python
|
30
|
+
wandb.login()
|
31
|
+
```
|
32
|
+
|
33
|
+
4. Use the example code snippet below as a template to integrate W&B to your Python script:
|
35
34
|
|
36
35
|
```python
|
37
36
|
import wandb
|
38
37
|
|
39
|
-
#
|
40
|
-
|
38
|
+
# Start a W&B Run with wandb.init
|
39
|
+
run = wandb.init(project="my_first_project")
|
41
40
|
|
42
|
-
wandb.
|
43
|
-
|
41
|
+
# Save model inputs and hyperparameters in a wandb.config object
|
42
|
+
config = run.config
|
43
|
+
config.learning_rate = 0.01
|
44
44
|
|
45
|
+
# Model training code here ...
|
45
46
|
|
46
|
-
|
47
|
-
|
48
|
-
|
49
|
-
|
50
|
-
|
51
|
-
|
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()
|
52
53
|
```
|
53
54
|
|
54
|
-
|
55
|
+
For example, if the preceding code was stored in a script called train.py:
|
55
56
|
|
56
|
-
```
|
57
|
-
|
57
|
+
```shell
|
58
|
+
python train.py
|
58
59
|
```
|
59
60
|
|
60
|
-
|
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!
|
61
62
|
|
62
|
-
|
63
|
+
|
63
64
|
|
64
|
-
|
65
|
+
# Integrations
|
65
66
|
|
66
|
-
|
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.
|
67
68
|
|
68
|
-
|
69
|
+
|
69
70
|
|
70
|
-
|
71
|
-
[](https://youtu.be/EeqhOSvNX-A)
|
72
|
-
[Introduction video →](https://youtu.be/EeqhOSvNX-A)
|
71
|
+
# Python Version Support
|
73
72
|
|
74
|
-
|
73
|
+
We are committed to supporting our minimum required Python version for *at least* six months after its official end-of-life (EOL) date, as defined by the Python Software Foundation. You can find a list of Python EOL dates [here](https://devguide.python.org/versions/).
|
75
74
|
|
76
|
-
|
75
|
+
When we discontinue support for a Python version, we will increment the library’s minor version number to reflect this change.
|
77
76
|
|
78
|
-
|
77
|
+
|
79
78
|
|
80
|
-
|
79
|
+
# Contribution guidelines
|
80
|
+
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.
|
81
81
|
|
82
|
-
|
82
|
+
|
83
83
|
|
84
84
|
# Academic Researchers
|
85
|
-
|
85
|
+
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.
|
86
|
+
|
87
|
+
|
88
|
+
|
89
|
+
# W&B Community
|
86
90
|
|
87
|
-
|
88
|
-
[](https://www.wandb.com/academic)
|
91
|
+
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
92
|
|
90
|
-
|
91
|
-
Got questions, feedback or want to join a community of ML engineers working on exciting projects?
|
93
|
+
|
92
94
|
|
93
|
-
|
95
|
+
# License
|
94
96
|
|
95
|
-
[
|
97
|
+
[MIT License](https://github.com/wandb/wandb/blob/main/LICENSE)
|
wandb/__init__.py
CHANGED
@@ -8,9 +8,10 @@ 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
|
-
|
11
|
+
from __future__ import annotations
|
12
|
+
|
13
|
+
__version__ = "0.19.7"
|
12
14
|
|
13
|
-
from typing import Optional
|
14
15
|
|
15
16
|
from wandb.errors import Error
|
16
17
|
|
@@ -28,8 +29,6 @@ setup = wandb_sdk.setup
|
|
28
29
|
_attach = wandb_sdk._attach
|
29
30
|
_sync = wandb_sdk._sync
|
30
31
|
_teardown = wandb_sdk.teardown
|
31
|
-
watch = wandb_sdk.watch
|
32
|
-
unwatch = wandb_sdk.unwatch
|
33
32
|
finish = wandb_sdk.finish
|
34
33
|
join = finish
|
35
34
|
login = wandb_sdk.login
|
@@ -51,7 +50,7 @@ _lazyloader = wandb.wandb_lib.lazyloader # type: ignore
|
|
51
50
|
# Call import module hook to set up any needed require hooks
|
52
51
|
wandb.sdk.wandb_require._import_module_hook()
|
53
52
|
|
54
|
-
from wandb import wandb_torch
|
53
|
+
from wandb.integration.torch import wandb_torch
|
55
54
|
|
56
55
|
# Move this (keras.__init__ expects it at top level)
|
57
56
|
from wandb.sdk.data_types._private import _cleanup_media_tmp_dir
|
@@ -76,9 +75,7 @@ from wandb.data_types import JoinedTable
|
|
76
75
|
|
77
76
|
from wandb.wandb_agent import agent
|
78
77
|
|
79
|
-
|
80
|
-
from wandb.viz import visualize
|
81
|
-
from wandb import plot
|
78
|
+
from wandb.plot import visualize, plot_table
|
82
79
|
from wandb.integration.sagemaker import sagemaker_auth
|
83
80
|
from wandb.sdk.internal import profiler
|
84
81
|
|
@@ -114,10 +111,12 @@ def _assert_is_user_process():
|
|
114
111
|
# globals
|
115
112
|
Api = PublicApi
|
116
113
|
api = InternalApi()
|
117
|
-
run:
|
114
|
+
run: wandb_sdk.wandb_run.Run | None = None
|
118
115
|
config = _preinit.PreInitObject("wandb.config", wandb_sdk.wandb_config.Config)
|
119
116
|
summary = _preinit.PreInitObject("wandb.summary", wandb_sdk.wandb_summary.Summary)
|
120
117
|
log = _preinit.PreInitCallable("wandb.log", wandb_sdk.wandb_run.Run.log) # type: ignore
|
118
|
+
watch = _preinit.PreInitCallable("wandb.watch", wandb_sdk.wandb_run.Run.watch) # type: ignore
|
119
|
+
unwatch = _preinit.PreInitCallable("wandb.unwatch", wandb_sdk.wandb_run.Run.unwatch) # type: ignore
|
121
120
|
save = _preinit.PreInitCallable("wandb.save", wandb_sdk.wandb_run.Run.save) # type: ignore
|
122
121
|
restore = wandb_sdk.wandb_run.restore
|
123
122
|
use_artifact = _preinit.PreInitCallable(
|
@@ -143,9 +142,6 @@ mark_preempting = _preinit.PreInitCallable(
|
|
143
142
|
"wandb.mark_preempting", wandb_sdk.wandb_run.Run.mark_preempting # type: ignore
|
144
143
|
)
|
145
144
|
|
146
|
-
plot_table = _preinit.PreInitCallable(
|
147
|
-
"wandb.plot_table", wandb_sdk.wandb_run.Run.plot_table
|
148
|
-
)
|
149
145
|
alert = _preinit.PreInitCallable("wandb.alert", wandb_sdk.wandb_run.Run.alert) # type: ignore
|
150
146
|
|
151
147
|
# record of patched libraries
|
@@ -199,14 +195,14 @@ if wandb_sdk.lib.ipython.in_notebook():
|
|
199
195
|
from .analytics import Sentry as _Sentry
|
200
196
|
|
201
197
|
if "dev" in __version__:
|
198
|
+
import wandb.env
|
202
199
|
import os
|
203
200
|
|
204
|
-
#
|
205
|
-
os.environ[
|
206
|
-
|
201
|
+
# Disable error reporting in dev versions.
|
202
|
+
os.environ[wandb.env.ERROR_REPORTING] = os.environ.get(
|
203
|
+
wandb.env.ERROR_REPORTING,
|
204
|
+
"false",
|
207
205
|
)
|
208
|
-
# turn on wandb-core for dev versions
|
209
|
-
os.environ["WANDB__REQUIRE_CORE"] = os.environ.get("WANDB__REQUIRE_CORE", "true")
|
210
206
|
|
211
207
|
_sentry = _Sentry()
|
212
208
|
_sentry.setup()
|
@@ -215,6 +211,7 @@ _sentry.setup()
|
|
215
211
|
__all__ = (
|
216
212
|
"__version__",
|
217
213
|
"init",
|
214
|
+
"finish",
|
218
215
|
"setup",
|
219
216
|
"save",
|
220
217
|
"sweep",
|
@@ -237,8 +234,13 @@ __all__ = (
|
|
237
234
|
"Molecule",
|
238
235
|
"Histogram",
|
239
236
|
"ArtifactTTL",
|
237
|
+
"log_artifact",
|
238
|
+
"use_artifact",
|
240
239
|
"log_model",
|
241
240
|
"use_model",
|
242
241
|
"link_model",
|
243
242
|
"define_metric",
|
243
|
+
"watch",
|
244
|
+
"unwatch",
|
245
|
+
"plot_table",
|
244
246
|
)
|