wandb 0.17.0rc2__py3-none-win32.whl → 0.17.1__py3-none-win32.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (159) hide show
  1. wandb/__init__.py +1 -2
  2. wandb/apis/importers/internals/internal.py +0 -1
  3. wandb/apis/importers/wandb.py +12 -7
  4. wandb/apis/internal.py +0 -3
  5. wandb/apis/public/api.py +213 -79
  6. wandb/apis/public/artifacts.py +335 -100
  7. wandb/apis/public/files.py +9 -9
  8. wandb/apis/public/jobs.py +16 -4
  9. wandb/apis/public/projects.py +26 -28
  10. wandb/apis/public/query_generator.py +1 -1
  11. wandb/apis/public/runs.py +163 -65
  12. wandb/apis/public/sweeps.py +2 -2
  13. wandb/apis/reports/__init__.py +1 -7
  14. wandb/apis/reports/v1/__init__.py +5 -27
  15. wandb/apis/reports/v2/__init__.py +7 -19
  16. wandb/apis/workspaces/__init__.py +8 -0
  17. wandb/beta/workflows.py +8 -3
  18. wandb/bin/wandb-core +0 -0
  19. wandb/cli/cli.py +131 -59
  20. wandb/docker/__init__.py +1 -1
  21. wandb/errors/term.py +10 -2
  22. wandb/filesync/step_checksum.py +1 -4
  23. wandb/filesync/step_prepare.py +4 -24
  24. wandb/filesync/step_upload.py +5 -107
  25. wandb/filesync/upload_job.py +0 -76
  26. wandb/integration/gym/__init__.py +35 -15
  27. wandb/integration/openai/fine_tuning.py +21 -3
  28. wandb/integration/prodigy/prodigy.py +1 -1
  29. wandb/jupyter.py +16 -17
  30. wandb/plot/pr_curve.py +2 -1
  31. wandb/plot/roc_curve.py +2 -1
  32. wandb/{plots → plot}/utils.py +13 -25
  33. wandb/proto/v3/wandb_internal_pb2.py +54 -54
  34. wandb/proto/v3/wandb_settings_pb2.py +2 -2
  35. wandb/proto/v3/wandb_telemetry_pb2.py +10 -10
  36. wandb/proto/v4/wandb_internal_pb2.py +54 -54
  37. wandb/proto/v4/wandb_settings_pb2.py +2 -2
  38. wandb/proto/v4/wandb_telemetry_pb2.py +10 -10
  39. wandb/proto/v5/wandb_base_pb2.py +30 -0
  40. wandb/proto/v5/wandb_internal_pb2.py +355 -0
  41. wandb/proto/v5/wandb_server_pb2.py +63 -0
  42. wandb/proto/v5/wandb_settings_pb2.py +45 -0
  43. wandb/proto/v5/wandb_telemetry_pb2.py +41 -0
  44. wandb/proto/wandb_base_pb2.py +2 -0
  45. wandb/proto/wandb_deprecated.py +9 -1
  46. wandb/proto/wandb_generate_deprecated.py +34 -0
  47. wandb/proto/{wandb_internal_codegen.py → wandb_generate_proto.py} +1 -35
  48. wandb/proto/wandb_internal_pb2.py +2 -0
  49. wandb/proto/wandb_server_pb2.py +2 -0
  50. wandb/proto/wandb_settings_pb2.py +2 -0
  51. wandb/proto/wandb_telemetry_pb2.py +2 -0
  52. wandb/sdk/artifacts/artifact.py +68 -22
  53. wandb/sdk/artifacts/artifact_manifest.py +1 -1
  54. wandb/sdk/artifacts/artifact_manifest_entry.py +6 -3
  55. wandb/sdk/artifacts/artifact_manifests/artifact_manifest_v1.py +1 -1
  56. wandb/sdk/artifacts/artifact_saver.py +1 -10
  57. wandb/sdk/artifacts/storage_handlers/local_file_handler.py +6 -2
  58. wandb/sdk/artifacts/storage_handlers/multi_handler.py +1 -1
  59. wandb/sdk/artifacts/storage_handlers/tracking_handler.py +6 -4
  60. wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py +2 -42
  61. wandb/sdk/artifacts/storage_policy.py +1 -12
  62. wandb/sdk/data_types/image.py +1 -1
  63. wandb/sdk/data_types/video.py +4 -2
  64. wandb/sdk/interface/interface.py +13 -0
  65. wandb/sdk/interface/interface_shared.py +1 -1
  66. wandb/sdk/internal/file_pusher.py +2 -5
  67. wandb/sdk/internal/file_stream.py +6 -19
  68. wandb/sdk/internal/internal_api.py +148 -136
  69. wandb/sdk/internal/job_builder.py +207 -135
  70. wandb/sdk/internal/progress.py +0 -28
  71. wandb/sdk/internal/sender.py +102 -39
  72. wandb/sdk/internal/settings_static.py +8 -1
  73. wandb/sdk/internal/system/assets/trainium.py +3 -3
  74. wandb/sdk/internal/system/system_info.py +4 -2
  75. wandb/sdk/internal/update.py +1 -1
  76. wandb/sdk/launch/__init__.py +9 -1
  77. wandb/sdk/launch/_launch.py +4 -24
  78. wandb/sdk/launch/_launch_add.py +1 -3
  79. wandb/sdk/launch/_project_spec.py +184 -224
  80. wandb/sdk/launch/agent/agent.py +58 -18
  81. wandb/sdk/launch/agent/config.py +0 -3
  82. wandb/sdk/launch/builder/abstract.py +67 -0
  83. wandb/sdk/launch/builder/build.py +165 -576
  84. wandb/sdk/launch/builder/context_manager.py +235 -0
  85. wandb/sdk/launch/builder/docker_builder.py +7 -23
  86. wandb/sdk/launch/builder/kaniko_builder.py +10 -23
  87. wandb/sdk/launch/builder/templates/dockerfile.py +92 -0
  88. wandb/sdk/launch/create_job.py +51 -45
  89. wandb/sdk/launch/environment/aws_environment.py +26 -1
  90. wandb/sdk/launch/inputs/files.py +148 -0
  91. wandb/sdk/launch/inputs/internal.py +224 -0
  92. wandb/sdk/launch/inputs/manage.py +95 -0
  93. wandb/sdk/launch/runner/abstract.py +2 -2
  94. wandb/sdk/launch/runner/kubernetes_monitor.py +45 -12
  95. wandb/sdk/launch/runner/kubernetes_runner.py +6 -8
  96. wandb/sdk/launch/runner/local_container.py +2 -3
  97. wandb/sdk/launch/runner/local_process.py +8 -29
  98. wandb/sdk/launch/runner/sagemaker_runner.py +20 -14
  99. wandb/sdk/launch/runner/vertex_runner.py +8 -7
  100. wandb/sdk/launch/sweeps/scheduler.py +2 -0
  101. wandb/sdk/launch/sweeps/utils.py +2 -2
  102. wandb/sdk/launch/utils.py +16 -138
  103. wandb/sdk/lib/_settings_toposort_generated.py +2 -5
  104. wandb/sdk/lib/apikey.py +4 -2
  105. wandb/sdk/lib/config_util.py +3 -3
  106. wandb/sdk/lib/proto_util.py +22 -1
  107. wandb/sdk/lib/redirect.py +1 -1
  108. wandb/sdk/service/service.py +2 -1
  109. wandb/sdk/service/streams.py +5 -5
  110. wandb/sdk/wandb_init.py +25 -59
  111. wandb/sdk/wandb_login.py +28 -25
  112. wandb/sdk/wandb_run.py +112 -45
  113. wandb/sdk/wandb_settings.py +33 -64
  114. wandb/sdk/wandb_watch.py +1 -1
  115. wandb/sklearn/plot/classifier.py +4 -6
  116. wandb/sync/sync.py +2 -2
  117. wandb/testing/relay.py +32 -17
  118. wandb/util.py +36 -37
  119. wandb/wandb_agent.py +3 -3
  120. wandb/wandb_controller.py +3 -2
  121. {wandb-0.17.0rc2.dist-info → wandb-0.17.1.dist-info}/METADATA +7 -9
  122. {wandb-0.17.0rc2.dist-info → wandb-0.17.1.dist-info}/RECORD +125 -147
  123. wandb/apis/reports/v1/_blocks.py +0 -1406
  124. wandb/apis/reports/v1/_helpers.py +0 -70
  125. wandb/apis/reports/v1/_panels.py +0 -1282
  126. wandb/apis/reports/v1/_templates.py +0 -478
  127. wandb/apis/reports/v1/blocks.py +0 -27
  128. wandb/apis/reports/v1/helpers.py +0 -2
  129. wandb/apis/reports/v1/mutations.py +0 -66
  130. wandb/apis/reports/v1/panels.py +0 -17
  131. wandb/apis/reports/v1/report.py +0 -268
  132. wandb/apis/reports/v1/runset.py +0 -144
  133. wandb/apis/reports/v1/templates.py +0 -7
  134. wandb/apis/reports/v1/util.py +0 -406
  135. wandb/apis/reports/v1/validators.py +0 -131
  136. wandb/apis/reports/v2/blocks.py +0 -25
  137. wandb/apis/reports/v2/expr_parsing.py +0 -257
  138. wandb/apis/reports/v2/gql.py +0 -68
  139. wandb/apis/reports/v2/interface.py +0 -1911
  140. wandb/apis/reports/v2/internal.py +0 -867
  141. wandb/apis/reports/v2/metrics.py +0 -6
  142. wandb/apis/reports/v2/panels.py +0 -15
  143. wandb/catboost/__init__.py +0 -9
  144. wandb/fastai/__init__.py +0 -9
  145. wandb/keras/__init__.py +0 -19
  146. wandb/lightgbm/__init__.py +0 -9
  147. wandb/plots/__init__.py +0 -6
  148. wandb/plots/explain_text.py +0 -36
  149. wandb/plots/heatmap.py +0 -81
  150. wandb/plots/named_entity.py +0 -43
  151. wandb/plots/part_of_speech.py +0 -50
  152. wandb/plots/plot_definitions.py +0 -768
  153. wandb/plots/precision_recall.py +0 -121
  154. wandb/plots/roc.py +0 -103
  155. wandb/sacred/__init__.py +0 -3
  156. wandb/xgboost/__init__.py +0 -9
  157. {wandb-0.17.0rc2.dist-info → wandb-0.17.1.dist-info}/WHEEL +0 -0
  158. {wandb-0.17.0rc2.dist-info → wandb-0.17.1.dist-info}/entry_points.txt +0 -0
  159. {wandb-0.17.0rc2.dist-info → wandb-0.17.1.dist-info}/licenses/LICENSE +0 -0
@@ -1,6 +0,0 @@
1
- # flake8: noqa
2
- from .interface import (
3
- Config,
4
- Metric,
5
- SummaryMetric,
6
- )
@@ -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
- )
@@ -1,9 +0,0 @@
1
- """Compatibility catboost module.
2
-
3
- In the future use:
4
- from wandb.integration.catboost import WandbCallback
5
- """
6
-
7
- from wandb.integration.catboost import WandbCallback, log_summary
8
-
9
- __all__ = ["log_summary", "WandbCallback"]
wandb/fastai/__init__.py DELETED
@@ -1,9 +0,0 @@
1
- """Compatibility fastai module.
2
-
3
- In the future use:
4
- from wandb.integration.fastai import WandbCallback
5
- """
6
-
7
- from wandb.integration.fastai import WandbCallback
8
-
9
- __all__ = ["WandbCallback"]
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
- )
@@ -1,9 +0,0 @@
1
- """Compatibility lightgbm module.
2
-
3
- In the future use:
4
- from wandb.integration.lightgbm import wandb_callback
5
- """
6
-
7
- from wandb.integration.lightgbm import log_summary, wandb_callback
8
-
9
- __all__ = ["wandb_callback", "log_summary"]
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
@@ -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)
@@ -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)
@@ -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)