wandb 0.18.0rc1__py3-none-any.whl → 0.18.1__py3-none-any.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (62) hide show
  1. wandb/__init__.py +2 -2
  2. wandb/__init__.pyi +1 -1
  3. wandb/apis/public/runs.py +2 -0
  4. wandb/bin/nvidia_gpu_stats +0 -0
  5. wandb/cli/cli.py +0 -2
  6. wandb/data_types.py +9 -2019
  7. wandb/env.py +0 -5
  8. wandb/{sklearn → integration/sklearn}/calculate/calibration_curves.py +7 -7
  9. wandb/{sklearn → integration/sklearn}/calculate/class_proportions.py +1 -1
  10. wandb/{sklearn → integration/sklearn}/calculate/confusion_matrix.py +3 -2
  11. wandb/{sklearn → integration/sklearn}/calculate/elbow_curve.py +6 -6
  12. wandb/{sklearn → integration/sklearn}/calculate/learning_curve.py +2 -2
  13. wandb/{sklearn → integration/sklearn}/calculate/outlier_candidates.py +2 -2
  14. wandb/{sklearn → integration/sklearn}/calculate/residuals.py +8 -8
  15. wandb/{sklearn → integration/sklearn}/calculate/silhouette.py +2 -2
  16. wandb/{sklearn → integration/sklearn}/calculate/summary_metrics.py +2 -2
  17. wandb/{sklearn → integration/sklearn}/plot/classifier.py +5 -5
  18. wandb/{sklearn → integration/sklearn}/plot/clusterer.py +10 -6
  19. wandb/{sklearn → integration/sklearn}/plot/regressor.py +5 -5
  20. wandb/{sklearn → integration/sklearn}/plot/shared.py +3 -3
  21. wandb/{sklearn → integration/sklearn}/utils.py +8 -8
  22. wandb/{wandb_torch.py → integration/torch/wandb_torch.py} +36 -32
  23. wandb/proto/v3/wandb_base_pb2.py +2 -1
  24. wandb/proto/v3/wandb_internal_pb2.py +2 -1
  25. wandb/proto/v3/wandb_server_pb2.py +2 -1
  26. wandb/proto/v3/wandb_settings_pb2.py +2 -1
  27. wandb/proto/v3/wandb_telemetry_pb2.py +2 -1
  28. wandb/proto/v4/wandb_base_pb2.py +2 -1
  29. wandb/proto/v4/wandb_internal_pb2.py +2 -1
  30. wandb/proto/v4/wandb_server_pb2.py +2 -1
  31. wandb/proto/v4/wandb_settings_pb2.py +2 -1
  32. wandb/proto/v4/wandb_telemetry_pb2.py +2 -1
  33. wandb/proto/v5/wandb_base_pb2.py +3 -2
  34. wandb/proto/v5/wandb_internal_pb2.py +3 -2
  35. wandb/proto/v5/wandb_server_pb2.py +3 -2
  36. wandb/proto/v5/wandb_settings_pb2.py +3 -2
  37. wandb/proto/v5/wandb_telemetry_pb2.py +3 -2
  38. wandb/sdk/data_types/audio.py +165 -0
  39. wandb/sdk/data_types/bokeh.py +70 -0
  40. wandb/sdk/data_types/graph.py +405 -0
  41. wandb/sdk/data_types/image.py +156 -0
  42. wandb/sdk/data_types/table.py +1204 -0
  43. wandb/sdk/data_types/trace_tree.py +2 -2
  44. wandb/sdk/data_types/utils.py +49 -0
  45. wandb/sdk/service/service.py +2 -9
  46. wandb/sdk/service/streams.py +0 -7
  47. wandb/sdk/wandb_init.py +10 -3
  48. wandb/sdk/wandb_run.py +6 -152
  49. wandb/sdk/wandb_setup.py +1 -1
  50. wandb/sklearn.py +35 -0
  51. wandb/util.py +6 -2
  52. {wandb-0.18.0rc1.dist-info → wandb-0.18.1.dist-info}/METADATA +5 -5
  53. {wandb-0.18.0rc1.dist-info → wandb-0.18.1.dist-info}/RECORD +61 -57
  54. wandb/sdk/lib/console.py +0 -39
  55. /wandb/{sklearn → integration/sklearn}/__init__.py +0 -0
  56. /wandb/{sklearn → integration/sklearn}/calculate/__init__.py +0 -0
  57. /wandb/{sklearn → integration/sklearn}/calculate/decision_boundaries.py +0 -0
  58. /wandb/{sklearn → integration/sklearn}/calculate/feature_importances.py +0 -0
  59. /wandb/{sklearn → integration/sklearn}/plot/__init__.py +0 -0
  60. {wandb-0.18.0rc1.dist-info → wandb-0.18.1.dist-info}/WHEEL +0 -0
  61. {wandb-0.18.0rc1.dist-info → wandb-0.18.1.dist-info}/entry_points.txt +0 -0
  62. {wandb-0.18.0rc1.dist-info → wandb-0.18.1.dist-info}/licenses/LICENSE +0 -0
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.18.0rc1"
11
+ __version__ = "0.18.1"
12
12
 
13
13
  from typing import Optional
14
14
 
@@ -51,7 +51,7 @@ _lazyloader = wandb.wandb_lib.lazyloader # type: ignore
51
51
  # Call import module hook to set up any needed require hooks
52
52
  wandb.sdk.wandb_require._import_module_hook()
53
53
 
54
- from wandb import wandb_torch
54
+ from wandb.integration.torch import wandb_torch
55
55
 
56
56
  # Move this (keras.__init__ expects it at top level)
57
57
  from wandb.sdk.data_types._private import _cleanup_media_tmp_dir
wandb/__init__.pyi CHANGED
@@ -79,7 +79,7 @@ from wandb.sdk.wandb_run import Run
79
79
  from wandb.sdk.wandb_setup import _WandbSetup
80
80
  from wandb.wandb_controller import _WandbController
81
81
 
82
- __version__: str = "0.18.0rc1"
82
+ __version__: str = "0.18.1"
83
83
 
84
84
  run: Optional[Run] = None
85
85
  config = wandb_config.Config
wandb/apis/public/runs.py CHANGED
@@ -562,6 +562,8 @@ class Run(Attrs):
562
562
  @property
563
563
  def json_config(self):
564
564
  config = {}
565
+ if "_wandb" in self.rawconfig:
566
+ config["_wandb"] = {"value": self.rawconfig["_wandb"], "desc": None}
565
567
  for k, v in self.config.items():
566
568
  config[k] = {"value": v, "desc": None}
567
569
  return json.dumps(config)
wandb/bin/nvidia_gpu_stats CHANGED
Binary file
wandb/cli/cli.py CHANGED
@@ -2928,7 +2928,6 @@ def disabled(service):
2928
2928
  try:
2929
2929
  api.set_setting("mode", "disabled", persist=True)
2930
2930
  click.echo("W&B disabled.")
2931
- os.environ[wandb.env._DISABLE_SERVICE] = str(service)
2932
2931
  except configparser.Error:
2933
2932
  click.echo(
2934
2933
  "Unable to write config, copy and paste the following in your terminal to turn off W&B:\nexport WANDB_MODE=disabled"
@@ -2948,7 +2947,6 @@ def enabled(service):
2948
2947
  try:
2949
2948
  api.set_setting("mode", "online", persist=True)
2950
2949
  click.echo("W&B enabled.")
2951
- os.environ[wandb.env._DISABLE_SERVICE] = str(not service)
2952
2950
  except configparser.Error:
2953
2951
  click.echo(
2954
2952
  "Unable to write config, copy and paste the following in your terminal to turn on W&B:\nexport WANDB_MODE=online"