wandb 0.22.1__py3-none-win32.whl → 0.22.3__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.
Files changed (166) hide show
  1. wandb/__init__.py +1 -1
  2. wandb/__init__.pyi +7 -4
  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.exe +0 -0
  31. wandb/bin/wandb-core +0 -0
  32. wandb/cli/beta.py +23 -3
  33. wandb/cli/beta_leet.py +75 -0
  34. wandb/cli/beta_sync.py +1 -1
  35. wandb/cli/cli.py +34 -7
  36. wandb/errors/term.py +8 -8
  37. wandb/jupyter.py +0 -51
  38. wandb/old/settings.py +6 -6
  39. wandb/proto/v3/wandb_api_pb2.py +86 -0
  40. wandb/proto/v3/wandb_server_pb2.py +38 -37
  41. wandb/proto/v3/wandb_settings_pb2.py +2 -2
  42. wandb/proto/v3/wandb_sync_pb2.py +19 -6
  43. wandb/proto/v4/wandb_api_pb2.py +37 -0
  44. wandb/proto/v4/wandb_server_pb2.py +38 -37
  45. wandb/proto/v4/wandb_settings_pb2.py +2 -2
  46. wandb/proto/v4/wandb_sync_pb2.py +10 -6
  47. wandb/proto/v5/wandb_api_pb2.py +38 -0
  48. wandb/proto/v5/wandb_server_pb2.py +38 -37
  49. wandb/proto/v5/wandb_settings_pb2.py +2 -2
  50. wandb/proto/v5/wandb_sync_pb2.py +10 -6
  51. wandb/proto/v6/wandb_api_pb2.py +48 -0
  52. wandb/proto/v6/wandb_server_pb2.py +38 -37
  53. wandb/proto/v6/wandb_settings_pb2.py +2 -2
  54. wandb/proto/v6/wandb_sync_pb2.py +10 -6
  55. wandb/proto/wandb_api_pb2.py +18 -0
  56. wandb/proto/wandb_generate_proto.py +1 -0
  57. wandb/sdk/artifacts/_generated/__init__.py +96 -40
  58. wandb/sdk/artifacts/_generated/add_aliases.py +3 -3
  59. wandb/sdk/artifacts/_generated/add_artifact_collection_tags.py +26 -0
  60. wandb/sdk/artifacts/_generated/artifact_by_id.py +2 -2
  61. wandb/sdk/artifacts/_generated/artifact_by_name.py +3 -3
  62. wandb/sdk/artifacts/_generated/artifact_collection_membership_file_urls.py +27 -8
  63. wandb/sdk/artifacts/_generated/artifact_collection_membership_files.py +27 -8
  64. wandb/sdk/artifacts/_generated/artifact_created_by.py +7 -20
  65. wandb/sdk/artifacts/_generated/artifact_file_urls.py +19 -6
  66. wandb/sdk/artifacts/_generated/artifact_membership_by_name.py +26 -0
  67. wandb/sdk/artifacts/_generated/artifact_type.py +5 -5
  68. wandb/sdk/artifacts/_generated/artifact_used_by.py +8 -17
  69. wandb/sdk/artifacts/_generated/artifact_version_files.py +19 -8
  70. wandb/sdk/artifacts/_generated/delete_aliases.py +3 -3
  71. wandb/sdk/artifacts/_generated/delete_artifact.py +4 -4
  72. wandb/sdk/artifacts/_generated/delete_artifact_collection_tags.py +23 -0
  73. wandb/sdk/artifacts/_generated/delete_artifact_portfolio.py +4 -4
  74. wandb/sdk/artifacts/_generated/delete_artifact_sequence.py +4 -4
  75. wandb/sdk/artifacts/_generated/delete_registry.py +21 -0
  76. wandb/sdk/artifacts/_generated/fetch_artifact_manifest.py +8 -20
  77. wandb/sdk/artifacts/_generated/fetch_linked_artifacts.py +13 -35
  78. wandb/sdk/artifacts/_generated/fetch_org_info_from_entity.py +28 -0
  79. wandb/sdk/artifacts/_generated/fetch_registries.py +18 -8
  80. wandb/sdk/{projects → artifacts}/_generated/fetch_registry.py +4 -4
  81. wandb/sdk/artifacts/_generated/fragments.py +183 -333
  82. wandb/sdk/artifacts/_generated/input_types.py +133 -7
  83. wandb/sdk/artifacts/_generated/link_artifact.py +5 -5
  84. wandb/sdk/artifacts/_generated/operations.py +1053 -548
  85. wandb/sdk/artifacts/_generated/project_artifact_collection.py +9 -77
  86. wandb/sdk/artifacts/_generated/project_artifact_collections.py +21 -9
  87. wandb/sdk/artifacts/_generated/project_artifact_type.py +3 -3
  88. wandb/sdk/artifacts/_generated/project_artifact_types.py +19 -6
  89. wandb/sdk/artifacts/_generated/project_artifacts.py +7 -8
  90. wandb/sdk/artifacts/_generated/registry_collections.py +21 -9
  91. wandb/sdk/artifacts/_generated/registry_versions.py +20 -9
  92. wandb/sdk/artifacts/_generated/rename_registry.py +25 -0
  93. wandb/sdk/artifacts/_generated/run_input_artifacts.py +5 -9
  94. wandb/sdk/artifacts/_generated/run_output_artifacts.py +5 -9
  95. wandb/sdk/artifacts/_generated/type_info.py +2 -2
  96. wandb/sdk/artifacts/_generated/unlink_artifact.py +3 -5
  97. wandb/sdk/artifacts/_generated/update_artifact.py +3 -3
  98. wandb/sdk/artifacts/_generated/update_artifact_collection_type.py +28 -0
  99. wandb/sdk/artifacts/_generated/update_artifact_portfolio.py +7 -16
  100. wandb/sdk/artifacts/_generated/update_artifact_sequence.py +7 -16
  101. wandb/sdk/artifacts/_generated/upsert_registry.py +25 -0
  102. wandb/sdk/artifacts/_gqlutils.py +170 -6
  103. wandb/sdk/artifacts/_models/__init__.py +9 -0
  104. wandb/sdk/artifacts/_models/artifact_collection.py +109 -0
  105. wandb/sdk/artifacts/_models/manifest.py +26 -0
  106. wandb/sdk/artifacts/_models/pagination.py +26 -0
  107. wandb/sdk/artifacts/_models/registry.py +100 -0
  108. wandb/sdk/artifacts/_validators.py +45 -27
  109. wandb/sdk/artifacts/artifact.py +249 -244
  110. wandb/sdk/artifacts/artifact_file_cache.py +1 -1
  111. wandb/sdk/artifacts/artifact_manifest.py +37 -32
  112. wandb/sdk/artifacts/artifact_manifest_entry.py +82 -133
  113. wandb/sdk/artifacts/artifact_manifests/artifact_manifest_v1.py +43 -61
  114. wandb/sdk/artifacts/storage_handler.py +18 -12
  115. wandb/sdk/artifacts/storage_handlers/azure_handler.py +11 -6
  116. wandb/sdk/artifacts/storage_handlers/gcs_handler.py +17 -12
  117. wandb/sdk/artifacts/storage_handlers/http_handler.py +9 -4
  118. wandb/sdk/artifacts/storage_handlers/local_file_handler.py +10 -6
  119. wandb/sdk/artifacts/storage_handlers/multi_handler.py +5 -4
  120. wandb/sdk/artifacts/storage_handlers/s3_handler.py +10 -8
  121. wandb/sdk/artifacts/storage_handlers/tracking_handler.py +6 -4
  122. wandb/sdk/artifacts/storage_handlers/wb_artifact_handler.py +24 -21
  123. wandb/sdk/artifacts/storage_handlers/wb_local_artifact_handler.py +4 -2
  124. wandb/sdk/artifacts/storage_policies/_multipart.py +187 -0
  125. wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py +61 -242
  126. wandb/sdk/artifacts/storage_policy.py +25 -12
  127. wandb/sdk/data_types/image.py +2 -2
  128. wandb/sdk/data_types/object_3d.py +67 -2
  129. wandb/sdk/interface/interface.py +72 -64
  130. wandb/sdk/interface/interface_queue.py +27 -18
  131. wandb/sdk/interface/interface_shared.py +61 -23
  132. wandb/sdk/interface/interface_sock.py +9 -5
  133. wandb/sdk/internal/_generated/server_features_query.py +4 -4
  134. wandb/sdk/internal/job_builder.py +27 -10
  135. wandb/sdk/internal/sender.py +4 -1
  136. wandb/sdk/launch/create_job.py +2 -1
  137. wandb/sdk/launch/inputs/schema.py +13 -10
  138. wandb/sdk/lib/apikey.py +8 -12
  139. wandb/sdk/lib/asyncio_compat.py +1 -1
  140. wandb/sdk/lib/asyncio_manager.py +5 -5
  141. wandb/sdk/lib/console_capture.py +38 -30
  142. wandb/sdk/lib/progress.py +151 -125
  143. wandb/sdk/lib/retry.py +3 -2
  144. wandb/sdk/lib/service/service_connection.py +2 -2
  145. wandb/sdk/lib/wb_logging.py +2 -1
  146. wandb/sdk/mailbox/mailbox.py +1 -1
  147. wandb/sdk/wandb_init.py +11 -14
  148. wandb/sdk/wandb_run.py +14 -48
  149. wandb/sdk/wandb_settings.py +114 -30
  150. {wandb-0.22.1.dist-info → wandb-0.22.3.dist-info}/METADATA +2 -1
  151. {wandb-0.22.1.dist-info → wandb-0.22.3.dist-info}/RECORD +154 -146
  152. wandb/sdk/artifacts/_generated/artifact_via_membership_by_name.py +0 -26
  153. wandb/sdk/artifacts/_generated/create_artifact_collection_tag_assignments.py +0 -36
  154. wandb/sdk/artifacts/_generated/delete_artifact_collection_tag_assignments.py +0 -25
  155. wandb/sdk/artifacts/_generated/move_artifact_collection.py +0 -35
  156. wandb/sdk/projects/_generated/__init__.py +0 -26
  157. wandb/sdk/projects/_generated/delete_project.py +0 -22
  158. wandb/sdk/projects/_generated/enums.py +0 -4
  159. wandb/sdk/projects/_generated/fragments.py +0 -41
  160. wandb/sdk/projects/_generated/input_types.py +0 -13
  161. wandb/sdk/projects/_generated/operations.py +0 -88
  162. wandb/sdk/projects/_generated/rename_project.py +0 -27
  163. wandb/sdk/projects/_generated/upsert_registry_project.py +0 -27
  164. {wandb-0.22.1.dist-info → wandb-0.22.3.dist-info}/WHEEL +0 -0
  165. {wandb-0.22.1.dist-info → wandb-0.22.3.dist-info}/entry_points.txt +0 -0
  166. {wandb-0.22.1.dist-info → wandb-0.22.3.dist-info}/licenses/LICENSE +0 -0
@@ -3,26 +3,26 @@
3
3
 
4
4
  from __future__ import annotations
5
5
 
6
- from typing import Literal, Optional
6
+ from typing import List, Literal, Optional
7
7
 
8
8
  from pydantic import Field
9
9
 
10
- from wandb._pydantic import GQLBase, Typename
10
+ from wandb._pydantic import GQLResult, Typename
11
11
 
12
- from .fragments import FileUrlsFragment
12
+ from .fragments import FileWithUrlFragment, PageInfoFragment
13
13
 
14
14
 
15
- class ArtifactCollectionMembershipFileUrls(GQLBase):
15
+ class ArtifactCollectionMembershipFileUrls(GQLResult):
16
16
  project: Optional[ArtifactCollectionMembershipFileUrlsProject]
17
17
 
18
18
 
19
- class ArtifactCollectionMembershipFileUrlsProject(GQLBase):
19
+ class ArtifactCollectionMembershipFileUrlsProject(GQLResult):
20
20
  artifact_collection: Optional[
21
21
  ArtifactCollectionMembershipFileUrlsProjectArtifactCollection
22
22
  ] = Field(alias="artifactCollection")
23
23
 
24
24
 
25
- class ArtifactCollectionMembershipFileUrlsProjectArtifactCollection(GQLBase):
25
+ class ArtifactCollectionMembershipFileUrlsProjectArtifactCollection(GQLResult):
26
26
  typename__: Typename[
27
27
  Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
28
28
  ]
@@ -32,12 +32,31 @@ class ArtifactCollectionMembershipFileUrlsProjectArtifactCollection(GQLBase):
32
32
 
33
33
 
34
34
  class ArtifactCollectionMembershipFileUrlsProjectArtifactCollectionArtifactMembership(
35
- GQLBase
35
+ GQLResult
36
36
  ):
37
- files: Optional[FileUrlsFragment]
37
+ files: Optional[
38
+ ArtifactCollectionMembershipFileUrlsProjectArtifactCollectionArtifactMembershipFiles
39
+ ]
40
+
41
+
42
+ class ArtifactCollectionMembershipFileUrlsProjectArtifactCollectionArtifactMembershipFiles(
43
+ GQLResult
44
+ ):
45
+ page_info: PageInfoFragment = Field(alias="pageInfo")
46
+ edges: List[
47
+ ArtifactCollectionMembershipFileUrlsProjectArtifactCollectionArtifactMembershipFilesEdges
48
+ ]
49
+
50
+
51
+ class ArtifactCollectionMembershipFileUrlsProjectArtifactCollectionArtifactMembershipFilesEdges(
52
+ GQLResult
53
+ ):
54
+ node: Optional[FileWithUrlFragment]
38
55
 
39
56
 
40
57
  ArtifactCollectionMembershipFileUrls.model_rebuild()
41
58
  ArtifactCollectionMembershipFileUrlsProject.model_rebuild()
42
59
  ArtifactCollectionMembershipFileUrlsProjectArtifactCollection.model_rebuild()
43
60
  ArtifactCollectionMembershipFileUrlsProjectArtifactCollectionArtifactMembership.model_rebuild()
61
+ ArtifactCollectionMembershipFileUrlsProjectArtifactCollectionArtifactMembershipFiles.model_rebuild()
62
+ ArtifactCollectionMembershipFileUrlsProjectArtifactCollectionArtifactMembershipFilesEdges.model_rebuild()
@@ -3,26 +3,26 @@
3
3
 
4
4
  from __future__ import annotations
5
5
 
6
- from typing import Literal, Optional
6
+ from typing import List, Literal, Optional
7
7
 
8
8
  from pydantic import Field
9
9
 
10
- from wandb._pydantic import GQLBase, Typename
10
+ from wandb._pydantic import GQLResult, Typename
11
11
 
12
- from .fragments import FilesFragment
12
+ from .fragments import FileFragment, PageInfoFragment
13
13
 
14
14
 
15
- class ArtifactCollectionMembershipFiles(GQLBase):
15
+ class ArtifactCollectionMembershipFiles(GQLResult):
16
16
  project: Optional[ArtifactCollectionMembershipFilesProject]
17
17
 
18
18
 
19
- class ArtifactCollectionMembershipFilesProject(GQLBase):
19
+ class ArtifactCollectionMembershipFilesProject(GQLResult):
20
20
  artifact_collection: Optional[
21
21
  ArtifactCollectionMembershipFilesProjectArtifactCollection
22
22
  ] = Field(alias="artifactCollection")
23
23
 
24
24
 
25
- class ArtifactCollectionMembershipFilesProjectArtifactCollection(GQLBase):
25
+ class ArtifactCollectionMembershipFilesProjectArtifactCollection(GQLResult):
26
26
  typename__: Typename[
27
27
  Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
28
28
  ]
@@ -32,12 +32,31 @@ class ArtifactCollectionMembershipFilesProjectArtifactCollection(GQLBase):
32
32
 
33
33
 
34
34
  class ArtifactCollectionMembershipFilesProjectArtifactCollectionArtifactMembership(
35
- GQLBase
35
+ GQLResult
36
36
  ):
37
- files: Optional[FilesFragment]
37
+ files: Optional[
38
+ ArtifactCollectionMembershipFilesProjectArtifactCollectionArtifactMembershipFiles
39
+ ]
40
+
41
+
42
+ class ArtifactCollectionMembershipFilesProjectArtifactCollectionArtifactMembershipFiles(
43
+ GQLResult
44
+ ):
45
+ page_info: PageInfoFragment = Field(alias="pageInfo")
46
+ edges: List[
47
+ ArtifactCollectionMembershipFilesProjectArtifactCollectionArtifactMembershipFilesEdges
48
+ ]
49
+
50
+
51
+ class ArtifactCollectionMembershipFilesProjectArtifactCollectionArtifactMembershipFilesEdges(
52
+ GQLResult
53
+ ):
54
+ node: Optional[FileFragment]
38
55
 
39
56
 
40
57
  ArtifactCollectionMembershipFiles.model_rebuild()
41
58
  ArtifactCollectionMembershipFilesProject.model_rebuild()
42
59
  ArtifactCollectionMembershipFilesProjectArtifactCollection.model_rebuild()
43
60
  ArtifactCollectionMembershipFilesProjectArtifactCollectionArtifactMembership.model_rebuild()
61
+ ArtifactCollectionMembershipFilesProjectArtifactCollectionArtifactMembershipFiles.model_rebuild()
62
+ ArtifactCollectionMembershipFilesProjectArtifactCollectionArtifactMembershipFilesEdges.model_rebuild()
@@ -8,40 +8,27 @@ from typing import Literal, Optional, Union
8
8
  from pydantic import Field
9
9
  from typing_extensions import Annotated
10
10
 
11
- from wandb._pydantic import GQLBase, Typename
11
+ from wandb._pydantic import GQLResult, Typename
12
12
 
13
+ from .fragments import RunInfoFragment
13
14
 
14
- class ArtifactCreatedBy(GQLBase):
15
+
16
+ class ArtifactCreatedBy(GQLResult):
15
17
  artifact: Optional[ArtifactCreatedByArtifact]
16
18
 
17
19
 
18
- class ArtifactCreatedByArtifact(GQLBase):
20
+ class ArtifactCreatedByArtifact(GQLResult):
19
21
  created_by: Optional[
20
22
  Annotated[
21
- Union[
22
- ArtifactCreatedByArtifactCreatedByRun,
23
- ArtifactCreatedByArtifactCreatedByUser,
24
- ],
23
+ Union[RunInfoFragment, ArtifactCreatedByArtifactCreatedByUser],
25
24
  Field(discriminator="typename__"),
26
25
  ]
27
26
  ] = Field(alias="createdBy")
28
27
 
29
28
 
30
- class ArtifactCreatedByArtifactCreatedByRun(GQLBase):
31
- typename__: Typename[Literal["Run"]]
32
- name: str
33
- project: Optional[ArtifactCreatedByArtifactCreatedByRunProject]
34
-
35
-
36
- class ArtifactCreatedByArtifactCreatedByRunProject(GQLBase):
37
- name: str
38
- entity_name: str = Field(alias="entityName")
39
-
40
-
41
- class ArtifactCreatedByArtifactCreatedByUser(GQLBase):
29
+ class ArtifactCreatedByArtifactCreatedByUser(GQLResult):
42
30
  typename__: Typename[Literal["User"]]
43
31
 
44
32
 
45
33
  ArtifactCreatedBy.model_rebuild()
46
34
  ArtifactCreatedByArtifact.model_rebuild()
47
- ArtifactCreatedByArtifactCreatedByRun.model_rebuild()
@@ -3,20 +3,33 @@
3
3
 
4
4
  from __future__ import annotations
5
5
 
6
- from typing import Optional
6
+ from typing import List, Optional
7
7
 
8
- from wandb._pydantic import GQLBase
8
+ from pydantic import Field
9
9
 
10
- from .fragments import FileUrlsFragment
10
+ from wandb._pydantic import GQLResult
11
11
 
12
+ from .fragments import FileWithUrlFragment, PageInfoFragment
12
13
 
13
- class ArtifactFileUrls(GQLBase):
14
+
15
+ class ArtifactFileUrls(GQLResult):
14
16
  artifact: Optional[ArtifactFileUrlsArtifact]
15
17
 
16
18
 
17
- class ArtifactFileUrlsArtifact(GQLBase):
18
- files: Optional[FileUrlsFragment]
19
+ class ArtifactFileUrlsArtifact(GQLResult):
20
+ files: Optional[ArtifactFileUrlsArtifactFiles]
21
+
22
+
23
+ class ArtifactFileUrlsArtifactFiles(GQLResult):
24
+ page_info: PageInfoFragment = Field(alias="pageInfo")
25
+ edges: List[ArtifactFileUrlsArtifactFilesEdges]
26
+
27
+
28
+ class ArtifactFileUrlsArtifactFilesEdges(GQLResult):
29
+ node: Optional[FileWithUrlFragment]
19
30
 
20
31
 
21
32
  ArtifactFileUrls.model_rebuild()
22
33
  ArtifactFileUrlsArtifact.model_rebuild()
34
+ ArtifactFileUrlsArtifactFiles.model_rebuild()
35
+ ArtifactFileUrlsArtifactFilesEdges.model_rebuild()
@@ -0,0 +1,26 @@
1
+ # Generated by ariadne-codegen
2
+ # Source: tools/graphql_codegen/artifacts/
3
+
4
+ from __future__ import annotations
5
+
6
+ from typing import Optional
7
+
8
+ from pydantic import Field
9
+
10
+ from wandb._pydantic import GQLResult
11
+
12
+ from .fragments import ArtifactMembershipFragment
13
+
14
+
15
+ class ArtifactMembershipByName(GQLResult):
16
+ project: Optional[ArtifactMembershipByNameProject]
17
+
18
+
19
+ class ArtifactMembershipByNameProject(GQLResult):
20
+ artifact_collection_membership: Optional[ArtifactMembershipFragment] = Field(
21
+ alias="artifactCollectionMembership"
22
+ )
23
+
24
+
25
+ ArtifactMembershipByName.model_rebuild()
26
+ ArtifactMembershipByNameProject.model_rebuild()
@@ -7,22 +7,22 @@ from typing import Optional
7
7
 
8
8
  from pydantic import Field
9
9
 
10
- from wandb._pydantic import GQLBase
10
+ from wandb._pydantic import GQLResult
11
11
 
12
12
 
13
- class ArtifactType(GQLBase):
13
+ class ArtifactType(GQLResult):
14
14
  project: Optional[ArtifactTypeProject]
15
15
 
16
16
 
17
- class ArtifactTypeProject(GQLBase):
17
+ class ArtifactTypeProject(GQLResult):
18
18
  artifact: Optional[ArtifactTypeProjectArtifact]
19
19
 
20
20
 
21
- class ArtifactTypeProjectArtifact(GQLBase):
21
+ class ArtifactTypeProjectArtifact(GQLResult):
22
22
  artifact_type: ArtifactTypeProjectArtifactArtifactType = Field(alias="artifactType")
23
23
 
24
24
 
25
- class ArtifactTypeProjectArtifactArtifactType(GQLBase):
25
+ class ArtifactTypeProjectArtifactArtifactType(GQLResult):
26
26
  name: str
27
27
 
28
28
 
@@ -7,37 +7,28 @@ from typing import List, Optional
7
7
 
8
8
  from pydantic import Field
9
9
 
10
- from wandb._pydantic import GQLBase
10
+ from wandb._pydantic import GQLResult
11
11
 
12
+ from .fragments import RunInfoFragment
12
13
 
13
- class ArtifactUsedBy(GQLBase):
14
+
15
+ class ArtifactUsedBy(GQLResult):
14
16
  artifact: Optional[ArtifactUsedByArtifact]
15
17
 
16
18
 
17
- class ArtifactUsedByArtifact(GQLBase):
19
+ class ArtifactUsedByArtifact(GQLResult):
18
20
  used_by: ArtifactUsedByArtifactUsedBy = Field(alias="usedBy")
19
21
 
20
22
 
21
- class ArtifactUsedByArtifactUsedBy(GQLBase):
23
+ class ArtifactUsedByArtifactUsedBy(GQLResult):
22
24
  edges: List[ArtifactUsedByArtifactUsedByEdges]
23
25
 
24
26
 
25
- class ArtifactUsedByArtifactUsedByEdges(GQLBase):
26
- node: ArtifactUsedByArtifactUsedByEdgesNode
27
-
28
-
29
- class ArtifactUsedByArtifactUsedByEdgesNode(GQLBase):
30
- name: str
31
- project: Optional[ArtifactUsedByArtifactUsedByEdgesNodeProject]
32
-
33
-
34
- class ArtifactUsedByArtifactUsedByEdgesNodeProject(GQLBase):
35
- name: str
36
- entity_name: str = Field(alias="entityName")
27
+ class ArtifactUsedByArtifactUsedByEdges(GQLResult):
28
+ node: RunInfoFragment
37
29
 
38
30
 
39
31
  ArtifactUsedBy.model_rebuild()
40
32
  ArtifactUsedByArtifact.model_rebuild()
41
33
  ArtifactUsedByArtifactUsedBy.model_rebuild()
42
34
  ArtifactUsedByArtifactUsedByEdges.model_rebuild()
43
- ArtifactUsedByArtifactUsedByEdgesNode.model_rebuild()
@@ -3,34 +3,45 @@
3
3
 
4
4
  from __future__ import annotations
5
5
 
6
- from typing import Optional
6
+ from typing import List, Optional
7
7
 
8
8
  from pydantic import Field
9
9
 
10
- from wandb._pydantic import GQLBase
10
+ from wandb._pydantic import GQLResult
11
11
 
12
- from .fragments import FilesFragment
12
+ from .fragments import FileFragment, PageInfoFragment
13
13
 
14
14
 
15
- class ArtifactVersionFiles(GQLBase):
15
+ class ArtifactVersionFiles(GQLResult):
16
16
  project: Optional[ArtifactVersionFilesProject]
17
17
 
18
18
 
19
- class ArtifactVersionFilesProject(GQLBase):
19
+ class ArtifactVersionFilesProject(GQLResult):
20
20
  artifact_type: Optional[ArtifactVersionFilesProjectArtifactType] = Field(
21
21
  alias="artifactType"
22
22
  )
23
23
 
24
24
 
25
- class ArtifactVersionFilesProjectArtifactType(GQLBase):
25
+ class ArtifactVersionFilesProjectArtifactType(GQLResult):
26
26
  artifact: Optional[ArtifactVersionFilesProjectArtifactTypeArtifact]
27
27
 
28
28
 
29
- class ArtifactVersionFilesProjectArtifactTypeArtifact(GQLBase):
30
- files: Optional[FilesFragment]
29
+ class ArtifactVersionFilesProjectArtifactTypeArtifact(GQLResult):
30
+ files: Optional[ArtifactVersionFilesProjectArtifactTypeArtifactFiles]
31
+
32
+
33
+ class ArtifactVersionFilesProjectArtifactTypeArtifactFiles(GQLResult):
34
+ page_info: PageInfoFragment = Field(alias="pageInfo")
35
+ edges: List[ArtifactVersionFilesProjectArtifactTypeArtifactFilesEdges]
36
+
37
+
38
+ class ArtifactVersionFilesProjectArtifactTypeArtifactFilesEdges(GQLResult):
39
+ node: Optional[FileFragment]
31
40
 
32
41
 
33
42
  ArtifactVersionFiles.model_rebuild()
34
43
  ArtifactVersionFilesProject.model_rebuild()
35
44
  ArtifactVersionFilesProjectArtifactType.model_rebuild()
36
45
  ArtifactVersionFilesProjectArtifactTypeArtifact.model_rebuild()
46
+ ArtifactVersionFilesProjectArtifactTypeArtifactFiles.model_rebuild()
47
+ ArtifactVersionFilesProjectArtifactTypeArtifactFilesEdges.model_rebuild()
@@ -7,14 +7,14 @@ from typing import Optional
7
7
 
8
8
  from pydantic import Field
9
9
 
10
- from wandb._pydantic import GQLBase
10
+ from wandb._pydantic import GQLResult
11
11
 
12
12
 
13
- class DeleteAliases(GQLBase):
13
+ class DeleteAliases(GQLResult):
14
14
  delete_aliases: Optional[DeleteAliasesDeleteAliases] = Field(alias="deleteAliases")
15
15
 
16
16
 
17
- class DeleteAliasesDeleteAliases(GQLBase):
17
+ class DeleteAliasesDeleteAliases(GQLResult):
18
18
  success: bool
19
19
 
20
20
 
@@ -7,20 +7,20 @@ from typing import Optional
7
7
 
8
8
  from pydantic import Field
9
9
 
10
- from wandb._pydantic import GQLBase, GQLId
10
+ from wandb._pydantic import GQLId, GQLResult
11
11
 
12
12
 
13
- class DeleteArtifact(GQLBase):
13
+ class DeleteArtifact(GQLResult):
14
14
  delete_artifact: Optional[DeleteArtifactDeleteArtifact] = Field(
15
15
  alias="deleteArtifact"
16
16
  )
17
17
 
18
18
 
19
- class DeleteArtifactDeleteArtifact(GQLBase):
19
+ class DeleteArtifactDeleteArtifact(GQLResult):
20
20
  artifact: DeleteArtifactDeleteArtifactArtifact
21
21
 
22
22
 
23
- class DeleteArtifactDeleteArtifactArtifact(GQLBase):
23
+ class DeleteArtifactDeleteArtifactArtifact(GQLResult):
24
24
  id: GQLId
25
25
 
26
26
 
@@ -0,0 +1,23 @@
1
+ # Generated by ariadne-codegen
2
+ # Source: tools/graphql_codegen/artifacts/
3
+
4
+ from __future__ import annotations
5
+
6
+ from typing import Optional
7
+
8
+ from pydantic import Field
9
+
10
+ from wandb._pydantic import GQLResult
11
+
12
+
13
+ class DeleteArtifactCollectionTags(GQLResult):
14
+ delete_artifact_collection_tag_assignments: Optional[
15
+ DeleteArtifactCollectionTagsDeleteArtifactCollectionTagAssignments
16
+ ] = Field(alias="deleteArtifactCollectionTagAssignments")
17
+
18
+
19
+ class DeleteArtifactCollectionTagsDeleteArtifactCollectionTagAssignments(GQLResult):
20
+ success: bool
21
+
22
+
23
+ DeleteArtifactCollectionTags.model_rebuild()
@@ -7,24 +7,24 @@ from typing import Literal, Optional
7
7
 
8
8
  from pydantic import Field
9
9
 
10
- from wandb._pydantic import GQLBase, Typename
10
+ from wandb._pydantic import GQLResult, Typename
11
11
 
12
12
  from .enums import ArtifactCollectionState
13
13
 
14
14
 
15
- class DeleteArtifactPortfolio(GQLBase):
15
+ class DeleteArtifactPortfolio(GQLResult):
16
16
  delete_artifact_portfolio: Optional[
17
17
  DeleteArtifactPortfolioDeleteArtifactPortfolio
18
18
  ] = Field(alias="deleteArtifactPortfolio")
19
19
 
20
20
 
21
- class DeleteArtifactPortfolioDeleteArtifactPortfolio(GQLBase):
21
+ class DeleteArtifactPortfolioDeleteArtifactPortfolio(GQLResult):
22
22
  artifact_collection: DeleteArtifactPortfolioDeleteArtifactPortfolioArtifactCollection = Field(
23
23
  alias="artifactCollection"
24
24
  )
25
25
 
26
26
 
27
- class DeleteArtifactPortfolioDeleteArtifactPortfolioArtifactCollection(GQLBase):
27
+ class DeleteArtifactPortfolioDeleteArtifactPortfolioArtifactCollection(GQLResult):
28
28
  typename__: Typename[
29
29
  Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
30
30
  ]
@@ -7,24 +7,24 @@ from typing import Literal, Optional
7
7
 
8
8
  from pydantic import Field
9
9
 
10
- from wandb._pydantic import GQLBase, Typename
10
+ from wandb._pydantic import GQLResult, Typename
11
11
 
12
12
  from .enums import ArtifactCollectionState
13
13
 
14
14
 
15
- class DeleteArtifactSequence(GQLBase):
15
+ class DeleteArtifactSequence(GQLResult):
16
16
  delete_artifact_sequence: Optional[DeleteArtifactSequenceDeleteArtifactSequence] = (
17
17
  Field(alias="deleteArtifactSequence")
18
18
  )
19
19
 
20
20
 
21
- class DeleteArtifactSequenceDeleteArtifactSequence(GQLBase):
21
+ class DeleteArtifactSequenceDeleteArtifactSequence(GQLResult):
22
22
  artifact_collection: DeleteArtifactSequenceDeleteArtifactSequenceArtifactCollection = Field(
23
23
  alias="artifactCollection"
24
24
  )
25
25
 
26
26
 
27
- class DeleteArtifactSequenceDeleteArtifactSequenceArtifactCollection(GQLBase):
27
+ class DeleteArtifactSequenceDeleteArtifactSequenceArtifactCollection(GQLResult):
28
28
  typename__: Typename[
29
29
  Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
30
30
  ]
@@ -0,0 +1,21 @@
1
+ # Generated by ariadne-codegen
2
+ # Source: tools/graphql_codegen/artifacts/
3
+
4
+ from __future__ import annotations
5
+
6
+ from typing import Optional
7
+
8
+ from pydantic import Field
9
+
10
+ from wandb._pydantic import GQLResult
11
+
12
+
13
+ class DeleteRegistry(GQLResult):
14
+ delete_model: Optional[DeleteRegistryDeleteModel] = Field(alias="deleteModel")
15
+
16
+
17
+ class DeleteRegistryDeleteModel(GQLResult):
18
+ success: Optional[bool]
19
+
20
+
21
+ DeleteRegistry.model_rebuild()
@@ -7,32 +7,20 @@ from typing import Optional
7
7
 
8
8
  from pydantic import Field
9
9
 
10
- from wandb._pydantic import GQLBase
10
+ from wandb._pydantic import GQLResult
11
11
 
12
+ from .fragments import DeferredManifestFragment
12
13
 
13
- class FetchArtifactManifest(GQLBase):
14
- project: Optional[FetchArtifactManifestProject]
15
14
 
15
+ class FetchArtifactManifest(GQLResult):
16
+ artifact: Optional[FetchArtifactManifestArtifact]
16
17
 
17
- class FetchArtifactManifestProject(GQLBase):
18
- artifact: Optional[FetchArtifactManifestProjectArtifact]
19
18
 
20
-
21
- class FetchArtifactManifestProjectArtifact(GQLBase):
22
- current_manifest: Optional[FetchArtifactManifestProjectArtifactCurrentManifest] = (
23
- Field(alias="currentManifest")
19
+ class FetchArtifactManifestArtifact(GQLResult):
20
+ current_manifest: Optional[DeferredManifestFragment] = Field(
21
+ alias="currentManifest"
24
22
  )
25
23
 
26
24
 
27
- class FetchArtifactManifestProjectArtifactCurrentManifest(GQLBase):
28
- file: FetchArtifactManifestProjectArtifactCurrentManifestFile
29
-
30
-
31
- class FetchArtifactManifestProjectArtifactCurrentManifestFile(GQLBase):
32
- direct_url: str = Field(alias="directUrl")
33
-
34
-
35
25
  FetchArtifactManifest.model_rebuild()
36
- FetchArtifactManifestProject.model_rebuild()
37
- FetchArtifactManifestProjectArtifact.model_rebuild()
38
- FetchArtifactManifestProjectArtifactCurrentManifest.model_rebuild()
26
+ FetchArtifactManifestArtifact.model_rebuild()
@@ -3,60 +3,39 @@
3
3
 
4
4
  from __future__ import annotations
5
5
 
6
- from typing import List, Literal, Optional
6
+ from typing import List, Optional
7
7
 
8
8
  from pydantic import Field
9
9
 
10
- from wandb._pydantic import GQLBase, Typename
10
+ from wandb._pydantic import GQLResult
11
11
 
12
+ from .fragments import ArtifactAliasFragment, CollectionInfoFragment
12
13
 
13
- class FetchLinkedArtifacts(GQLBase):
14
+
15
+ class FetchLinkedArtifacts(GQLResult):
14
16
  artifact: Optional[FetchLinkedArtifactsArtifact]
15
17
 
16
18
 
17
- class FetchLinkedArtifactsArtifact(GQLBase):
19
+ class FetchLinkedArtifactsArtifact(GQLResult):
18
20
  artifact_memberships: FetchLinkedArtifactsArtifactArtifactMemberships = Field(
19
21
  alias="artifactMemberships"
20
22
  )
21
23
 
22
24
 
23
- class FetchLinkedArtifactsArtifactArtifactMemberships(GQLBase):
25
+ class FetchLinkedArtifactsArtifactArtifactMemberships(GQLResult):
24
26
  edges: List[FetchLinkedArtifactsArtifactArtifactMembershipsEdges]
25
27
 
26
28
 
27
- class FetchLinkedArtifactsArtifactArtifactMembershipsEdges(GQLBase):
29
+ class FetchLinkedArtifactsArtifactArtifactMembershipsEdges(GQLResult):
28
30
  node: Optional[FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNode]
29
31
 
30
32
 
31
- class FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNode(GQLBase):
32
- aliases: List[FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNodeAliases]
33
+ class FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNode(GQLResult):
33
34
  version_index: Optional[int] = Field(alias="versionIndex")
34
- artifact_collection: Optional[
35
- FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNodeArtifactCollection
36
- ] = Field(alias="artifactCollection")
37
-
38
-
39
- class FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNodeAliases(GQLBase):
40
- alias: str
41
-
42
-
43
- class FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNodeArtifactCollection(
44
- GQLBase
45
- ):
46
- project: Optional[
47
- FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNodeArtifactCollectionProject
48
- ]
49
- name: str
50
- typename__: Typename[
51
- Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
52
- ]
53
-
54
-
55
- class FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNodeArtifactCollectionProject(
56
- GQLBase
57
- ):
58
- entity_name: str = Field(alias="entityName")
59
- name: str
35
+ aliases: List[ArtifactAliasFragment]
36
+ artifact_collection: Optional[CollectionInfoFragment] = Field(
37
+ alias="artifactCollection"
38
+ )
60
39
 
61
40
 
62
41
  FetchLinkedArtifacts.model_rebuild()
@@ -64,4 +43,3 @@ FetchLinkedArtifactsArtifact.model_rebuild()
64
43
  FetchLinkedArtifactsArtifactArtifactMemberships.model_rebuild()
65
44
  FetchLinkedArtifactsArtifactArtifactMembershipsEdges.model_rebuild()
66
45
  FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNode.model_rebuild()
67
- FetchLinkedArtifactsArtifactArtifactMembershipsEdgesNodeArtifactCollection.model_rebuild()