wandb 0.17.5__py3-none-win_amd64.whl → 0.17.6__py3-none-win_amd64.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (47) hide show
  1. wandb/__init__.py +3 -1
  2. wandb/apis/public/api.py +1 -1
  3. wandb/apis/public/jobs.py +5 -0
  4. wandb/bin/wandb-core +0 -0
  5. wandb/data_types.py +2 -1
  6. wandb/env.py +6 -0
  7. wandb/integration/lightning/fabric/logger.py +4 -4
  8. wandb/proto/v3/wandb_internal_pb2.py +226 -226
  9. wandb/proto/v3/wandb_settings_pb2.py +1 -1
  10. wandb/proto/v3/wandb_telemetry_pb2.py +10 -10
  11. wandb/proto/v4/wandb_internal_pb2.py +226 -226
  12. wandb/proto/v4/wandb_settings_pb2.py +1 -1
  13. wandb/proto/v4/wandb_telemetry_pb2.py +10 -10
  14. wandb/proto/v5/wandb_internal_pb2.py +226 -226
  15. wandb/proto/v5/wandb_settings_pb2.py +1 -1
  16. wandb/proto/v5/wandb_telemetry_pb2.py +10 -10
  17. wandb/proto/wandb_deprecated.py +4 -0
  18. wandb/proto/wandb_internal_pb2.py +6 -0
  19. wandb/sdk/artifacts/artifact.py +5 -0
  20. wandb/sdk/artifacts/artifact_manifest_entry.py +31 -0
  21. wandb/sdk/artifacts/storage_handlers/azure_handler.py +35 -23
  22. wandb/sdk/data_types/object_3d.py +113 -2
  23. wandb/sdk/interface/interface.py +23 -0
  24. wandb/sdk/internal/sender.py +31 -15
  25. wandb/sdk/launch/_launch.py +4 -2
  26. wandb/sdk/launch/_project_spec.py +34 -8
  27. wandb/sdk/launch/agent/agent.py +6 -2
  28. wandb/sdk/launch/agent/run_queue_item_file_saver.py +2 -4
  29. wandb/sdk/launch/builder/build.py +4 -2
  30. wandb/sdk/launch/builder/templates/_wandb_bootstrap.py +2 -1
  31. wandb/sdk/launch/inputs/internal.py +42 -28
  32. wandb/sdk/launch/inputs/schema.py +39 -0
  33. wandb/sdk/launch/runner/kubernetes_runner.py +72 -0
  34. wandb/sdk/launch/runner/local_container.py +13 -10
  35. wandb/sdk/launch/runner/sagemaker_runner.py +3 -5
  36. wandb/sdk/launch/utils.py +2 -0
  37. wandb/sdk/lib/disabled.py +13 -174
  38. wandb/sdk/wandb_init.py +23 -27
  39. wandb/sdk/wandb_login.py +6 -6
  40. wandb/sdk/wandb_run.py +41 -22
  41. wandb/sdk/wandb_settings.py +3 -2
  42. wandb/wandb_agent.py +2 -0
  43. {wandb-0.17.5.dist-info → wandb-0.17.6.dist-info}/METADATA +3 -2
  44. {wandb-0.17.5.dist-info → wandb-0.17.6.dist-info}/RECORD +47 -46
  45. {wandb-0.17.5.dist-info → wandb-0.17.6.dist-info}/WHEEL +0 -0
  46. {wandb-0.17.5.dist-info → wandb-0.17.6.dist-info}/entry_points.txt +0 -0
  47. {wandb-0.17.5.dist-info → wandb-0.17.6.dist-info}/licenses/LICENSE +0 -0
wandb/__init__.py CHANGED
@@ -11,7 +11,7 @@ For scripts and interactive notebooks, see https://github.com/wandb/examples.
11
11
 
12
12
  For reference documentation, see https://docs.wandb.com/ref/python.
13
13
  """
14
- __version__ = "0.17.5"
14
+ __version__ = "0.17.6"
15
15
 
16
16
 
17
17
  # Used with pypi checks and other messages related to pip
@@ -74,6 +74,7 @@ from wandb.data_types import Video
74
74
  from wandb.data_types import Audio
75
75
  from wandb.data_types import Table
76
76
  from wandb.data_types import Html
77
+ from wandb.data_types import box3d
77
78
  from wandb.data_types import Object3D
78
79
  from wandb.data_types import Molecule
79
80
  from wandb.data_types import Histogram
@@ -245,6 +246,7 @@ __all__ = (
245
246
  "Audio",
246
247
  "Table",
247
248
  "Html",
249
+ "box3d",
248
250
  "Object3D",
249
251
  "Molecule",
250
252
  "Histogram",
wandb/apis/public/api.py CHANGED
@@ -821,7 +821,7 @@ class Api:
821
821
  order: (str) Order can be `created_at`, `heartbeat_at`, `config.*.value`, or `summary_metrics.*`.
822
822
  If you prepend order with a + order is ascending.
823
823
  If you prepend order with a - order is descending (default).
824
- The default order is run.created_at from newest to oldest.
824
+ The default order is run.created_at from oldest to newest.
825
825
  per_page: (int) Sets the page size for query pagination.
826
826
  include_sweeps: (bool) Whether to include the sweep runs in the results.
827
827
 
wandb/apis/public/jobs.py CHANGED
@@ -65,6 +65,7 @@ class Job:
65
65
  self._entrypoint = source_info.get("entrypoint")
66
66
  self._dockerfile = source_info.get("dockerfile")
67
67
  self._build_context = source_info.get("build_context")
68
+ self._base_image = source_info.get("base_image")
68
69
  self._args = source_info.get("args")
69
70
  self._partial = self._job_info.get("_partial", False)
70
71
  self._requirements_file = os.path.join(self._fpath, "requirements.frozen.txt")
@@ -131,6 +132,8 @@ class Job:
131
132
  launch_project.set_job_dockerfile(self._dockerfile)
132
133
  if self._build_context:
133
134
  launch_project.set_job_build_context(self._build_context)
135
+ if self._base_image:
136
+ launch_project.set_job_base_image(self._base_image)
134
137
 
135
138
  def _configure_launch_project_artifact(self, launch_project):
136
139
  artifact_string = self._job_info.get("source", {}).get("artifact")
@@ -152,6 +155,8 @@ class Job:
152
155
  launch_project.set_job_dockerfile(self._dockerfile)
153
156
  if self._build_context:
154
157
  launch_project.set_job_build_context(self._build_context)
158
+ if self._base_image:
159
+ launch_project.set_job_base_image(self._base_image)
155
160
 
156
161
  def _configure_launch_project_container(self, launch_project):
157
162
  launch_project.docker_image = self._job_info.get("source", {}).get("image")
wandb/bin/wandb-core CHANGED
Binary file
wandb/data_types.py CHANGED
@@ -44,7 +44,7 @@ from .sdk.data_types.histogram import Histogram
44
44
  from .sdk.data_types.html import Html
45
45
  from .sdk.data_types.image import Image
46
46
  from .sdk.data_types.molecule import Molecule
47
- from .sdk.data_types.object_3d import Object3D
47
+ from .sdk.data_types.object_3d import Object3D, box3d
48
48
  from .sdk.data_types.plotly import Plotly
49
49
  from .sdk.data_types.saved_model import _SavedModel
50
50
  from .sdk.data_types.trace_tree import WBTraceTree
@@ -65,6 +65,7 @@ __all__ = [
65
65
  "Html",
66
66
  "Image",
67
67
  "Molecule",
68
+ "box3d",
68
69
  "Object3D",
69
70
  "Plotly",
70
71
  "Video",
wandb/env.py CHANGED
@@ -161,6 +161,12 @@ def is_debug(default: Optional[str] = None, env: Optional[Env] = None) -> bool:
161
161
  return _env_as_bool(DEBUG, default=default, env=env)
162
162
 
163
163
 
164
+ def is_offline(env: Optional[Env] = None) -> bool:
165
+ if env is None:
166
+ env = os.environ
167
+ return env.get(MODE) == "offline"
168
+
169
+
164
170
  def error_reporting_enabled() -> bool:
165
171
  return _env_as_bool(ERROR_REPORTING, default="True")
166
172
 
@@ -8,7 +8,7 @@ from typing_extensions import override
8
8
 
9
9
  import wandb
10
10
  from wandb import Artifact
11
- from wandb.sdk.lib import RunDisabled, telemetry
11
+ from wandb.sdk.lib import telemetry
12
12
  from wandb.sdk.wandb_run import Run
13
13
 
14
14
  try:
@@ -293,7 +293,7 @@ class WandbLogger(Logger):
293
293
  anonymous: Optional[bool] = None,
294
294
  project: Optional[str] = None,
295
295
  log_model: Union[Literal["all"], bool] = False,
296
- experiment: Union["Run", "RunDisabled", None] = None,
296
+ experiment: Optional["Run"] = None,
297
297
  prefix: str = "",
298
298
  checkpoint_name: Optional[str] = None,
299
299
  log_checkpoint_on: Union[Literal["success"], Literal["all"]] = "success",
@@ -360,7 +360,7 @@ class WandbLogger(Logger):
360
360
 
361
361
  @property
362
362
  @rank_zero_experiment
363
- def experiment(self) -> Union["Run", "RunDisabled"]:
363
+ def experiment(self) -> "Run":
364
364
  r"""Actual wandb object.
365
365
 
366
366
  To use wandb features in your :class:`~lightning.pytorch.core.LightningModule`, do the
@@ -393,7 +393,7 @@ class WandbLogger(Logger):
393
393
  self._experiment = wandb.init(**self._wandb_init)
394
394
 
395
395
  # define default x-axis
396
- if isinstance(self._experiment, (Run, RunDisabled)) and getattr(
396
+ if isinstance(self._experiment, Run) and getattr(
397
397
  self._experiment, "define_metric", None
398
398
  ):
399
399
  self._experiment.define_metric("trainer/global_step")