wandb 0.18.6__py3-none-win32.whl → 0.19.0rc1__py3-none-win32.whl
Sign up to get free protection for your applications and to get access to all the features.
- package_readme.md +8 -0
- wandb/__init__.py +5 -7
- wandb/__init__.pyi +51 -30
- wandb/analytics/sentry.py +4 -10
- wandb/apis/importers/internals/internal.py +6 -6
- wandb/apis/importers/internals/protocols.py +11 -7
- wandb/apis/public/api.py +5 -1
- wandb/apis/public/jobs.py +1 -7
- wandb/apis/public/reports.py +6 -17
- wandb/apis/public/runs.py +12 -10
- wandb/bin/gpu_stats.exe +0 -0
- wandb/bin/wandb-core +0 -0
- wandb/cli/cli.py +9 -45
- wandb/env.py +3 -5
- wandb/errors/links.py +1 -1
- wandb/errors/term.py +1 -6
- wandb/filesync/dir_watcher.py +3 -3
- wandb/filesync/step_upload.py +2 -5
- wandb/integration/fastai/__init__.py +1 -6
- wandb/integration/gym/__init__.py +1 -7
- wandb/integration/keras/callbacks/metrics_logger.py +1 -8
- wandb/integration/keras/callbacks/model_checkpoint.py +1 -8
- wandb/integration/keras/keras.py +3 -5
- wandb/integration/lightgbm/__init__.py +1 -1
- wandb/integration/sb3/sb3.py +1 -7
- wandb/integration/sklearn/utils.py +1 -1
- wandb/integration/tensorboard/log.py +1 -2
- wandb/integration/torch/wandb_torch.py +1 -1
- wandb/integration/ultralytics/bbox_utils.py +9 -2
- wandb/jupyter.py +4 -4
- wandb/proto/v3/wandb_internal_pb2.py +31 -31
- wandb/proto/v3/wandb_settings_pb2.py +2 -2
- wandb/proto/v3/wandb_telemetry_pb2.py +4 -4
- wandb/proto/v4/wandb_internal_pb2.py +31 -31
- wandb/proto/v4/wandb_settings_pb2.py +2 -2
- wandb/proto/v4/wandb_telemetry_pb2.py +4 -4
- wandb/proto/v5/wandb_internal_pb2.py +31 -31
- wandb/proto/v5/wandb_settings_pb2.py +2 -2
- wandb/proto/v5/wandb_telemetry_pb2.py +4 -4
- wandb/proto/wandb_deprecated.py +3 -11
- wandb/proto/wandb_generate_deprecated.py +3 -7
- wandb/sdk/artifacts/artifact.py +3 -11
- wandb/sdk/artifacts/artifact_file_cache.py +2 -5
- wandb/sdk/artifacts/artifact_saver.py +2 -6
- wandb/sdk/artifacts/storage_handlers/gcs_handler.py +2 -4
- wandb/sdk/artifacts/storage_handlers/local_file_handler.py +2 -4
- wandb/sdk/artifacts/storage_handlers/s3_handler.py +2 -4
- wandb/sdk/backend/backend.py +1 -1
- wandb/sdk/data_types/base_types/wb_value.py +20 -10
- wandb/sdk/data_types/histogram.py +1 -3
- wandb/sdk/data_types/object_3d.py +2 -6
- wandb/sdk/data_types/table.py +1 -1
- wandb/sdk/data_types/utils.py +1 -2
- wandb/sdk/data_types/video.py +15 -4
- wandb/sdk/integration_utils/auto_logging.py +1 -8
- wandb/sdk/interface/interface.py +12 -5
- wandb/sdk/interface/interface_queue.py +0 -6
- wandb/sdk/interface/interface_shared.py +9 -0
- wandb/sdk/interface/router.py +1 -2
- wandb/sdk/interface/router_queue.py +0 -3
- wandb/sdk/interface/router_relay.py +0 -2
- wandb/sdk/internal/file_stream.py +1 -4
- wandb/sdk/internal/flow_control.py +1 -1
- wandb/sdk/internal/handler.py +8 -5
- wandb/sdk/internal/internal.py +3 -17
- wandb/sdk/internal/internal_api.py +3 -10
- wandb/sdk/internal/internal_util.py +0 -3
- wandb/sdk/internal/job_builder.py +20 -12
- wandb/sdk/internal/progress.py +1 -5
- wandb/sdk/internal/sender.py +9 -15
- wandb/sdk/internal/settings_static.py +4 -10
- wandb/sdk/internal/system/assets/cpu.py +2 -2
- wandb/sdk/internal/system/assets/disk.py +3 -3
- wandb/sdk/internal/system/assets/gpu.py +7 -7
- wandb/sdk/internal/system/assets/gpu_amd.py +1 -7
- wandb/sdk/internal/system/assets/interfaces.py +11 -13
- wandb/sdk/internal/system/assets/ipu.py +1 -1
- wandb/sdk/internal/system/assets/memory.py +2 -2
- wandb/sdk/internal/system/assets/open_metrics.py +2 -8
- wandb/sdk/internal/system/assets/trainium.py +3 -9
- wandb/sdk/internal/system/system_info.py +14 -13
- wandb/sdk/internal/system/system_monitor.py +5 -12
- wandb/sdk/internal/tb_watcher.py +1 -1
- wandb/sdk/internal/writer.py +2 -4
- wandb/sdk/launch/__init__.py +2 -1
- wandb/sdk/launch/agent/run_queue_item_file_saver.py +1 -7
- wandb/sdk/launch/create_job.py +2 -3
- wandb/sdk/launch/runner/abstract.py +1 -6
- wandb/sdk/launch/runner/kubernetes_monitor.py +2 -4
- wandb/sdk/lib/apikey.py +2 -6
- wandb/sdk/lib/fsm.py +12 -6
- wandb/sdk/lib/ipython.py +1 -6
- wandb/sdk/lib/module.py +0 -3
- wandb/sdk/lib/progress.py +2 -3
- wandb/sdk/lib/run_moment.py +1 -7
- wandb/sdk/lib/server.py +10 -24
- wandb/sdk/lib/sock_client.py +0 -5
- wandb/sdk/service/server.py +3 -12
- wandb/sdk/service/server_sock.py +0 -2
- wandb/sdk/service/service.py +5 -5
- wandb/sdk/wandb_init.py +215 -166
- wandb/sdk/wandb_login.py +17 -27
- wandb/sdk/wandb_run.py +129 -161
- wandb/sdk/wandb_settings.py +978 -1760
- wandb/sdk/wandb_setup.py +87 -94
- wandb/sdk/wandb_watch.py +1 -1
- wandb/sync/sync.py +1 -2
- wandb/util.py +7 -40
- wandb/wandb_controller.py +10 -12
- {wandb-0.18.6.dist-info → wandb-0.19.0rc1.dist-info}/METADATA +14 -4
- {wandb-0.18.6.dist-info → wandb-0.19.0rc1.dist-info}/RECORD +114 -120
- {wandb-0.18.6.dist-info → wandb-0.19.0rc1.dist-info}/WHEEL +1 -1
- wandb/integration/magic.py +0 -556
- wandb/magic.py +0 -3
- wandb/sdk/lib/_settings_toposort_generate.py +0 -159
- wandb/sdk/lib/_settings_toposort_generated.py +0 -250
- wandb/sdk/lib/reporting.py +0 -99
- wandb/sdk/lib/tracelog.py +0 -255
- {wandb-0.18.6.dist-info → wandb-0.19.0rc1.dist-info}/entry_points.txt +0 -0
- {wandb-0.18.6.dist-info → wandb-0.19.0rc1.dist-info}/licenses/LICENSE +0 -0
wandb/sdk/wandb_login.py
CHANGED
@@ -5,13 +5,7 @@ This authenticates your machine to log data to your account.
|
|
5
5
|
|
6
6
|
import enum
|
7
7
|
import os
|
8
|
-
import
|
9
|
-
from typing import Optional, Tuple
|
10
|
-
|
11
|
-
if sys.version_info >= (3, 8):
|
12
|
-
from typing import Literal
|
13
|
-
else:
|
14
|
-
from typing_extensions import Literal
|
8
|
+
from typing import Literal, Optional, Tuple
|
15
9
|
|
16
10
|
import click
|
17
11
|
|
@@ -124,7 +118,7 @@ class _WandbLogin:
|
|
124
118
|
def setup(
|
125
119
|
self,
|
126
120
|
*,
|
127
|
-
anonymous: Optional[Literal["
|
121
|
+
anonymous: Optional[Literal["allow", "must", "never"]] = None,
|
128
122
|
key: Optional[str] = None,
|
129
123
|
relogin: Optional[bool] = None,
|
130
124
|
host: Optional[str] = None,
|
@@ -134,23 +128,20 @@ class _WandbLogin:
|
|
134
128
|
"""Updates login-related settings on the global setup object."""
|
135
129
|
self._relogin = relogin
|
136
130
|
|
137
|
-
|
138
|
-
|
139
|
-
|
140
|
-
|
141
|
-
|
142
|
-
|
143
|
-
|
144
|
-
"key": key,
|
145
|
-
"host": host,
|
146
|
-
"force": force,
|
147
|
-
"timeout": timeout,
|
148
|
-
},
|
149
|
-
_logger=logger,
|
150
|
-
)
|
131
|
+
login_settings = {
|
132
|
+
"anonymous": anonymous,
|
133
|
+
"api_key": key,
|
134
|
+
"base_url": host,
|
135
|
+
"force": force,
|
136
|
+
"login_timeout": timeout,
|
137
|
+
}
|
151
138
|
|
152
139
|
# make sure they are applied globally
|
153
|
-
self._wl = wandb.setup(
|
140
|
+
self._wl = wandb.setup(
|
141
|
+
settings=wandb.Settings(
|
142
|
+
**{k: v for k, v in login_settings.items() if v is not None}
|
143
|
+
)
|
144
|
+
)
|
154
145
|
self._settings = self._wl.settings
|
155
146
|
|
156
147
|
def is_apikey_configured(self) -> bool:
|
@@ -232,7 +223,6 @@ class _WandbLogin:
|
|
232
223
|
|
233
224
|
If we're online, this also pulls in user settings from the server.
|
234
225
|
"""
|
235
|
-
_logger = wandb.setup()._get_logger()
|
236
226
|
login_settings = dict()
|
237
227
|
if status == ApiKeyStatus.OFFLINE:
|
238
228
|
login_settings = dict(mode="offline")
|
@@ -240,7 +230,7 @@ class _WandbLogin:
|
|
240
230
|
login_settings = dict(mode="disabled")
|
241
231
|
elif key:
|
242
232
|
login_settings = dict(api_key=key)
|
243
|
-
self._wl._settings.
|
233
|
+
self._wl._settings.update_from_dict(login_settings)
|
244
234
|
# Whenever the key changes, make sure to pull in user settings
|
245
235
|
# from server.
|
246
236
|
if not self._wl.settings._offline:
|
@@ -275,7 +265,7 @@ class _WandbLogin:
|
|
275
265
|
if status == ApiKeyStatus.NOTTY:
|
276
266
|
directive = (
|
277
267
|
"wandb login [your_api_key]"
|
278
|
-
if self._settings.
|
268
|
+
if self._settings.x_cli_only_mode
|
279
269
|
else "wandb.login(key=[your_api_key])"
|
280
270
|
)
|
281
271
|
raise UsageError("api_key not configured (no-tty). call " + directive)
|
@@ -286,7 +276,7 @@ class _WandbLogin:
|
|
286
276
|
|
287
277
|
def _login(
|
288
278
|
*,
|
289
|
-
anonymous: Optional[Literal["
|
279
|
+
anonymous: Optional[Literal["allow", "must", "never"]] = None,
|
290
280
|
key: Optional[str] = None,
|
291
281
|
relogin: Optional[bool] = None,
|
292
282
|
host: Optional[str] = None,
|