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.
Files changed (300) hide show
  1. package_readme.md +52 -50
  2. wandb/__init__.py +19 -17
  3. wandb/__init__.pyi +530 -276
  4. wandb/_iterutils.py +46 -0
  5. wandb/agents/pyagent.py +2 -2
  6. wandb/analytics/sentry.py +92 -95
  7. wandb/apis/attrs.py +15 -4
  8. wandb/apis/importers/internals/internal.py +6 -6
  9. wandb/apis/importers/internals/protocols.py +11 -7
  10. wandb/apis/importers/wandb.py +2 -2
  11. wandb/apis/internal.py +7 -0
  12. wandb/apis/normalize.py +15 -23
  13. wandb/apis/public/_generated/__init__.py +21 -0
  14. wandb/apis/public/_generated/base.py +128 -0
  15. wandb/apis/public/_generated/enums.py +4 -0
  16. wandb/apis/public/_generated/input_types.py +4 -0
  17. wandb/apis/public/_generated/operations.py +15 -0
  18. wandb/apis/public/_generated/server_features_query.py +27 -0
  19. wandb/apis/public/_generated/typing_compat.py +14 -0
  20. wandb/apis/public/api.py +450 -56
  21. wandb/apis/public/artifacts.py +31 -54
  22. wandb/apis/public/files.py +82 -14
  23. wandb/apis/public/jobs.py +12 -10
  24. wandb/apis/public/query_generator.py +1 -1
  25. wandb/apis/public/registries.py +573 -0
  26. wandb/apis/public/reports.py +6 -17
  27. wandb/apis/public/runs.py +137 -31
  28. wandb/apis/public/sweeps.py +1 -1
  29. wandb/apis/public/teams.py +3 -3
  30. wandb/apis/public/users.py +1 -1
  31. wandb/apis/public/utils.py +104 -0
  32. wandb/apis/reports/v1/__init__.py +1 -1
  33. wandb/apis/reports/v2/__init__.py +1 -1
  34. wandb/apis/workspaces/__init__.py +1 -1
  35. wandb/bin/gpu_stats +0 -0
  36. wandb/bin/wandb-core +0 -0
  37. wandb/cli/beta.py +181 -0
  38. wandb/cli/cli.py +50 -246
  39. wandb/data_types.py +13 -2019
  40. wandb/docker/__init__.py +6 -5
  41. wandb/docker/auth.py +2 -3
  42. wandb/env.py +90 -85
  43. wandb/errors/__init__.py +11 -40
  44. wandb/errors/errors.py +37 -0
  45. wandb/errors/links.py +73 -0
  46. wandb/errors/term.py +355 -43
  47. wandb/errors/warnings.py +2 -0
  48. wandb/filesync/dir_watcher.py +3 -3
  49. wandb/filesync/step_prepare.py +1 -1
  50. wandb/filesync/step_upload.py +2 -5
  51. wandb/filesync/upload_job.py +1 -1
  52. wandb/integration/catboost/catboost.py +2 -2
  53. wandb/integration/diffusers/pipeline_resolver.py +1 -1
  54. wandb/integration/diffusers/resolvers/multimodal.py +6 -6
  55. wandb/integration/diffusers/resolvers/utils.py +1 -1
  56. wandb/integration/fastai/__init__.py +8 -14
  57. wandb/integration/gym/__init__.py +1 -7
  58. wandb/integration/keras/__init__.py +2 -5
  59. wandb/integration/keras/callbacks/metrics_logger.py +12 -13
  60. wandb/integration/keras/callbacks/model_checkpoint.py +2 -14
  61. wandb/integration/keras/callbacks/tables_builder.py +3 -1
  62. wandb/integration/keras/keras.py +21 -10
  63. wandb/integration/kfp/kfp_patch.py +26 -16
  64. wandb/integration/lightgbm/__init__.py +3 -3
  65. wandb/integration/lightning/fabric/logger.py +3 -1
  66. wandb/integration/metaflow/metaflow.py +15 -17
  67. wandb/integration/openai/fine_tuning.py +5 -5
  68. wandb/integration/prodigy/prodigy.py +3 -11
  69. wandb/integration/sacred/__init__.py +1 -1
  70. wandb/integration/sagemaker/__init__.py +5 -3
  71. wandb/integration/sagemaker/auth.py +1 -1
  72. wandb/integration/sagemaker/config.py +17 -8
  73. wandb/integration/sagemaker/files.py +0 -1
  74. wandb/integration/sagemaker/resources.py +47 -18
  75. wandb/integration/sb3/sb3.py +1 -7
  76. wandb/{sklearn → integration/sklearn}/calculate/calibration_curves.py +7 -7
  77. wandb/{sklearn → integration/sklearn}/calculate/class_proportions.py +1 -1
  78. wandb/{sklearn → integration/sklearn}/calculate/confusion_matrix.py +3 -2
  79. wandb/{sklearn → integration/sklearn}/calculate/elbow_curve.py +6 -6
  80. wandb/{sklearn → integration/sklearn}/calculate/learning_curve.py +2 -2
  81. wandb/{sklearn → integration/sklearn}/calculate/outlier_candidates.py +2 -2
  82. wandb/{sklearn → integration/sklearn}/calculate/residuals.py +8 -8
  83. wandb/{sklearn → integration/sklearn}/calculate/silhouette.py +2 -2
  84. wandb/{sklearn → integration/sklearn}/calculate/summary_metrics.py +2 -2
  85. wandb/{sklearn → integration/sklearn}/plot/classifier.py +12 -12
  86. wandb/{sklearn → integration/sklearn}/plot/clusterer.py +13 -9
  87. wandb/{sklearn → integration/sklearn}/plot/regressor.py +8 -8
  88. wandb/{sklearn → integration/sklearn}/plot/shared.py +5 -5
  89. wandb/{sklearn → integration/sklearn}/utils.py +9 -9
  90. wandb/integration/tensorboard/log.py +1 -5
  91. wandb/integration/tensorboard/monkeypatch.py +3 -2
  92. wandb/{wandb_torch.py → integration/torch/wandb_torch.py} +38 -34
  93. wandb/integration/ultralytics/bbox_utils.py +9 -2
  94. wandb/integration/ultralytics/callback.py +2 -2
  95. wandb/integration/xgboost/xgboost.py +1 -1
  96. wandb/jupyter.py +26 -13
  97. wandb/old/core.py +2 -80
  98. wandb/plot/__init__.py +17 -8
  99. wandb/plot/bar.py +54 -26
  100. wandb/plot/confusion_matrix.py +150 -68
  101. wandb/plot/custom_chart.py +124 -0
  102. wandb/plot/histogram.py +48 -19
  103. wandb/plot/line.py +59 -25
  104. wandb/plot/line_series.py +148 -60
  105. wandb/plot/pr_curve.py +93 -44
  106. wandb/plot/roc_curve.py +82 -37
  107. wandb/plot/scatter.py +54 -20
  108. wandb/plot/viz.py +41 -0
  109. wandb/proto/v3/wandb_base_pb2.py +2 -1
  110. wandb/proto/v3/wandb_internal_pb2.py +368 -248
  111. wandb/proto/v3/wandb_server_pb2.py +58 -37
  112. wandb/proto/v3/wandb_settings_pb2.py +26 -15
  113. wandb/proto/v3/wandb_telemetry_pb2.py +13 -12
  114. wandb/proto/v4/wandb_base_pb2.py +2 -1
  115. wandb/proto/v4/wandb_internal_pb2.py +253 -228
  116. wandb/proto/v4/wandb_server_pb2.py +42 -37
  117. wandb/proto/v4/wandb_settings_pb2.py +18 -15
  118. wandb/proto/v4/wandb_telemetry_pb2.py +13 -12
  119. wandb/proto/v5/wandb_base_pb2.py +3 -2
  120. wandb/proto/v5/wandb_internal_pb2.py +254 -229
  121. wandb/proto/v5/wandb_server_pb2.py +43 -38
  122. wandb/proto/v5/wandb_settings_pb2.py +19 -16
  123. wandb/proto/v5/wandb_telemetry_pb2.py +14 -13
  124. wandb/proto/wandb_deprecated.py +7 -11
  125. wandb/proto/wandb_generate_deprecated.py +3 -7
  126. wandb/sdk/__init__.py +4 -4
  127. wandb/sdk/artifacts/_graphql_fragments.py +126 -0
  128. wandb/sdk/artifacts/_validators.py +121 -0
  129. wandb/sdk/artifacts/artifact.py +571 -508
  130. wandb/sdk/artifacts/artifact_file_cache.py +16 -18
  131. wandb/sdk/artifacts/artifact_instance_cache.py +4 -2
  132. wandb/sdk/artifacts/artifact_manifest.py +22 -19
  133. wandb/sdk/artifacts/artifact_manifest_entry.py +36 -26
  134. wandb/sdk/artifacts/artifact_manifests/artifact_manifest_v1.py +9 -7
  135. wandb/sdk/artifacts/artifact_saver.py +33 -29
  136. wandb/sdk/artifacts/exceptions.py +26 -25
  137. wandb/sdk/artifacts/storage_handler.py +11 -9
  138. wandb/sdk/artifacts/storage_handlers/azure_handler.py +25 -18
  139. wandb/sdk/artifacts/storage_handlers/gcs_handler.py +47 -18
  140. wandb/sdk/artifacts/storage_handlers/http_handler.py +15 -14
  141. wandb/sdk/artifacts/storage_handlers/local_file_handler.py +24 -17
  142. wandb/sdk/artifacts/storage_handlers/multi_handler.py +14 -12
  143. wandb/sdk/artifacts/storage_handlers/s3_handler.py +22 -24
  144. wandb/sdk/artifacts/storage_handlers/tracking_handler.py +10 -8
  145. wandb/sdk/artifacts/storage_handlers/wb_artifact_handler.py +14 -12
  146. wandb/sdk/artifacts/storage_handlers/wb_local_artifact_handler.py +11 -9
  147. wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py +31 -29
  148. wandb/sdk/artifacts/storage_policy.py +20 -20
  149. wandb/sdk/backend/backend.py +17 -40
  150. wandb/sdk/data_types/_dtypes.py +7 -3
  151. wandb/sdk/data_types/audio.py +165 -0
  152. wandb/sdk/data_types/base_types/media.py +66 -5
  153. wandb/sdk/data_types/base_types/wb_value.py +22 -14
  154. wandb/sdk/data_types/bokeh.py +70 -0
  155. wandb/sdk/data_types/graph.py +405 -0
  156. wandb/sdk/data_types/helper_types/bounding_boxes_2d.py +29 -11
  157. wandb/sdk/data_types/helper_types/image_mask.py +24 -12
  158. wandb/sdk/data_types/histogram.py +5 -7
  159. wandb/sdk/data_types/html.py +1 -1
  160. wandb/sdk/data_types/image.py +160 -5
  161. wandb/sdk/data_types/molecule.py +3 -3
  162. wandb/sdk/data_types/object_3d.py +6 -10
  163. wandb/sdk/data_types/plotly.py +1 -1
  164. wandb/sdk/data_types/saved_model.py +37 -49
  165. wandb/sdk/data_types/table.py +1204 -0
  166. wandb/sdk/data_types/trace_tree.py +3 -3
  167. wandb/sdk/data_types/utils.py +52 -8
  168. wandb/sdk/data_types/video.py +25 -13
  169. wandb/sdk/integration_utils/auto_logging.py +1 -8
  170. wandb/sdk/interface/interface.py +129 -88
  171. wandb/sdk/interface/interface_queue.py +5 -14
  172. wandb/sdk/interface/interface_relay.py +1 -6
  173. wandb/sdk/interface/interface_shared.py +51 -116
  174. wandb/sdk/interface/interface_sock.py +9 -24
  175. wandb/sdk/interface/router.py +21 -18
  176. wandb/sdk/interface/router_queue.py +2 -4
  177. wandb/sdk/interface/router_relay.py +2 -3
  178. wandb/sdk/interface/router_sock.py +5 -4
  179. wandb/sdk/internal/datastore.py +1 -1
  180. wandb/sdk/internal/file_pusher.py +1 -1
  181. wandb/sdk/internal/file_stream.py +9 -9
  182. wandb/sdk/internal/flow_control.py +1 -1
  183. wandb/sdk/internal/handler.py +33 -20
  184. wandb/sdk/internal/internal.py +5 -19
  185. wandb/sdk/internal/internal_api.py +374 -46
  186. wandb/sdk/internal/internal_util.py +0 -3
  187. wandb/sdk/internal/job_builder.py +24 -15
  188. wandb/sdk/internal/progress.py +1 -5
  189. wandb/sdk/internal/sample.py +2 -2
  190. wandb/sdk/internal/sender.py +32 -83
  191. wandb/sdk/internal/settings_static.py +23 -14
  192. wandb/sdk/internal/system/assets/__init__.py +0 -2
  193. wandb/sdk/internal/system/assets/cpu.py +3 -3
  194. wandb/sdk/internal/system/assets/disk.py +7 -7
  195. wandb/sdk/internal/system/assets/gpu.py +8 -8
  196. wandb/sdk/internal/system/assets/gpu_amd.py +12 -9
  197. wandb/sdk/internal/system/assets/interfaces.py +11 -13
  198. wandb/sdk/internal/system/assets/ipu.py +1 -1
  199. wandb/sdk/internal/system/assets/memory.py +3 -3
  200. wandb/sdk/internal/system/assets/open_metrics.py +2 -8
  201. wandb/sdk/internal/system/assets/trainium.py +5 -10
  202. wandb/sdk/internal/system/system_info.py +15 -14
  203. wandb/sdk/internal/system/system_monitor.py +8 -13
  204. wandb/sdk/internal/tb_watcher.py +16 -13
  205. wandb/sdk/internal/writer.py +2 -4
  206. wandb/sdk/launch/__init__.py +2 -1
  207. wandb/sdk/launch/_launch.py +4 -3
  208. wandb/sdk/launch/_launch_add.py +2 -2
  209. wandb/sdk/launch/_project_spec.py +3 -3
  210. wandb/sdk/launch/agent/run_queue_item_file_saver.py +1 -7
  211. wandb/sdk/launch/builder/abstract.py +1 -1
  212. wandb/sdk/launch/builder/kaniko_builder.py +0 -1
  213. wandb/sdk/launch/create_job.py +2 -2
  214. wandb/sdk/launch/environment/local_environment.py +0 -1
  215. wandb/sdk/launch/errors.py +0 -6
  216. wandb/sdk/launch/inputs/internal.py +2 -2
  217. wandb/sdk/launch/registry/local_registry.py +0 -2
  218. wandb/sdk/launch/runner/abstract.py +1 -11
  219. wandb/sdk/launch/runner/kubernetes_monitor.py +2 -4
  220. wandb/sdk/launch/sweeps/__init__.py +0 -2
  221. wandb/sdk/launch/sweeps/scheduler.py +0 -2
  222. wandb/sdk/launch/sweeps/scheduler_sweep.py +0 -1
  223. wandb/sdk/launch/utils.py +1 -0
  224. wandb/sdk/lib/apikey.py +100 -56
  225. wandb/sdk/lib/asyncio_compat.py +210 -0
  226. wandb/sdk/lib/file_stream_utils.py +1 -1
  227. wandb/sdk/lib/filesystem.py +1 -1
  228. wandb/sdk/lib/fsm.py +13 -7
  229. wandb/sdk/lib/gitlib.py +1 -1
  230. wandb/sdk/lib/gql_request.py +1 -1
  231. wandb/sdk/lib/hashutil.py +34 -12
  232. wandb/sdk/lib/interrupt.py +37 -0
  233. wandb/sdk/lib/ipython.py +22 -42
  234. wandb/sdk/lib/lazyloader.py +1 -1
  235. wandb/sdk/lib/module.py +12 -3
  236. wandb/sdk/lib/printer.py +358 -123
  237. wandb/sdk/lib/progress.py +341 -0
  238. wandb/sdk/lib/retry.py +7 -4
  239. wandb/sdk/lib/run_moment.py +19 -13
  240. wandb/sdk/lib/server.py +30 -24
  241. wandb/sdk/lib/service_connection.py +216 -0
  242. wandb/sdk/lib/service_token.py +94 -0
  243. wandb/sdk/lib/sock_client.py +14 -15
  244. wandb/sdk/lib/sparkline.py +1 -2
  245. wandb/sdk/lib/telemetry.py +1 -1
  246. wandb/sdk/mailbox/__init__.py +23 -0
  247. wandb/sdk/mailbox/handles.py +199 -0
  248. wandb/sdk/mailbox/mailbox.py +121 -0
  249. wandb/sdk/mailbox/wait_with_progress.py +134 -0
  250. wandb/sdk/service/_startup_debug.py +1 -1
  251. wandb/sdk/service/server.py +5 -17
  252. wandb/sdk/service/server_sock.py +9 -6
  253. wandb/sdk/service/service.py +12 -37
  254. wandb/sdk/service/streams.py +94 -99
  255. wandb/sdk/verify/verify.py +67 -15
  256. wandb/sdk/wandb_config.py +2 -2
  257. wandb/sdk/wandb_init.py +943 -697
  258. wandb/sdk/wandb_login.py +103 -133
  259. wandb/sdk/wandb_metadata.py +607 -0
  260. wandb/sdk/wandb_require.py +22 -14
  261. wandb/sdk/wandb_run.py +933 -1152
  262. wandb/sdk/wandb_settings.py +1331 -1735
  263. wandb/sdk/wandb_setup.py +170 -165
  264. wandb/sdk/wandb_sweep.py +2 -2
  265. wandb/sdk/wandb_sync.py +25 -23
  266. wandb/sdk/wandb_watch.py +49 -27
  267. wandb/sklearn.py +35 -0
  268. wandb/sync/sync.py +11 -12
  269. wandb/util.py +118 -66
  270. wandb/wandb_agent.py +14 -19
  271. wandb/wandb_controller.py +19 -21
  272. {wandb-0.17.8rc1.dist-info → wandb-0.19.7.dist-info}/METADATA +70 -63
  273. {wandb-0.17.8rc1.dist-info → wandb-0.19.7.dist-info}/RECORD +281 -266
  274. {wandb-0.17.8rc1.dist-info → wandb-0.19.7.dist-info}/WHEEL +1 -1
  275. wandb/bin/apple_gpu_stats +0 -0
  276. wandb/integration/magic.py +0 -556
  277. wandb/magic.py +0 -3
  278. wandb/sdk/internal/system/assets/gpu_apple.py +0 -177
  279. wandb/sdk/internal/update.py +0 -113
  280. wandb/sdk/lib/_settings_toposort_generate.py +0 -159
  281. wandb/sdk/lib/_settings_toposort_generated.py +0 -249
  282. wandb/sdk/lib/_wburls_generate.py +0 -25
  283. wandb/sdk/lib/_wburls_generated.py +0 -22
  284. wandb/sdk/lib/console.py +0 -39
  285. wandb/sdk/lib/mailbox.py +0 -460
  286. wandb/sdk/lib/reporting.py +0 -99
  287. wandb/sdk/lib/tracelog.py +0 -255
  288. wandb/sdk/lib/wburls.py +0 -46
  289. wandb/sdk/service/service_base.py +0 -50
  290. wandb/sdk/service/service_sock.py +0 -70
  291. wandb/sdk/wandb_manager.py +0 -235
  292. wandb/testing/relay.py +0 -874
  293. wandb/viz.py +0 -123
  294. /wandb/{sklearn → integration/sklearn}/__init__.py +0 -0
  295. /wandb/{sklearn → integration/sklearn}/calculate/__init__.py +0 -0
  296. /wandb/{sklearn → integration/sklearn}/calculate/decision_boundaries.py +0 -0
  297. /wandb/{sklearn → integration/sklearn}/calculate/feature_importances.py +0 -0
  298. /wandb/{sklearn → integration/sklearn}/plot/__init__.py +0 -0
  299. {wandb-0.17.8rc1.dist-info → wandb-0.19.7.dist-info}/entry_points.txt +0 -0
  300. {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 [![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:
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
- # Your custom arguments defined here
40
- args = ...
38
+ # Start a W&B Run with wandb.init
39
+ run = wandb.init(project="my_first_project")
41
40
 
42
- wandb.init(config=args, project="my-project")
43
- wandb.config["more"] = "custom"
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
- 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})
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
- If you're already using Tensorboard or [TensorboardX](https://github.com/lanpa/tensorboardX), you can integrate with one line:
55
+ For example, if the preceding code was stored in a script called train.py:
55
56
 
56
- ```python
57
- wandb.init(sync_tensorboard=True)
57
+ ```shell
58
+ python train.py
58
59
  ```
59
60
 
60
- ## Running your script
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
- 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).
63
+  
63
64
 
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`.
65
+ # Integrations
65
66
 
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.
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
- ## Web Interface
69
+  
69
70
 
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)
71
+ # Python Version Support
73
72
 
74
- ## Detailed Usage
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
- Framework specific and detailed usage can be found in our [documentation](http://docs.wandb.com/).
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
- ## Testing
77
+  
79
78
 
80
- To run basic test use `make test`. More detailed information can be found at CONTRIBUTING.md.
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
- We use [circleci](https://circleci.com) for CI.
82
+  
83
83
 
84
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)
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
- 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)
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
- ## Community
91
- Got questions, feedback or want to join a community of ML engineers working on exciting projects?
93
+  
92
94
 
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.
95
+ # License
94
96
 
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).
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
- __version__ = "0.17.8rc1"
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
- # from wandb.core import *
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: Optional["wandb_sdk.wandb_run.Run"] = None
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
- # disable error reporting in dev versions for the python client
205
- os.environ["WANDB_ERROR_REPORTING"] = os.environ.get(
206
- "WANDB_ERROR_REPORTING", "false"
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
  )