wandb 0.22.2__py3-none-macosx_12_0_arm64.whl → 0.22.3__py3-none-macosx_12_0_arm64.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.
Files changed (147) hide show
  1. wandb/__init__.py +1 -1
  2. wandb/__init__.pyi +2 -2
  3. wandb/_pydantic/__init__.py +8 -1
  4. wandb/_pydantic/base.py +54 -18
  5. wandb/_pydantic/field_types.py +8 -3
  6. wandb/_pydantic/pagination.py +46 -0
  7. wandb/_pydantic/utils.py +2 -2
  8. wandb/apis/public/api.py +24 -19
  9. wandb/apis/public/artifacts.py +259 -270
  10. wandb/apis/public/registries/_utils.py +40 -54
  11. wandb/apis/public/registries/registries_search.py +70 -85
  12. wandb/apis/public/registries/registry.py +173 -156
  13. wandb/apis/public/runs.py +27 -6
  14. wandb/apis/public/utils.py +43 -20
  15. wandb/automations/_generated/create_automation.py +2 -2
  16. wandb/automations/_generated/create_generic_webhook_integration.py +4 -4
  17. wandb/automations/_generated/delete_automation.py +2 -2
  18. wandb/automations/_generated/fragments.py +31 -52
  19. wandb/automations/_generated/generic_webhook_integrations_by_entity.py +3 -3
  20. wandb/automations/_generated/get_automations.py +3 -3
  21. wandb/automations/_generated/get_automations_by_entity.py +3 -3
  22. wandb/automations/_generated/input_types.py +9 -9
  23. wandb/automations/_generated/integrations_by_entity.py +3 -3
  24. wandb/automations/_generated/operations.py +6 -6
  25. wandb/automations/_generated/slack_integrations_by_entity.py +3 -3
  26. wandb/automations/_generated/update_automation.py +2 -2
  27. wandb/automations/_utils.py +3 -3
  28. wandb/automations/actions.py +3 -3
  29. wandb/automations/automations.py +6 -5
  30. wandb/bin/gpu_stats +0 -0
  31. wandb/bin/wandb-core +0 -0
  32. wandb/cli/beta.py +8 -2
  33. wandb/cli/beta_leet.py +2 -1
  34. wandb/cli/beta_sync.py +1 -1
  35. wandb/errors/term.py +8 -8
  36. wandb/jupyter.py +0 -51
  37. wandb/old/settings.py +6 -6
  38. wandb/proto/v3/wandb_internal_pb2.py +351 -352
  39. wandb/proto/v3/wandb_server_pb2.py +38 -37
  40. wandb/proto/v3/wandb_settings_pb2.py +2 -2
  41. wandb/proto/v3/wandb_sync_pb2.py +19 -6
  42. wandb/proto/v4/wandb_internal_pb2.py +351 -352
  43. wandb/proto/v4/wandb_server_pb2.py +38 -37
  44. wandb/proto/v4/wandb_settings_pb2.py +2 -2
  45. wandb/proto/v4/wandb_sync_pb2.py +10 -6
  46. wandb/proto/v5/wandb_internal_pb2.py +351 -352
  47. wandb/proto/v5/wandb_server_pb2.py +38 -37
  48. wandb/proto/v5/wandb_settings_pb2.py +2 -2
  49. wandb/proto/v5/wandb_sync_pb2.py +10 -6
  50. wandb/proto/v6/wandb_internal_pb2.py +351 -352
  51. wandb/proto/v6/wandb_server_pb2.py +38 -37
  52. wandb/proto/v6/wandb_settings_pb2.py +2 -2
  53. wandb/proto/v6/wandb_sync_pb2.py +10 -6
  54. wandb/sdk/artifacts/_generated/__init__.py +96 -40
  55. wandb/sdk/artifacts/_generated/add_aliases.py +3 -3
  56. wandb/sdk/artifacts/_generated/add_artifact_collection_tags.py +26 -0
  57. wandb/sdk/artifacts/_generated/artifact_by_id.py +2 -2
  58. wandb/sdk/artifacts/_generated/artifact_by_name.py +3 -3
  59. wandb/sdk/artifacts/_generated/artifact_collection_membership_file_urls.py +27 -8
  60. wandb/sdk/artifacts/_generated/artifact_collection_membership_files.py +27 -8
  61. wandb/sdk/artifacts/_generated/artifact_created_by.py +7 -20
  62. wandb/sdk/artifacts/_generated/artifact_file_urls.py +19 -6
  63. wandb/sdk/artifacts/_generated/artifact_membership_by_name.py +26 -0
  64. wandb/sdk/artifacts/_generated/artifact_type.py +5 -5
  65. wandb/sdk/artifacts/_generated/artifact_used_by.py +8 -17
  66. wandb/sdk/artifacts/_generated/artifact_version_files.py +19 -8
  67. wandb/sdk/artifacts/_generated/delete_aliases.py +3 -3
  68. wandb/sdk/artifacts/_generated/delete_artifact.py +4 -4
  69. wandb/sdk/artifacts/_generated/delete_artifact_collection_tags.py +23 -0
  70. wandb/sdk/artifacts/_generated/delete_artifact_portfolio.py +4 -4
  71. wandb/sdk/artifacts/_generated/delete_artifact_sequence.py +4 -4
  72. wandb/sdk/artifacts/_generated/delete_registry.py +21 -0
  73. wandb/sdk/artifacts/_generated/fetch_artifact_manifest.py +8 -20
  74. wandb/sdk/artifacts/_generated/fetch_linked_artifacts.py +13 -35
  75. wandb/sdk/artifacts/_generated/fetch_org_info_from_entity.py +28 -0
  76. wandb/sdk/artifacts/_generated/fetch_registries.py +18 -8
  77. wandb/sdk/{projects → artifacts}/_generated/fetch_registry.py +4 -4
  78. wandb/sdk/artifacts/_generated/fragments.py +183 -333
  79. wandb/sdk/artifacts/_generated/input_types.py +133 -7
  80. wandb/sdk/artifacts/_generated/link_artifact.py +5 -5
  81. wandb/sdk/artifacts/_generated/operations.py +1053 -548
  82. wandb/sdk/artifacts/_generated/project_artifact_collection.py +9 -77
  83. wandb/sdk/artifacts/_generated/project_artifact_collections.py +21 -9
  84. wandb/sdk/artifacts/_generated/project_artifact_type.py +3 -3
  85. wandb/sdk/artifacts/_generated/project_artifact_types.py +19 -6
  86. wandb/sdk/artifacts/_generated/project_artifacts.py +7 -8
  87. wandb/sdk/artifacts/_generated/registry_collections.py +21 -9
  88. wandb/sdk/artifacts/_generated/registry_versions.py +20 -9
  89. wandb/sdk/artifacts/_generated/rename_registry.py +25 -0
  90. wandb/sdk/artifacts/_generated/run_input_artifacts.py +5 -9
  91. wandb/sdk/artifacts/_generated/run_output_artifacts.py +5 -9
  92. wandb/sdk/artifacts/_generated/type_info.py +2 -2
  93. wandb/sdk/artifacts/_generated/unlink_artifact.py +3 -5
  94. wandb/sdk/artifacts/_generated/update_artifact.py +3 -3
  95. wandb/sdk/artifacts/_generated/update_artifact_collection_type.py +28 -0
  96. wandb/sdk/artifacts/_generated/update_artifact_portfolio.py +7 -16
  97. wandb/sdk/artifacts/_generated/update_artifact_sequence.py +7 -16
  98. wandb/sdk/artifacts/_generated/upsert_registry.py +25 -0
  99. wandb/sdk/artifacts/_gqlutils.py +170 -6
  100. wandb/sdk/artifacts/_models/__init__.py +9 -0
  101. wandb/sdk/artifacts/_models/artifact_collection.py +109 -0
  102. wandb/sdk/artifacts/_models/manifest.py +26 -0
  103. wandb/sdk/artifacts/_models/pagination.py +26 -0
  104. wandb/sdk/artifacts/_models/registry.py +100 -0
  105. wandb/sdk/artifacts/_validators.py +45 -27
  106. wandb/sdk/artifacts/artifact.py +220 -215
  107. wandb/sdk/artifacts/artifact_file_cache.py +1 -1
  108. wandb/sdk/artifacts/artifact_manifest.py +37 -32
  109. wandb/sdk/artifacts/artifact_manifest_entry.py +80 -125
  110. wandb/sdk/artifacts/artifact_manifests/artifact_manifest_v1.py +43 -61
  111. wandb/sdk/artifacts/storage_handlers/gcs_handler.py +8 -6
  112. wandb/sdk/data_types/image.py +2 -2
  113. wandb/sdk/interface/interface.py +72 -64
  114. wandb/sdk/interface/interface_queue.py +27 -18
  115. wandb/sdk/interface/interface_shared.py +61 -23
  116. wandb/sdk/interface/interface_sock.py +9 -5
  117. wandb/sdk/internal/_generated/server_features_query.py +4 -4
  118. wandb/sdk/launch/inputs/schema.py +13 -10
  119. wandb/sdk/lib/apikey.py +8 -12
  120. wandb/sdk/lib/asyncio_compat.py +1 -1
  121. wandb/sdk/lib/asyncio_manager.py +5 -5
  122. wandb/sdk/lib/console_capture.py +38 -30
  123. wandb/sdk/lib/progress.py +159 -64
  124. wandb/sdk/lib/retry.py +3 -2
  125. wandb/sdk/lib/service/service_connection.py +2 -2
  126. wandb/sdk/lib/wb_logging.py +2 -1
  127. wandb/sdk/mailbox/mailbox.py +1 -1
  128. wandb/sdk/wandb_init.py +10 -13
  129. wandb/sdk/wandb_run.py +9 -46
  130. wandb/sdk/wandb_settings.py +102 -19
  131. {wandb-0.22.2.dist-info → wandb-0.22.3.dist-info}/METADATA +2 -1
  132. {wandb-0.22.2.dist-info → wandb-0.22.3.dist-info}/RECORD +135 -134
  133. wandb/sdk/artifacts/_generated/artifact_via_membership_by_name.py +0 -26
  134. wandb/sdk/artifacts/_generated/create_artifact_collection_tag_assignments.py +0 -36
  135. wandb/sdk/artifacts/_generated/delete_artifact_collection_tag_assignments.py +0 -25
  136. wandb/sdk/artifacts/_generated/move_artifact_collection.py +0 -35
  137. wandb/sdk/projects/_generated/__init__.py +0 -26
  138. wandb/sdk/projects/_generated/delete_project.py +0 -22
  139. wandb/sdk/projects/_generated/enums.py +0 -4
  140. wandb/sdk/projects/_generated/fragments.py +0 -41
  141. wandb/sdk/projects/_generated/input_types.py +0 -13
  142. wandb/sdk/projects/_generated/operations.py +0 -88
  143. wandb/sdk/projects/_generated/rename_project.py +0 -27
  144. wandb/sdk/projects/_generated/upsert_registry_project.py +0 -27
  145. {wandb-0.22.2.dist-info → wandb-0.22.3.dist-info}/WHEEL +0 -0
  146. {wandb-0.22.2.dist-info → wandb-0.22.3.dist-info}/entry_points.txt +0 -0
  147. {wandb-0.22.2.dist-info → wandb-0.22.3.dist-info}/licenses/LICENSE +0 -0
@@ -1,26 +0,0 @@
1
- # Generated by ariadne-codegen
2
-
3
- __all__ = [
4
- "DELETE_PROJECT_GQL",
5
- "FETCH_REGISTRY_GQL",
6
- "RENAME_PROJECT_GQL",
7
- "UPSERT_REGISTRY_PROJECT_GQL",
8
- "FetchRegistry",
9
- "RenameProject",
10
- "UpsertRegistryProject",
11
- "DeleteProject",
12
- "ArtifactTypeInput",
13
- "RegistryFragment",
14
- ]
15
- from .delete_project import DeleteProject
16
- from .fetch_registry import FetchRegistry
17
- from .fragments import RegistryFragment
18
- from .input_types import ArtifactTypeInput
19
- from .operations import (
20
- DELETE_PROJECT_GQL,
21
- FETCH_REGISTRY_GQL,
22
- RENAME_PROJECT_GQL,
23
- UPSERT_REGISTRY_PROJECT_GQL,
24
- )
25
- from .rename_project import RenameProject
26
- from .upsert_registry_project import UpsertRegistryProject
@@ -1,22 +0,0 @@
1
- # Generated by ariadne-codegen
2
- # Source: tools/graphql_codegen/projects/
3
-
4
- from __future__ import annotations
5
-
6
- from typing import Literal, Optional
7
-
8
- from pydantic import Field
9
-
10
- from wandb._pydantic import GQLBase, Typename
11
-
12
-
13
- class DeleteProject(GQLBase):
14
- delete_model: Optional[DeleteProjectDeleteModel] = Field(alias="deleteModel")
15
-
16
-
17
- class DeleteProjectDeleteModel(GQLBase):
18
- success: Optional[bool]
19
- typename__: Typename[Literal["DeleteModelPayload"]]
20
-
21
-
22
- DeleteProject.model_rebuild()
@@ -1,4 +0,0 @@
1
- # Generated by ariadne-codegen
2
- # Source: core/api/graphql/schemas/schema-latest.graphql
3
-
4
- from __future__ import annotations
@@ -1,41 +0,0 @@
1
- # Generated by ariadne-codegen
2
- # Source: tools/graphql_codegen/projects/
3
-
4
- from __future__ import annotations
5
-
6
- from typing import List, Optional
7
-
8
- from pydantic import Field
9
-
10
- from wandb._pydantic import GQLBase, GQLId
11
-
12
-
13
- class RegistryFragment(GQLBase):
14
- id: GQLId
15
- allow_all_artifact_types_in_registry: bool = Field(
16
- alias="allowAllArtifactTypesInRegistry"
17
- )
18
- artifact_types: RegistryFragmentArtifactTypes = Field(alias="artifactTypes")
19
- name: str
20
- description: Optional[str]
21
- created_at: str = Field(alias="createdAt")
22
- updated_at: Optional[str] = Field(alias="updatedAt")
23
- access: Optional[str]
24
-
25
-
26
- class RegistryFragmentArtifactTypes(GQLBase):
27
- edges: List[RegistryFragmentArtifactTypesEdges]
28
-
29
-
30
- class RegistryFragmentArtifactTypesEdges(GQLBase):
31
- node: Optional[RegistryFragmentArtifactTypesEdgesNode]
32
-
33
-
34
- class RegistryFragmentArtifactTypesEdgesNode(GQLBase):
35
- name: str
36
-
37
-
38
- RegistryFragment.model_rebuild()
39
- RegistryFragmentArtifactTypes.model_rebuild()
40
- RegistryFragmentArtifactTypesEdges.model_rebuild()
41
- RegistryFragmentArtifactTypesEdgesNode.model_rebuild()
@@ -1,13 +0,0 @@
1
- # Generated by ariadne-codegen
2
- # Source: core/api/graphql/schemas/schema-latest.graphql
3
-
4
- from __future__ import annotations
5
-
6
- from typing import Optional
7
-
8
- from wandb._pydantic import GQLBase
9
-
10
-
11
- class ArtifactTypeInput(GQLBase):
12
- name: str
13
- description: Optional[str] = None
@@ -1,88 +0,0 @@
1
- # Generated by ariadne-codegen
2
- # Source: tools/graphql_codegen/projects/
3
-
4
- __all__ = [
5
- "DELETE_PROJECT_GQL",
6
- "FETCH_REGISTRY_GQL",
7
- "RENAME_PROJECT_GQL",
8
- "UPSERT_REGISTRY_PROJECT_GQL",
9
- ]
10
-
11
- FETCH_REGISTRY_GQL = """
12
- query FetchRegistry($name: String, $entityName: String) {
13
- entity(name: $entityName) {
14
- project(name: $name) {
15
- ...RegistryFragment
16
- }
17
- }
18
- }
19
-
20
- fragment RegistryFragment on Project {
21
- id
22
- allowAllArtifactTypesInRegistry
23
- artifactTypes(includeAll: true) {
24
- edges {
25
- node {
26
- name
27
- }
28
- }
29
- }
30
- name
31
- description
32
- createdAt
33
- updatedAt
34
- access
35
- }
36
- """
37
-
38
- RENAME_PROJECT_GQL = """
39
- mutation renameProject($entityName: String!, $oldProjectName: String!, $newProjectName: String!) {
40
- renameProject(
41
- input: {entityName: $entityName, oldProjectName: $oldProjectName, newProjectName: $newProjectName}
42
- ) {
43
- project {
44
- name
45
- }
46
- inserted
47
- }
48
- }
49
- """
50
-
51
- UPSERT_REGISTRY_PROJECT_GQL = """
52
- mutation UpsertRegistryProject($description: String, $entityName: String, $name: String, $access: String, $allowAllArtifactTypesInRegistry: Boolean, $artifactTypes: [ArtifactTypeInput!]) {
53
- upsertModel(
54
- input: {description: $description, entityName: $entityName, name: $name, access: $access, allowAllArtifactTypesInRegistry: $allowAllArtifactTypesInRegistry, artifactTypes: $artifactTypes}
55
- ) {
56
- project {
57
- ...RegistryFragment
58
- }
59
- inserted
60
- }
61
- }
62
-
63
- fragment RegistryFragment on Project {
64
- id
65
- allowAllArtifactTypesInRegistry
66
- artifactTypes(includeAll: true) {
67
- edges {
68
- node {
69
- name
70
- }
71
- }
72
- }
73
- name
74
- description
75
- createdAt
76
- updatedAt
77
- access
78
- }
79
- """
80
-
81
- DELETE_PROJECT_GQL = """
82
- mutation deleteProject($id: String!) {
83
- deleteModel(input: {id: $id}) {
84
- success
85
- __typename
86
- }
87
- }
88
- """
@@ -1,27 +0,0 @@
1
- # Generated by ariadne-codegen
2
- # Source: tools/graphql_codegen/projects/
3
-
4
- from __future__ import annotations
5
-
6
- from typing import Optional
7
-
8
- from pydantic import Field
9
-
10
- from wandb._pydantic import GQLBase
11
-
12
-
13
- class RenameProject(GQLBase):
14
- rename_project: Optional[RenameProjectRenameProject] = Field(alias="renameProject")
15
-
16
-
17
- class RenameProjectRenameProject(GQLBase):
18
- project: Optional[RenameProjectRenameProjectProject]
19
- inserted: Optional[bool]
20
-
21
-
22
- class RenameProjectRenameProjectProject(GQLBase):
23
- name: str
24
-
25
-
26
- RenameProject.model_rebuild()
27
- RenameProjectRenameProject.model_rebuild()
@@ -1,27 +0,0 @@
1
- # Generated by ariadne-codegen
2
- # Source: tools/graphql_codegen/projects/
3
-
4
- from __future__ import annotations
5
-
6
- from typing import Optional
7
-
8
- from pydantic import Field
9
-
10
- from wandb._pydantic import GQLBase
11
-
12
- from .fragments import RegistryFragment
13
-
14
-
15
- class UpsertRegistryProject(GQLBase):
16
- upsert_model: Optional[UpsertRegistryProjectUpsertModel] = Field(
17
- alias="upsertModel"
18
- )
19
-
20
-
21
- class UpsertRegistryProjectUpsertModel(GQLBase):
22
- project: Optional[RegistryFragment]
23
- inserted: Optional[bool]
24
-
25
-
26
- UpsertRegistryProject.model_rebuild()
27
- UpsertRegistryProjectUpsertModel.model_rebuild()
File without changes