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.
- wandb/__init__.py +3 -1
- wandb/apis/public/api.py +1 -1
- wandb/apis/public/jobs.py +5 -0
- wandb/bin/wandb-core +0 -0
- wandb/data_types.py +2 -1
- wandb/env.py +6 -0
- wandb/filesync/upload_job.py +1 -1
- wandb/integration/lightning/fabric/logger.py +4 -4
- wandb/proto/v3/wandb_internal_pb2.py +339 -328
- wandb/proto/v3/wandb_settings_pb2.py +1 -1
- wandb/proto/v3/wandb_telemetry_pb2.py +10 -10
- wandb/proto/v4/wandb_internal_pb2.py +326 -323
- wandb/proto/v4/wandb_settings_pb2.py +1 -1
- wandb/proto/v4/wandb_telemetry_pb2.py +10 -10
- wandb/proto/v5/wandb_internal_pb2.py +326 -323
- wandb/proto/v5/wandb_settings_pb2.py +1 -1
- wandb/proto/v5/wandb_telemetry_pb2.py +10 -10
- wandb/proto/wandb_deprecated.py +4 -0
- wandb/proto/wandb_internal_pb2.py +6 -0
- wandb/sdk/artifacts/artifact.py +16 -24
- wandb/sdk/artifacts/artifact_manifest_entry.py +31 -0
- wandb/sdk/artifacts/storage_handlers/azure_handler.py +35 -23
- wandb/sdk/data_types/object_3d.py +113 -2
- wandb/sdk/interface/interface.py +35 -5
- wandb/sdk/interface/interface_shared.py +9 -7
- wandb/sdk/internal/handler.py +1 -1
- wandb/sdk/internal/internal_api.py +4 -4
- wandb/sdk/internal/sender.py +40 -17
- wandb/sdk/launch/_launch.py +4 -2
- wandb/sdk/launch/_project_spec.py +34 -8
- wandb/sdk/launch/agent/agent.py +6 -2
- wandb/sdk/launch/agent/run_queue_item_file_saver.py +2 -4
- wandb/sdk/launch/builder/build.py +4 -2
- wandb/sdk/launch/builder/kaniko_builder.py +30 -9
- wandb/sdk/launch/builder/templates/_wandb_bootstrap.py +2 -1
- wandb/sdk/launch/inputs/internal.py +93 -2
- wandb/sdk/launch/inputs/manage.py +21 -3
- wandb/sdk/launch/inputs/schema.py +39 -0
- wandb/sdk/launch/runner/kubernetes_runner.py +72 -0
- wandb/sdk/launch/runner/local_container.py +13 -10
- wandb/sdk/launch/runner/sagemaker_runner.py +3 -5
- wandb/sdk/launch/utils.py +2 -0
- wandb/sdk/lib/disabled.py +13 -174
- wandb/sdk/lib/tracelog.py +2 -2
- wandb/sdk/wandb_init.py +23 -27
- wandb/sdk/wandb_login.py +6 -6
- wandb/sdk/wandb_manager.py +9 -5
- wandb/sdk/wandb_run.py +141 -97
- wandb/sdk/wandb_settings.py +3 -2
- wandb/util.py +29 -11
- wandb/wandb_agent.py +2 -0
- {wandb-0.17.4.dist-info → wandb-0.17.6.dist-info}/METADATA +3 -2
- {wandb-0.17.4.dist-info → wandb-0.17.6.dist-info}/RECORD +56 -55
- {wandb-0.17.4.dist-info → wandb-0.17.6.dist-info}/WHEEL +0 -0
- {wandb-0.17.4.dist-info → wandb-0.17.6.dist-info}/entry_points.txt +0 -0
- {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.
|
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
|
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
|
|
wandb/filesync/upload_job.py
CHANGED
@@ -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
|
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:
|
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) ->
|
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,
|
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")
|