wandb 0.17.4__py3-none-win32.whl → 0.17.6__py3-none-win32.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 (56) 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/filesync/upload_job.py +1 -1
  8. wandb/integration/lightning/fabric/logger.py +4 -4
  9. wandb/proto/v3/wandb_internal_pb2.py +339 -328
  10. wandb/proto/v3/wandb_settings_pb2.py +1 -1
  11. wandb/proto/v3/wandb_telemetry_pb2.py +10 -10
  12. wandb/proto/v4/wandb_internal_pb2.py +326 -323
  13. wandb/proto/v4/wandb_settings_pb2.py +1 -1
  14. wandb/proto/v4/wandb_telemetry_pb2.py +10 -10
  15. wandb/proto/v5/wandb_internal_pb2.py +326 -323
  16. wandb/proto/v5/wandb_settings_pb2.py +1 -1
  17. wandb/proto/v5/wandb_telemetry_pb2.py +10 -10
  18. wandb/proto/wandb_deprecated.py +4 -0
  19. wandb/proto/wandb_internal_pb2.py +6 -0
  20. wandb/sdk/artifacts/artifact.py +16 -24
  21. wandb/sdk/artifacts/artifact_manifest_entry.py +31 -0
  22. wandb/sdk/artifacts/storage_handlers/azure_handler.py +35 -23
  23. wandb/sdk/data_types/object_3d.py +113 -2
  24. wandb/sdk/interface/interface.py +35 -5
  25. wandb/sdk/interface/interface_shared.py +9 -7
  26. wandb/sdk/internal/handler.py +1 -1
  27. wandb/sdk/internal/internal_api.py +4 -4
  28. wandb/sdk/internal/sender.py +40 -17
  29. wandb/sdk/launch/_launch.py +4 -2
  30. wandb/sdk/launch/_project_spec.py +34 -8
  31. wandb/sdk/launch/agent/agent.py +6 -2
  32. wandb/sdk/launch/agent/run_queue_item_file_saver.py +2 -4
  33. wandb/sdk/launch/builder/build.py +4 -2
  34. wandb/sdk/launch/builder/kaniko_builder.py +30 -9
  35. wandb/sdk/launch/builder/templates/_wandb_bootstrap.py +2 -1
  36. wandb/sdk/launch/inputs/internal.py +93 -2
  37. wandb/sdk/launch/inputs/manage.py +21 -3
  38. wandb/sdk/launch/inputs/schema.py +39 -0
  39. wandb/sdk/launch/runner/kubernetes_runner.py +72 -0
  40. wandb/sdk/launch/runner/local_container.py +13 -10
  41. wandb/sdk/launch/runner/sagemaker_runner.py +3 -5
  42. wandb/sdk/launch/utils.py +2 -0
  43. wandb/sdk/lib/disabled.py +13 -174
  44. wandb/sdk/lib/tracelog.py +2 -2
  45. wandb/sdk/wandb_init.py +23 -27
  46. wandb/sdk/wandb_login.py +6 -6
  47. wandb/sdk/wandb_manager.py +9 -5
  48. wandb/sdk/wandb_run.py +141 -97
  49. wandb/sdk/wandb_settings.py +3 -2
  50. wandb/util.py +29 -11
  51. wandb/wandb_agent.py +2 -0
  52. {wandb-0.17.4.dist-info → wandb-0.17.6.dist-info}/METADATA +3 -2
  53. {wandb-0.17.4.dist-info → wandb-0.17.6.dist-info}/RECORD +56 -55
  54. {wandb-0.17.4.dist-info → wandb-0.17.6.dist-info}/WHEEL +0 -0
  55. {wandb-0.17.4.dist-info → wandb-0.17.6.dist-info}/entry_points.txt +0 -0
  56. {wandb-0.17.4.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.4"
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
 
@@ -110,7 +110,7 @@ class UploadJob:
110
110
  logger.info("Skipped uploading %s", self.save_path)
111
111
  self._stats.set_file_deduped(self.save_name)
112
112
  else:
113
- extra_headers = {}
113
+ extra_headers = self._api._extra_http_headers
114
114
  for upload_header in upload_headers:
115
115
  key, val = upload_header.split(":", 1)
116
116
  extra_headers[key] = val
@@ -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")