wandb 0.17.0rc2__py3-none-macosx_11_0_arm64.whl → 0.17.2__py3-none-macosx_11_0_arm64.whl
Sign up to get free protection for your applications and to get access to all the features.
- wandb/__init__.py +4 -2
- wandb/apis/importers/internals/internal.py +0 -1
- wandb/apis/importers/wandb.py +12 -7
- wandb/apis/internal.py +0 -3
- wandb/apis/public/api.py +213 -79
- wandb/apis/public/artifacts.py +335 -100
- wandb/apis/public/files.py +9 -9
- wandb/apis/public/jobs.py +16 -4
- wandb/apis/public/projects.py +26 -28
- wandb/apis/public/query_generator.py +1 -1
- wandb/apis/public/runs.py +163 -65
- wandb/apis/public/sweeps.py +2 -2
- wandb/apis/reports/__init__.py +1 -7
- wandb/apis/reports/v1/__init__.py +5 -27
- wandb/apis/reports/v2/__init__.py +7 -19
- wandb/apis/workspaces/__init__.py +8 -0
- wandb/beta/workflows.py +8 -3
- wandb/bin/apple_gpu_stats +0 -0
- wandb/bin/wandb-core +0 -0
- wandb/cli/cli.py +151 -59
- wandb/docker/__init__.py +1 -1
- wandb/errors/term.py +10 -2
- wandb/filesync/step_checksum.py +1 -4
- wandb/filesync/step_prepare.py +4 -24
- wandb/filesync/step_upload.py +5 -107
- wandb/filesync/upload_job.py +0 -76
- wandb/integration/gym/__init__.py +35 -15
- wandb/integration/openai/fine_tuning.py +21 -3
- wandb/integration/prodigy/prodigy.py +1 -1
- wandb/jupyter.py +16 -17
- wandb/old/summary.py +5 -0
- wandb/plot/pr_curve.py +2 -1
- wandb/plot/roc_curve.py +2 -1
- wandb/{plots → plot}/utils.py +13 -25
- wandb/proto/v3/wandb_internal_pb2.py +54 -54
- wandb/proto/v3/wandb_settings_pb2.py +2 -2
- wandb/proto/v3/wandb_telemetry_pb2.py +10 -10
- wandb/proto/v4/wandb_internal_pb2.py +54 -54
- wandb/proto/v4/wandb_settings_pb2.py +2 -2
- wandb/proto/v4/wandb_telemetry_pb2.py +10 -10
- wandb/proto/v5/wandb_base_pb2.py +30 -0
- wandb/proto/v5/wandb_internal_pb2.py +355 -0
- wandb/proto/v5/wandb_server_pb2.py +63 -0
- wandb/proto/v5/wandb_settings_pb2.py +45 -0
- wandb/proto/v5/wandb_telemetry_pb2.py +41 -0
- wandb/proto/wandb_base_pb2.py +2 -0
- wandb/proto/wandb_deprecated.py +9 -1
- wandb/proto/wandb_generate_deprecated.py +34 -0
- wandb/proto/{wandb_internal_codegen.py → wandb_generate_proto.py} +1 -35
- wandb/proto/wandb_internal_pb2.py +2 -0
- wandb/proto/wandb_server_pb2.py +2 -0
- wandb/proto/wandb_settings_pb2.py +2 -0
- wandb/proto/wandb_telemetry_pb2.py +2 -0
- wandb/sdk/artifacts/artifact.py +76 -23
- wandb/sdk/artifacts/artifact_manifest.py +1 -1
- wandb/sdk/artifacts/artifact_manifest_entry.py +6 -3
- wandb/sdk/artifacts/artifact_manifests/artifact_manifest_v1.py +1 -1
- wandb/sdk/artifacts/artifact_saver.py +1 -10
- wandb/sdk/artifacts/storage_handlers/local_file_handler.py +6 -2
- wandb/sdk/artifacts/storage_handlers/multi_handler.py +1 -1
- wandb/sdk/artifacts/storage_handlers/tracking_handler.py +6 -4
- wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py +2 -42
- wandb/sdk/artifacts/storage_policy.py +1 -12
- wandb/sdk/data_types/_dtypes.py +5 -2
- wandb/sdk/data_types/html.py +1 -1
- wandb/sdk/data_types/image.py +1 -1
- wandb/sdk/data_types/object_3d.py +1 -1
- wandb/sdk/data_types/video.py +4 -2
- wandb/sdk/interface/interface.py +13 -0
- wandb/sdk/interface/interface_shared.py +1 -1
- wandb/sdk/internal/file_pusher.py +2 -5
- wandb/sdk/internal/file_stream.py +6 -19
- wandb/sdk/internal/internal_api.py +160 -138
- wandb/sdk/internal/job_builder.py +207 -135
- wandb/sdk/internal/progress.py +0 -28
- wandb/sdk/internal/sender.py +105 -42
- wandb/sdk/internal/settings_static.py +8 -1
- wandb/sdk/internal/system/assets/gpu.py +2 -0
- wandb/sdk/internal/system/assets/trainium.py +3 -3
- wandb/sdk/internal/system/system_info.py +4 -2
- wandb/sdk/internal/update.py +1 -1
- wandb/sdk/launch/__init__.py +9 -1
- wandb/sdk/launch/_launch.py +4 -24
- wandb/sdk/launch/_launch_add.py +1 -3
- wandb/sdk/launch/_project_spec.py +184 -224
- wandb/sdk/launch/agent/agent.py +58 -18
- wandb/sdk/launch/agent/config.py +0 -3
- wandb/sdk/launch/builder/abstract.py +67 -0
- wandb/sdk/launch/builder/build.py +165 -576
- wandb/sdk/launch/builder/context_manager.py +235 -0
- wandb/sdk/launch/builder/docker_builder.py +7 -23
- wandb/sdk/launch/builder/kaniko_builder.py +10 -23
- wandb/sdk/launch/builder/templates/dockerfile.py +92 -0
- wandb/sdk/launch/create_job.py +51 -45
- wandb/sdk/launch/environment/aws_environment.py +26 -1
- wandb/sdk/launch/inputs/files.py +148 -0
- wandb/sdk/launch/inputs/internal.py +224 -0
- wandb/sdk/launch/inputs/manage.py +95 -0
- wandb/sdk/launch/runner/abstract.py +2 -2
- wandb/sdk/launch/runner/kubernetes_monitor.py +45 -12
- wandb/sdk/launch/runner/kubernetes_runner.py +6 -8
- wandb/sdk/launch/runner/local_container.py +2 -3
- wandb/sdk/launch/runner/local_process.py +8 -29
- wandb/sdk/launch/runner/sagemaker_runner.py +20 -14
- wandb/sdk/launch/runner/vertex_runner.py +8 -7
- wandb/sdk/launch/sweeps/scheduler.py +2 -0
- wandb/sdk/launch/sweeps/utils.py +2 -2
- wandb/sdk/launch/utils.py +16 -138
- wandb/sdk/lib/_settings_toposort_generated.py +2 -5
- wandb/sdk/lib/apikey.py +4 -2
- wandb/sdk/lib/config_util.py +3 -3
- wandb/sdk/lib/proto_util.py +22 -1
- wandb/sdk/lib/redirect.py +1 -1
- wandb/sdk/service/service.py +2 -1
- wandb/sdk/service/streams.py +5 -5
- wandb/sdk/wandb_init.py +25 -59
- wandb/sdk/wandb_login.py +28 -25
- wandb/sdk/wandb_run.py +135 -70
- wandb/sdk/wandb_settings.py +33 -64
- wandb/sdk/wandb_watch.py +1 -1
- wandb/sklearn/plot/classifier.py +4 -6
- wandb/sync/sync.py +2 -2
- wandb/testing/relay.py +32 -17
- wandb/util.py +39 -37
- wandb/wandb_agent.py +3 -3
- wandb/wandb_controller.py +3 -2
- {wandb-0.17.0rc2.dist-info → wandb-0.17.2.dist-info}/METADATA +7 -9
- {wandb-0.17.0rc2.dist-info → wandb-0.17.2.dist-info}/RECORD +131 -153
- {wandb-0.17.0rc2.dist-info → wandb-0.17.2.dist-info}/WHEEL +1 -1
- wandb/apis/reports/v1/_blocks.py +0 -1406
- wandb/apis/reports/v1/_helpers.py +0 -70
- wandb/apis/reports/v1/_panels.py +0 -1282
- wandb/apis/reports/v1/_templates.py +0 -478
- wandb/apis/reports/v1/blocks.py +0 -27
- wandb/apis/reports/v1/helpers.py +0 -2
- wandb/apis/reports/v1/mutations.py +0 -66
- wandb/apis/reports/v1/panels.py +0 -17
- wandb/apis/reports/v1/report.py +0 -268
- wandb/apis/reports/v1/runset.py +0 -144
- wandb/apis/reports/v1/templates.py +0 -7
- wandb/apis/reports/v1/util.py +0 -406
- wandb/apis/reports/v1/validators.py +0 -131
- wandb/apis/reports/v2/blocks.py +0 -25
- wandb/apis/reports/v2/expr_parsing.py +0 -257
- wandb/apis/reports/v2/gql.py +0 -68
- wandb/apis/reports/v2/interface.py +0 -1911
- wandb/apis/reports/v2/internal.py +0 -867
- wandb/apis/reports/v2/metrics.py +0 -6
- wandb/apis/reports/v2/panels.py +0 -15
- wandb/catboost/__init__.py +0 -9
- wandb/fastai/__init__.py +0 -9
- wandb/keras/__init__.py +0 -19
- wandb/lightgbm/__init__.py +0 -9
- wandb/plots/__init__.py +0 -6
- wandb/plots/explain_text.py +0 -36
- wandb/plots/heatmap.py +0 -81
- wandb/plots/named_entity.py +0 -43
- wandb/plots/part_of_speech.py +0 -50
- wandb/plots/plot_definitions.py +0 -768
- wandb/plots/precision_recall.py +0 -121
- wandb/plots/roc.py +0 -103
- wandb/sacred/__init__.py +0 -3
- wandb/xgboost/__init__.py +0 -9
- {wandb-0.17.0rc2.dist-info → wandb-0.17.2.dist-info}/entry_points.txt +0 -0
- {wandb-0.17.0rc2.dist-info → wandb-0.17.2.dist-info}/licenses/LICENSE +0 -0
wandb/apis/reports/v2/metrics.py
DELETED
wandb/apis/reports/v2/panels.py
DELETED
@@ -1,15 +0,0 @@
|
|
1
|
-
# flake8: noqa
|
2
|
-
from .interface import (
|
3
|
-
BarPlot,
|
4
|
-
CodeComparer,
|
5
|
-
CustomChart,
|
6
|
-
LinePlot,
|
7
|
-
MarkdownPanel,
|
8
|
-
MediaBrowser,
|
9
|
-
ParallelCoordinatesPlot,
|
10
|
-
ParameterImportancePlot,
|
11
|
-
RunComparer,
|
12
|
-
ScalarChart,
|
13
|
-
ScatterPlot,
|
14
|
-
WeavePanel,
|
15
|
-
)
|
wandb/catboost/__init__.py
DELETED
wandb/fastai/__init__.py
DELETED
wandb/keras/__init__.py
DELETED
@@ -1,19 +0,0 @@
|
|
1
|
-
"""Compatibility keras module.
|
2
|
-
|
3
|
-
In the future use e.g.:
|
4
|
-
from wandb.integration.keras import WandbCallback
|
5
|
-
"""
|
6
|
-
|
7
|
-
__all__ = (
|
8
|
-
"WandbCallback",
|
9
|
-
"WandbMetricsLogger",
|
10
|
-
"WandbModelCheckpoint",
|
11
|
-
"WandbEvalCallback",
|
12
|
-
)
|
13
|
-
|
14
|
-
from wandb.integration.keras import (
|
15
|
-
WandbCallback,
|
16
|
-
WandbEvalCallback,
|
17
|
-
WandbMetricsLogger,
|
18
|
-
WandbModelCheckpoint,
|
19
|
-
)
|
wandb/lightgbm/__init__.py
DELETED
wandb/plots/__init__.py
DELETED
@@ -1,6 +0,0 @@
|
|
1
|
-
from wandb.plots.explain_text import explain_text as ExplainText
|
2
|
-
from wandb.plots.heatmap import heatmap as HeatMap
|
3
|
-
from wandb.plots.named_entity import named_entity as NER
|
4
|
-
from wandb.plots.part_of_speech import part_of_speech as POS
|
5
|
-
from wandb.plots.precision_recall import precision_recall
|
6
|
-
from wandb.plots.roc import roc as ROC
|
wandb/plots/explain_text.py
DELETED
@@ -1,36 +0,0 @@
|
|
1
|
-
import wandb
|
2
|
-
from wandb import util
|
3
|
-
from wandb.plots.utils import (
|
4
|
-
deprecation_notice,
|
5
|
-
encode_labels,
|
6
|
-
test_missing,
|
7
|
-
test_types,
|
8
|
-
)
|
9
|
-
|
10
|
-
|
11
|
-
def explain_text(text, probas, target_names=None):
|
12
|
-
"""
|
13
|
-
ExplainText adds support for eli5's LIME based TextExplainer.
|
14
|
-
Arguments:
|
15
|
-
text (str): Text to explain
|
16
|
-
probas (black-box classification pipeline): A function which
|
17
|
-
takes a list of strings (documents) and returns a matrix
|
18
|
-
of shape (n_samples, n_classes) with probability values,
|
19
|
-
i.e. a row per document and a column per output label.
|
20
|
-
Returns:
|
21
|
-
Nothing. To see plots, go to your W&B run page.
|
22
|
-
Example:
|
23
|
-
wandb.log({'roc': wandb.plots.ExplainText(text, probas)})
|
24
|
-
"""
|
25
|
-
deprecation_notice()
|
26
|
-
eli5 = util.get_module(
|
27
|
-
"eli5",
|
28
|
-
required="explain_text requires the eli5 library, install with `pip install eli5`",
|
29
|
-
)
|
30
|
-
if test_missing(text=text, probas=probas):
|
31
|
-
# and test_types(proba=proba)):
|
32
|
-
wandb.termlog("Visualizing TextExplainer.")
|
33
|
-
te = eli5.lime.TextExplainer(random_state=42)
|
34
|
-
te.fit(text, probas)
|
35
|
-
html = te.show_prediction(target_names=target_names)
|
36
|
-
return wandb.Html(html.data)
|
wandb/plots/heatmap.py
DELETED
@@ -1,81 +0,0 @@
|
|
1
|
-
import wandb
|
2
|
-
from wandb import util
|
3
|
-
from wandb.plots.utils import (
|
4
|
-
deprecation_notice,
|
5
|
-
encode_labels,
|
6
|
-
test_missing,
|
7
|
-
test_types,
|
8
|
-
)
|
9
|
-
|
10
|
-
chart_limit = wandb.Table.MAX_ROWS
|
11
|
-
|
12
|
-
|
13
|
-
def heatmap(x_labels, y_labels, matrix_values, show_text=False):
|
14
|
-
"""
|
15
|
-
Generates a heatmap.
|
16
|
-
|
17
|
-
Arguments:
|
18
|
-
matrix_values (arr): 2D dataset of shape x_labels * y_labels, containing
|
19
|
-
heatmap values that can be coerced into an ndarray.
|
20
|
-
x_labels (list): Named labels for rows (x_axis).
|
21
|
-
y_labels (list): Named labels for columns (y_axis).
|
22
|
-
show_text (bool): Show text values in heatmap cells.
|
23
|
-
|
24
|
-
Returns:
|
25
|
-
Nothing. To see plots, go to your W&B run page then expand the 'media' tab
|
26
|
-
under 'auto visualizations'.
|
27
|
-
|
28
|
-
Example:
|
29
|
-
wandb.log({'heatmap': wandb.plots.HeatMap(x_labels, y_labels,
|
30
|
-
matrix_values)})
|
31
|
-
"""
|
32
|
-
deprecation_notice()
|
33
|
-
|
34
|
-
np = util.get_module(
|
35
|
-
"numpy",
|
36
|
-
required="roc requires the numpy library, install with `pip install numpy`",
|
37
|
-
)
|
38
|
-
scikit = util.get_module(
|
39
|
-
"sklearn",
|
40
|
-
required="roc requires the scikit library, install with `pip install scikit-learn`",
|
41
|
-
)
|
42
|
-
|
43
|
-
if test_missing(
|
44
|
-
x_labels=x_labels, y_labels=y_labels, matrix_values=matrix_values
|
45
|
-
) and test_types(x_labels=x_labels, y_labels=y_labels, matrix_values=matrix_values):
|
46
|
-
matrix_values = np.array(matrix_values)
|
47
|
-
wandb.termlog("Visualizing heatmap.")
|
48
|
-
|
49
|
-
def heatmap_table(x_labels, y_labels, matrix_values, show_text):
|
50
|
-
x_axis = []
|
51
|
-
y_axis = []
|
52
|
-
values = []
|
53
|
-
count = 0
|
54
|
-
for i, x in enumerate(x_labels):
|
55
|
-
for j, y in enumerate(y_labels):
|
56
|
-
x_axis.append(x)
|
57
|
-
y_axis.append(y)
|
58
|
-
values.append(matrix_values[j][i])
|
59
|
-
count += 1
|
60
|
-
if count >= chart_limit:
|
61
|
-
wandb.termwarn(
|
62
|
-
"wandb uses only the first %d datapoints to create the plots."
|
63
|
-
% wandb.Table.MAX_ROWS
|
64
|
-
)
|
65
|
-
break
|
66
|
-
if show_text:
|
67
|
-
heatmap_key = "wandb/heatmap/v1"
|
68
|
-
else:
|
69
|
-
heatmap_key = "wandb/heatmap_no_text/v1"
|
70
|
-
return wandb.visualize(
|
71
|
-
heatmap_key,
|
72
|
-
wandb.Table(
|
73
|
-
columns=["x_axis", "y_axis", "values"],
|
74
|
-
data=[
|
75
|
-
[x_axis[i], y_axis[i], round(values[i], 2)]
|
76
|
-
for i in range(len(x_axis))
|
77
|
-
],
|
78
|
-
),
|
79
|
-
)
|
80
|
-
|
81
|
-
return heatmap_table(x_labels, y_labels, matrix_values, show_text)
|
wandb/plots/named_entity.py
DELETED
@@ -1,43 +0,0 @@
|
|
1
|
-
import wandb
|
2
|
-
from wandb import util
|
3
|
-
from wandb.plots.utils import (
|
4
|
-
deprecation_notice,
|
5
|
-
encode_labels,
|
6
|
-
test_missing,
|
7
|
-
test_types,
|
8
|
-
)
|
9
|
-
|
10
|
-
|
11
|
-
def named_entity(docs):
|
12
|
-
"""
|
13
|
-
Adds support for spaCy's entity visualizer, which highlights named
|
14
|
-
entities and their labels in a text.
|
15
|
-
|
16
|
-
Arguments:
|
17
|
-
docs (list, Doc, Span): Document(s) to visualize.
|
18
|
-
|
19
|
-
Returns:
|
20
|
-
Nothing. To see plots, go to your W&B run page.
|
21
|
-
|
22
|
-
Example:
|
23
|
-
wandb.log({'NER': wandb.plots.NER(docs=doc)})
|
24
|
-
"""
|
25
|
-
deprecation_notice()
|
26
|
-
|
27
|
-
spacy = util.get_module(
|
28
|
-
"spacy",
|
29
|
-
required="part_of_speech requires the spacy library, install with `pip install spacy`",
|
30
|
-
)
|
31
|
-
en_core_web_md = util.get_module(
|
32
|
-
"en_core_web_md",
|
33
|
-
required="part_of_speech requires the en_core_web_md library, install with `python -m spacy download en_core_web_md`",
|
34
|
-
)
|
35
|
-
nlp = en_core_web_md.load()
|
36
|
-
|
37
|
-
if test_missing(docs=docs):
|
38
|
-
# and test_types(docs=docs)):
|
39
|
-
wandb.termlog("Visualizing named entity recognition.")
|
40
|
-
html = spacy.displacy.render(
|
41
|
-
nlp(str(docs)), style="ent", page=True, minify=True
|
42
|
-
)
|
43
|
-
return wandb.Html(html)
|
wandb/plots/part_of_speech.py
DELETED
@@ -1,50 +0,0 @@
|
|
1
|
-
import wandb
|
2
|
-
from wandb import util
|
3
|
-
from wandb.plots.utils import (
|
4
|
-
deprecation_notice,
|
5
|
-
encode_labels,
|
6
|
-
test_missing,
|
7
|
-
test_types,
|
8
|
-
)
|
9
|
-
|
10
|
-
|
11
|
-
def part_of_speech(docs):
|
12
|
-
"""
|
13
|
-
Adds support for spaCy's dependency visualizer which shows
|
14
|
-
part-of-speech tags and syntactic dependencies.
|
15
|
-
|
16
|
-
Arguments:
|
17
|
-
docs (list, Doc, Span): Document(s) to visualize.
|
18
|
-
|
19
|
-
Returns:
|
20
|
-
Nothing. To see plots, go to your W&B run page.
|
21
|
-
|
22
|
-
Example:
|
23
|
-
wandb.log({'part_of_speech': wandb.plots.POS(docs=doc)})
|
24
|
-
"""
|
25
|
-
deprecation_notice()
|
26
|
-
|
27
|
-
spacy = util.get_module(
|
28
|
-
"spacy",
|
29
|
-
required="part_of_speech requires the spacy library, install with `pip install spacy`",
|
30
|
-
)
|
31
|
-
en_core_web_md = util.get_module(
|
32
|
-
"en_core_web_md",
|
33
|
-
required="part_of_speech requires the en_core_web_md library, install with `python -m spacy download en_core_web_md`",
|
34
|
-
)
|
35
|
-
nlp = en_core_web_md.load()
|
36
|
-
|
37
|
-
if test_missing(docs=docs):
|
38
|
-
# and test_types(docs=docs)):
|
39
|
-
wandb.termlog("Visualizing part of speech.")
|
40
|
-
options = {
|
41
|
-
"compact": True,
|
42
|
-
"color": "#1a1c1f",
|
43
|
-
"font": "Source Sans Pro",
|
44
|
-
"collapse_punct": True,
|
45
|
-
"collapse_phrases": True,
|
46
|
-
}
|
47
|
-
html = spacy.displacy.render(
|
48
|
-
nlp(str(docs)), style="dep", minify=True, options=options, page=True
|
49
|
-
)
|
50
|
-
return wandb.Html(html)
|