wandb 0.22.0__py3-none-win32.whl → 0.22.2__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 +1 -1
- wandb/__init__.pyi +8 -5
- wandb/_pydantic/__init__.py +12 -11
- wandb/_pydantic/base.py +49 -19
- wandb/apis/__init__.py +2 -0
- wandb/apis/attrs.py +2 -0
- wandb/apis/importers/internals/internal.py +16 -23
- wandb/apis/internal.py +2 -0
- wandb/apis/normalize.py +2 -0
- wandb/apis/public/__init__.py +3 -2
- wandb/apis/public/api.py +215 -164
- wandb/apis/public/artifacts.py +23 -20
- wandb/apis/public/const.py +2 -0
- wandb/apis/public/files.py +33 -24
- wandb/apis/public/history.py +2 -0
- wandb/apis/public/jobs.py +20 -18
- wandb/apis/public/projects.py +4 -2
- wandb/apis/public/query_generator.py +3 -0
- wandb/apis/public/registries/__init__.py +7 -0
- wandb/apis/public/registries/_freezable_list.py +9 -12
- wandb/apis/public/registries/registries_search.py +8 -6
- wandb/apis/public/registries/registry.py +22 -17
- wandb/apis/public/reports.py +2 -0
- wandb/apis/public/runs.py +261 -57
- wandb/apis/public/sweeps.py +10 -9
- wandb/apis/public/teams.py +2 -0
- wandb/apis/public/users.py +2 -0
- wandb/apis/public/utils.py +16 -15
- wandb/automations/_generated/__init__.py +54 -127
- wandb/automations/_generated/create_generic_webhook_integration.py +1 -7
- wandb/automations/_generated/fragments.py +26 -91
- wandb/bin/gpu_stats.exe +0 -0
- wandb/bin/wandb-core +0 -0
- wandb/cli/beta.py +16 -2
- wandb/cli/beta_leet.py +74 -0
- wandb/cli/beta_sync.py +9 -11
- wandb/cli/cli.py +34 -7
- wandb/errors/errors.py +3 -3
- wandb/proto/v3/wandb_api_pb2.py +86 -0
- wandb/proto/v3/wandb_internal_pb2.py +352 -351
- wandb/proto/v3/wandb_settings_pb2.py +2 -2
- wandb/proto/v3/wandb_sync_pb2.py +19 -6
- wandb/proto/v4/wandb_api_pb2.py +37 -0
- wandb/proto/v4/wandb_internal_pb2.py +352 -351
- wandb/proto/v4/wandb_settings_pb2.py +2 -2
- wandb/proto/v4/wandb_sync_pb2.py +10 -6
- wandb/proto/v5/wandb_api_pb2.py +38 -0
- wandb/proto/v5/wandb_internal_pb2.py +352 -351
- wandb/proto/v5/wandb_settings_pb2.py +2 -2
- wandb/proto/v5/wandb_sync_pb2.py +10 -6
- wandb/proto/v6/wandb_api_pb2.py +48 -0
- wandb/proto/v6/wandb_internal_pb2.py +352 -351
- wandb/proto/v6/wandb_settings_pb2.py +2 -2
- wandb/proto/v6/wandb_sync_pb2.py +10 -6
- wandb/proto/wandb_api_pb2.py +18 -0
- wandb/proto/wandb_generate_proto.py +1 -0
- wandb/sdk/artifacts/_factories.py +7 -2
- wandb/sdk/artifacts/_generated/__init__.py +112 -412
- wandb/sdk/artifacts/_generated/fragments.py +65 -0
- wandb/sdk/artifacts/_generated/operations.py +52 -22
- wandb/sdk/artifacts/_generated/run_input_artifacts.py +3 -23
- wandb/sdk/artifacts/_generated/run_output_artifacts.py +3 -23
- wandb/sdk/artifacts/_generated/type_info.py +19 -0
- wandb/sdk/artifacts/_gqlutils.py +47 -0
- wandb/sdk/artifacts/_models/__init__.py +4 -0
- wandb/sdk/artifacts/_models/base_model.py +20 -0
- wandb/sdk/artifacts/_validators.py +40 -12
- wandb/sdk/artifacts/artifact.py +99 -118
- wandb/sdk/artifacts/artifact_file_cache.py +6 -1
- wandb/sdk/artifacts/artifact_manifest_entry.py +67 -14
- wandb/sdk/artifacts/storage_handler.py +18 -12
- wandb/sdk/artifacts/storage_handlers/azure_handler.py +11 -6
- wandb/sdk/artifacts/storage_handlers/gcs_handler.py +9 -6
- wandb/sdk/artifacts/storage_handlers/http_handler.py +9 -4
- wandb/sdk/artifacts/storage_handlers/local_file_handler.py +10 -6
- wandb/sdk/artifacts/storage_handlers/multi_handler.py +5 -4
- wandb/sdk/artifacts/storage_handlers/s3_handler.py +10 -8
- wandb/sdk/artifacts/storage_handlers/tracking_handler.py +6 -4
- wandb/sdk/artifacts/storage_handlers/wb_artifact_handler.py +24 -21
- wandb/sdk/artifacts/storage_handlers/wb_local_artifact_handler.py +4 -2
- wandb/sdk/artifacts/storage_policies/_multipart.py +187 -0
- wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py +71 -242
- wandb/sdk/artifacts/storage_policy.py +25 -12
- wandb/sdk/data_types/bokeh.py +5 -1
- wandb/sdk/data_types/image.py +17 -6
- wandb/sdk/data_types/object_3d.py +67 -2
- wandb/sdk/interface/interface.py +31 -4
- wandb/sdk/interface/interface_queue.py +10 -0
- wandb/sdk/interface/interface_shared.py +0 -7
- wandb/sdk/interface/interface_sock.py +9 -3
- wandb/sdk/internal/_generated/__init__.py +2 -12
- wandb/sdk/internal/job_builder.py +27 -10
- wandb/sdk/internal/sender.py +5 -2
- wandb/sdk/internal/settings_static.py +2 -82
- wandb/sdk/launch/create_job.py +2 -1
- wandb/sdk/launch/runner/kubernetes_runner.py +25 -20
- wandb/sdk/launch/utils.py +82 -1
- wandb/sdk/lib/progress.py +8 -74
- wandb/sdk/lib/service/service_client.py +5 -9
- wandb/sdk/lib/service/service_connection.py +39 -23
- wandb/sdk/mailbox/mailbox_handle.py +2 -0
- wandb/sdk/projects/_generated/__init__.py +12 -33
- wandb/sdk/wandb_init.py +23 -3
- wandb/sdk/wandb_login.py +53 -27
- wandb/sdk/wandb_run.py +10 -5
- wandb/sdk/wandb_settings.py +63 -25
- wandb/sync/sync.py +7 -2
- wandb/util.py +1 -1
- {wandb-0.22.0.dist-info → wandb-0.22.2.dist-info}/METADATA +1 -1
- {wandb-0.22.0.dist-info → wandb-0.22.2.dist-info}/RECORD +113 -103
- wandb/sdk/artifacts/_graphql_fragments.py +0 -19
- {wandb-0.22.0.dist-info → wandb-0.22.2.dist-info}/WHEEL +0 -0
- {wandb-0.22.0.dist-info → wandb-0.22.2.dist-info}/entry_points.txt +0 -0
- {wandb-0.22.0.dist-info → wandb-0.22.2.dist-info}/licenses/LICENSE +0 -0
wandb/errors/errors.py
CHANGED
@@ -1,4 +1,4 @@
|
|
1
|
-
from
|
1
|
+
from __future__ import annotations
|
2
2
|
|
3
3
|
|
4
4
|
class Error(Exception):
|
@@ -7,7 +7,7 @@ class Error(Exception):
|
|
7
7
|
<!-- lazydoc-ignore-class: internal -->
|
8
8
|
"""
|
9
9
|
|
10
|
-
def __init__(self, message, context:
|
10
|
+
def __init__(self, message: str, context: dict | None = None) -> None:
|
11
11
|
super().__init__(message)
|
12
12
|
self.message = message
|
13
13
|
# sentry context capture
|
@@ -18,7 +18,7 @@ class Error(Exception):
|
|
18
18
|
class CommError(Error):
|
19
19
|
"""Error communicating with W&B servers."""
|
20
20
|
|
21
|
-
def __init__(self, msg, exc=None) -> None:
|
21
|
+
def __init__(self, msg: str, exc: Exception | None = None) -> None:
|
22
22
|
self.exc = exc
|
23
23
|
self.message = msg
|
24
24
|
super().__init__(self.message)
|
@@ -0,0 +1,86 @@
|
|
1
|
+
# -*- coding: utf-8 -*-
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
3
|
+
# source: wandb/proto/wandb_api.proto
|
4
|
+
"""Generated protocol buffer code."""
|
5
|
+
from google.protobuf import descriptor as _descriptor
|
6
|
+
from google.protobuf import descriptor_pool as _descriptor_pool
|
7
|
+
from google.protobuf import message as _message
|
8
|
+
from google.protobuf import reflection as _reflection
|
9
|
+
from google.protobuf import symbol_database as _symbol_database
|
10
|
+
# @@protoc_insertion_point(imports)
|
11
|
+
|
12
|
+
_sym_db = _symbol_database.Default()
|
13
|
+
|
14
|
+
|
15
|
+
from wandb.proto import wandb_base_pb2 as wandb_dot_proto_dot_wandb__base__pb2
|
16
|
+
|
17
|
+
|
18
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1bwandb/proto/wandb_api.proto\x12\x0ewandb_internal\x1a\x1cwandb/proto/wandb_base.proto\"]\n\nApiRequest\x12\x44\n\x10read_run_history\x18\x01 \x01(\x0b\x32(.wandb_internal.ReadRunHistoryApiRequestH\x00\x42\t\n\x07request\"`\n\x0b\x41piResponse\x12\x45\n\x10read_run_history\x18\x01 \x01(\x0b\x32).wandb_internal.ReadRunHistoryApiResponseH\x00\x42\n\n\x08response\"\xaa\x01\n\x18ReadRunHistoryApiRequest\x12\x0e\n\x06\x65ntity\x18\x01 \x01(\t\x12\x0f\n\x07project\x18\x02 \x01(\t\x12\x0e\n\x06run_id\x18\x03 \x01(\t\x12\x0c\n\x04keys\x18\x04 \x03(\t\x12\x10\n\x08min_step\x18\x05 \x01(\x03\x12\x10\n\x08max_step\x18\x06 \x01(\x03\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"d\n\x19ReadRunHistoryApiResponse\x12\x30\n\x0chistory_rows\x18\x01 \x03(\x0b\x32\x1a.wandb_internal.HistoryRow\x12\x15\n\rerror_message\x18\x02 \x01(\t\"G\n\nHistoryRow\x12\x39\n\rhistory_items\x18\x01 \x03(\x0b\x32\".wandb_internal.ParquetHistoryItem\"5\n\x12ParquetHistoryItem\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x12\n\nvalue_json\x18\x10 \x01(\tB\x1bZ\x19\x63ore/pkg/service_go_protob\x06proto3')
|
19
|
+
|
20
|
+
|
21
|
+
|
22
|
+
_APIREQUEST = DESCRIPTOR.message_types_by_name['ApiRequest']
|
23
|
+
_APIRESPONSE = DESCRIPTOR.message_types_by_name['ApiResponse']
|
24
|
+
_READRUNHISTORYAPIREQUEST = DESCRIPTOR.message_types_by_name['ReadRunHistoryApiRequest']
|
25
|
+
_READRUNHISTORYAPIRESPONSE = DESCRIPTOR.message_types_by_name['ReadRunHistoryApiResponse']
|
26
|
+
_HISTORYROW = DESCRIPTOR.message_types_by_name['HistoryRow']
|
27
|
+
_PARQUETHISTORYITEM = DESCRIPTOR.message_types_by_name['ParquetHistoryItem']
|
28
|
+
ApiRequest = _reflection.GeneratedProtocolMessageType('ApiRequest', (_message.Message,), {
|
29
|
+
'DESCRIPTOR' : _APIREQUEST,
|
30
|
+
'__module__' : 'wandb.proto.wandb_api_pb2'
|
31
|
+
# @@protoc_insertion_point(class_scope:wandb_internal.ApiRequest)
|
32
|
+
})
|
33
|
+
_sym_db.RegisterMessage(ApiRequest)
|
34
|
+
|
35
|
+
ApiResponse = _reflection.GeneratedProtocolMessageType('ApiResponse', (_message.Message,), {
|
36
|
+
'DESCRIPTOR' : _APIRESPONSE,
|
37
|
+
'__module__' : 'wandb.proto.wandb_api_pb2'
|
38
|
+
# @@protoc_insertion_point(class_scope:wandb_internal.ApiResponse)
|
39
|
+
})
|
40
|
+
_sym_db.RegisterMessage(ApiResponse)
|
41
|
+
|
42
|
+
ReadRunHistoryApiRequest = _reflection.GeneratedProtocolMessageType('ReadRunHistoryApiRequest', (_message.Message,), {
|
43
|
+
'DESCRIPTOR' : _READRUNHISTORYAPIREQUEST,
|
44
|
+
'__module__' : 'wandb.proto.wandb_api_pb2'
|
45
|
+
# @@protoc_insertion_point(class_scope:wandb_internal.ReadRunHistoryApiRequest)
|
46
|
+
})
|
47
|
+
_sym_db.RegisterMessage(ReadRunHistoryApiRequest)
|
48
|
+
|
49
|
+
ReadRunHistoryApiResponse = _reflection.GeneratedProtocolMessageType('ReadRunHistoryApiResponse', (_message.Message,), {
|
50
|
+
'DESCRIPTOR' : _READRUNHISTORYAPIRESPONSE,
|
51
|
+
'__module__' : 'wandb.proto.wandb_api_pb2'
|
52
|
+
# @@protoc_insertion_point(class_scope:wandb_internal.ReadRunHistoryApiResponse)
|
53
|
+
})
|
54
|
+
_sym_db.RegisterMessage(ReadRunHistoryApiResponse)
|
55
|
+
|
56
|
+
HistoryRow = _reflection.GeneratedProtocolMessageType('HistoryRow', (_message.Message,), {
|
57
|
+
'DESCRIPTOR' : _HISTORYROW,
|
58
|
+
'__module__' : 'wandb.proto.wandb_api_pb2'
|
59
|
+
# @@protoc_insertion_point(class_scope:wandb_internal.HistoryRow)
|
60
|
+
})
|
61
|
+
_sym_db.RegisterMessage(HistoryRow)
|
62
|
+
|
63
|
+
ParquetHistoryItem = _reflection.GeneratedProtocolMessageType('ParquetHistoryItem', (_message.Message,), {
|
64
|
+
'DESCRIPTOR' : _PARQUETHISTORYITEM,
|
65
|
+
'__module__' : 'wandb.proto.wandb_api_pb2'
|
66
|
+
# @@protoc_insertion_point(class_scope:wandb_internal.ParquetHistoryItem)
|
67
|
+
})
|
68
|
+
_sym_db.RegisterMessage(ParquetHistoryItem)
|
69
|
+
|
70
|
+
if _descriptor._USE_C_DESCRIPTORS == False:
|
71
|
+
|
72
|
+
DESCRIPTOR._options = None
|
73
|
+
DESCRIPTOR._serialized_options = b'Z\031core/pkg/service_go_proto'
|
74
|
+
_APIREQUEST._serialized_start=77
|
75
|
+
_APIREQUEST._serialized_end=170
|
76
|
+
_APIRESPONSE._serialized_start=172
|
77
|
+
_APIRESPONSE._serialized_end=268
|
78
|
+
_READRUNHISTORYAPIREQUEST._serialized_start=271
|
79
|
+
_READRUNHISTORYAPIREQUEST._serialized_end=441
|
80
|
+
_READRUNHISTORYAPIRESPONSE._serialized_start=443
|
81
|
+
_READRUNHISTORYAPIRESPONSE._serialized_end=543
|
82
|
+
_HISTORYROW._serialized_start=545
|
83
|
+
_HISTORYROW._serialized_end=616
|
84
|
+
_PARQUETHISTORYITEM._serialized_start=618
|
85
|
+
_PARQUETHISTORYITEM._serialized_end=671
|
86
|
+
# @@protoc_insertion_point(module_scope)
|