wandb 0.21.1__py3-none-win32.whl → 0.21.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 (94) hide show
  1. wandb/__init__.py +1 -1
  2. wandb/__init__.pyi +1 -1
  3. wandb/apis/public/api.py +1 -2
  4. wandb/apis/public/artifacts.py +3 -5
  5. wandb/apis/public/registries/_utils.py +14 -16
  6. wandb/apis/public/registries/registries_search.py +176 -289
  7. wandb/apis/public/reports.py +13 -10
  8. wandb/automations/_generated/delete_automation.py +1 -3
  9. wandb/automations/_generated/enums.py +13 -11
  10. wandb/bin/gpu_stats.exe +0 -0
  11. wandb/bin/wandb-core +0 -0
  12. wandb/cli/cli.py +47 -2
  13. wandb/integration/metaflow/data_pandas.py +2 -2
  14. wandb/integration/metaflow/data_pytorch.py +75 -0
  15. wandb/integration/metaflow/data_sklearn.py +76 -0
  16. wandb/integration/metaflow/metaflow.py +16 -87
  17. wandb/integration/weave/__init__.py +6 -0
  18. wandb/integration/weave/interface.py +49 -0
  19. wandb/integration/weave/weave.py +63 -0
  20. wandb/proto/v3/wandb_internal_pb2.py +3 -2
  21. wandb/proto/v4/wandb_internal_pb2.py +2 -2
  22. wandb/proto/v5/wandb_internal_pb2.py +2 -2
  23. wandb/proto/v6/wandb_internal_pb2.py +2 -2
  24. wandb/sdk/artifacts/_factories.py +17 -0
  25. wandb/sdk/artifacts/_generated/__init__.py +221 -13
  26. wandb/sdk/artifacts/_generated/artifact_by_id.py +17 -0
  27. wandb/sdk/artifacts/_generated/artifact_by_name.py +22 -0
  28. wandb/sdk/artifacts/_generated/artifact_collection_membership_file_urls.py +43 -0
  29. wandb/sdk/artifacts/_generated/artifact_created_by.py +47 -0
  30. wandb/sdk/artifacts/_generated/artifact_file_urls.py +22 -0
  31. wandb/sdk/artifacts/_generated/artifact_type.py +31 -0
  32. wandb/sdk/artifacts/_generated/artifact_used_by.py +43 -0
  33. wandb/sdk/artifacts/_generated/artifact_via_membership_by_name.py +26 -0
  34. wandb/sdk/artifacts/_generated/delete_artifact.py +28 -0
  35. wandb/sdk/artifacts/_generated/enums.py +5 -0
  36. wandb/sdk/artifacts/_generated/fetch_artifact_manifest.py +38 -0
  37. wandb/sdk/artifacts/_generated/fetch_registries.py +32 -0
  38. wandb/sdk/artifacts/_generated/fragments.py +279 -41
  39. wandb/sdk/artifacts/_generated/link_artifact.py +6 -0
  40. wandb/sdk/artifacts/_generated/operations.py +654 -51
  41. wandb/sdk/artifacts/_generated/registry_collections.py +34 -0
  42. wandb/sdk/artifacts/_generated/registry_versions.py +34 -0
  43. wandb/sdk/artifacts/_generated/unlink_artifact.py +25 -0
  44. wandb/sdk/artifacts/_graphql_fragments.py +3 -86
  45. wandb/sdk/artifacts/_validators.py +6 -4
  46. wandb/sdk/artifacts/artifact.py +410 -547
  47. wandb/sdk/artifacts/artifact_file_cache.py +11 -7
  48. wandb/sdk/artifacts/artifact_manifest.py +10 -9
  49. wandb/sdk/artifacts/artifact_manifest_entry.py +15 -18
  50. wandb/sdk/artifacts/artifact_manifests/artifact_manifest_v1.py +5 -3
  51. wandb/sdk/artifacts/storage_handlers/gcs_handler.py +1 -1
  52. wandb/sdk/artifacts/storage_handlers/http_handler.py +1 -1
  53. wandb/sdk/artifacts/storage_handlers/s3_handler.py +1 -1
  54. wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py +1 -1
  55. wandb/sdk/data_types/video.py +2 -2
  56. wandb/sdk/interface/interface_queue.py +1 -4
  57. wandb/sdk/interface/interface_shared.py +26 -37
  58. wandb/sdk/interface/interface_sock.py +24 -14
  59. wandb/sdk/internal/settings_static.py +2 -3
  60. wandb/sdk/launch/create_job.py +12 -1
  61. wandb/sdk/launch/inputs/internal.py +25 -24
  62. wandb/sdk/launch/inputs/schema.py +31 -1
  63. wandb/sdk/launch/runner/kubernetes_runner.py +24 -29
  64. wandb/sdk/lib/asyncio_compat.py +16 -16
  65. wandb/sdk/lib/asyncio_manager.py +252 -0
  66. wandb/sdk/lib/hashutil.py +13 -4
  67. wandb/sdk/lib/paths.py +23 -21
  68. wandb/sdk/lib/printer.py +2 -2
  69. wandb/sdk/lib/printer_asyncio.py +3 -1
  70. wandb/sdk/lib/retry.py +185 -78
  71. wandb/sdk/lib/service/service_client.py +106 -0
  72. wandb/sdk/lib/service/service_connection.py +20 -26
  73. wandb/sdk/lib/service/service_token.py +30 -13
  74. wandb/sdk/mailbox/mailbox.py +13 -5
  75. wandb/sdk/mailbox/mailbox_handle.py +22 -13
  76. wandb/sdk/mailbox/response_handle.py +42 -106
  77. wandb/sdk/mailbox/wait_with_progress.py +7 -42
  78. wandb/sdk/wandb_init.py +11 -25
  79. wandb/sdk/wandb_login.py +1 -1
  80. wandb/sdk/wandb_run.py +92 -56
  81. wandb/sdk/wandb_settings.py +45 -32
  82. wandb/sdk/wandb_setup.py +176 -96
  83. wandb/util.py +1 -1
  84. {wandb-0.21.1.dist-info → wandb-0.21.3.dist-info}/METADATA +2 -2
  85. {wandb-0.21.1.dist-info → wandb-0.21.3.dist-info}/RECORD +88 -72
  86. wandb/sdk/interface/interface_relay.py +0 -38
  87. wandb/sdk/interface/router.py +0 -89
  88. wandb/sdk/interface/router_queue.py +0 -43
  89. wandb/sdk/interface/router_relay.py +0 -50
  90. wandb/sdk/interface/router_sock.py +0 -32
  91. wandb/sdk/lib/sock_client.py +0 -232
  92. {wandb-0.21.1.dist-info → wandb-0.21.3.dist-info}/WHEEL +0 -0
  93. {wandb-0.21.1.dist-info → wandb-0.21.3.dist-info}/entry_points.txt +0 -0
  94. {wandb-0.21.1.dist-info → wandb-0.21.3.dist-info}/licenses/LICENSE +0 -0
@@ -0,0 +1,43 @@
1
+ # Generated by ariadne-codegen
2
+ # Source: tools/graphql_codegen/artifacts/
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
+ from .fragments import FileUrlsFragment
13
+
14
+
15
+ class ArtifactCollectionMembershipFileUrls(GQLBase):
16
+ project: Optional[ArtifactCollectionMembershipFileUrlsProject]
17
+
18
+
19
+ class ArtifactCollectionMembershipFileUrlsProject(GQLBase):
20
+ artifact_collection: Optional[
21
+ ArtifactCollectionMembershipFileUrlsProjectArtifactCollection
22
+ ] = Field(alias="artifactCollection")
23
+
24
+
25
+ class ArtifactCollectionMembershipFileUrlsProjectArtifactCollection(GQLBase):
26
+ typename__: Typename[
27
+ Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
28
+ ]
29
+ artifact_membership: Optional[
30
+ ArtifactCollectionMembershipFileUrlsProjectArtifactCollectionArtifactMembership
31
+ ] = Field(alias="artifactMembership")
32
+
33
+
34
+ class ArtifactCollectionMembershipFileUrlsProjectArtifactCollectionArtifactMembership(
35
+ GQLBase
36
+ ):
37
+ files: Optional[FileUrlsFragment]
38
+
39
+
40
+ ArtifactCollectionMembershipFileUrls.model_rebuild()
41
+ ArtifactCollectionMembershipFileUrlsProject.model_rebuild()
42
+ ArtifactCollectionMembershipFileUrlsProjectArtifactCollection.model_rebuild()
43
+ ArtifactCollectionMembershipFileUrlsProjectArtifactCollectionArtifactMembership.model_rebuild()
@@ -0,0 +1,47 @@
1
+ # Generated by ariadne-codegen
2
+ # Source: tools/graphql_codegen/artifacts/
3
+
4
+ from __future__ import annotations
5
+
6
+ from typing import Literal, Optional, Union
7
+
8
+ from pydantic import Field
9
+ from typing_extensions import Annotated
10
+
11
+ from wandb._pydantic import GQLBase, Typename
12
+
13
+
14
+ class ArtifactCreatedBy(GQLBase):
15
+ artifact: Optional[ArtifactCreatedByArtifact]
16
+
17
+
18
+ class ArtifactCreatedByArtifact(GQLBase):
19
+ created_by: Optional[
20
+ Annotated[
21
+ Union[
22
+ ArtifactCreatedByArtifactCreatedByRun,
23
+ ArtifactCreatedByArtifactCreatedByUser,
24
+ ],
25
+ Field(discriminator="typename__"),
26
+ ]
27
+ ] = Field(alias="createdBy")
28
+
29
+
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):
42
+ typename__: Typename[Literal["User"]]
43
+
44
+
45
+ ArtifactCreatedBy.model_rebuild()
46
+ ArtifactCreatedByArtifact.model_rebuild()
47
+ ArtifactCreatedByArtifactCreatedByRun.model_rebuild()
@@ -0,0 +1,22 @@
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 wandb._pydantic import GQLBase
9
+
10
+ from .fragments import FileUrlsFragment
11
+
12
+
13
+ class ArtifactFileUrls(GQLBase):
14
+ artifact: Optional[ArtifactFileUrlsArtifact]
15
+
16
+
17
+ class ArtifactFileUrlsArtifact(GQLBase):
18
+ files: Optional[FileUrlsFragment]
19
+
20
+
21
+ ArtifactFileUrls.model_rebuild()
22
+ ArtifactFileUrlsArtifact.model_rebuild()
@@ -0,0 +1,31 @@
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 GQLBase
11
+
12
+
13
+ class ArtifactType(GQLBase):
14
+ project: Optional[ArtifactTypeProject]
15
+
16
+
17
+ class ArtifactTypeProject(GQLBase):
18
+ artifact: Optional[ArtifactTypeProjectArtifact]
19
+
20
+
21
+ class ArtifactTypeProjectArtifact(GQLBase):
22
+ artifact_type: ArtifactTypeProjectArtifactArtifactType = Field(alias="artifactType")
23
+
24
+
25
+ class ArtifactTypeProjectArtifactArtifactType(GQLBase):
26
+ name: str
27
+
28
+
29
+ ArtifactType.model_rebuild()
30
+ ArtifactTypeProject.model_rebuild()
31
+ ArtifactTypeProjectArtifact.model_rebuild()
@@ -0,0 +1,43 @@
1
+ # Generated by ariadne-codegen
2
+ # Source: tools/graphql_codegen/artifacts/
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
11
+
12
+
13
+ class ArtifactUsedBy(GQLBase):
14
+ artifact: Optional[ArtifactUsedByArtifact]
15
+
16
+
17
+ class ArtifactUsedByArtifact(GQLBase):
18
+ used_by: ArtifactUsedByArtifactUsedBy = Field(alias="usedBy")
19
+
20
+
21
+ class ArtifactUsedByArtifactUsedBy(GQLBase):
22
+ edges: List[ArtifactUsedByArtifactUsedByEdges]
23
+
24
+
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")
37
+
38
+
39
+ ArtifactUsedBy.model_rebuild()
40
+ ArtifactUsedByArtifact.model_rebuild()
41
+ ArtifactUsedByArtifactUsedBy.model_rebuild()
42
+ ArtifactUsedByArtifactUsedByEdges.model_rebuild()
43
+ ArtifactUsedByArtifactUsedByEdgesNode.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 GQLBase
11
+
12
+ from .fragments import MembershipWithArtifact
13
+
14
+
15
+ class ArtifactViaMembershipByName(GQLBase):
16
+ project: Optional[ArtifactViaMembershipByNameProject]
17
+
18
+
19
+ class ArtifactViaMembershipByNameProject(GQLBase):
20
+ artifact_collection_membership: Optional[MembershipWithArtifact] = Field(
21
+ alias="artifactCollectionMembership"
22
+ )
23
+
24
+
25
+ ArtifactViaMembershipByName.model_rebuild()
26
+ ArtifactViaMembershipByNameProject.model_rebuild()
@@ -0,0 +1,28 @@
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 GQLBase, GQLId
11
+
12
+
13
+ class DeleteArtifact(GQLBase):
14
+ delete_artifact: Optional[DeleteArtifactDeleteArtifact] = Field(
15
+ alias="deleteArtifact"
16
+ )
17
+
18
+
19
+ class DeleteArtifactDeleteArtifact(GQLBase):
20
+ artifact: DeleteArtifactDeleteArtifactArtifact
21
+
22
+
23
+ class DeleteArtifactDeleteArtifactArtifact(GQLBase):
24
+ id: GQLId
25
+
26
+
27
+ DeleteArtifact.model_rebuild()
28
+ DeleteArtifactDeleteArtifact.model_rebuild()
@@ -6,6 +6,11 @@ from __future__ import annotations
6
6
  from enum import Enum
7
7
 
8
8
 
9
+ class ArtifactCollectionType(str, Enum):
10
+ SEQUENCE = "SEQUENCE"
11
+ PORTFOLIO = "PORTFOLIO"
12
+
13
+
9
14
  class ArtifactState(str, Enum):
10
15
  PENDING = "PENDING"
11
16
  COMMITTED = "COMMITTED"
@@ -0,0 +1,38 @@
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 GQLBase
11
+
12
+
13
+ class FetchArtifactManifest(GQLBase):
14
+ project: Optional[FetchArtifactManifestProject]
15
+
16
+
17
+ class FetchArtifactManifestProject(GQLBase):
18
+ artifact: Optional[FetchArtifactManifestProjectArtifact]
19
+
20
+
21
+ class FetchArtifactManifestProjectArtifact(GQLBase):
22
+ current_manifest: Optional[FetchArtifactManifestProjectArtifactCurrentManifest] = (
23
+ Field(alias="currentManifest")
24
+ )
25
+
26
+
27
+ class FetchArtifactManifestProjectArtifactCurrentManifest(GQLBase):
28
+ file: FetchArtifactManifestProjectArtifactCurrentManifestFile
29
+
30
+
31
+ class FetchArtifactManifestProjectArtifactCurrentManifestFile(GQLBase):
32
+ direct_url: str = Field(alias="directUrl")
33
+
34
+
35
+ FetchArtifactManifest.model_rebuild()
36
+ FetchArtifactManifestProject.model_rebuild()
37
+ FetchArtifactManifestProjectArtifact.model_rebuild()
38
+ FetchArtifactManifestProjectArtifactCurrentManifest.model_rebuild()
@@ -0,0 +1,32 @@
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 GQLBase
11
+
12
+ from .fragments import RegistriesPage
13
+
14
+
15
+ class FetchRegistries(GQLBase):
16
+ organization: Optional[FetchRegistriesOrganization]
17
+
18
+
19
+ class FetchRegistriesOrganization(GQLBase):
20
+ org_entity: Optional[FetchRegistriesOrganizationOrgEntity] = Field(
21
+ alias="orgEntity"
22
+ )
23
+
24
+
25
+ class FetchRegistriesOrganizationOrgEntity(GQLBase):
26
+ name: str
27
+ projects: Optional[RegistriesPage]
28
+
29
+
30
+ FetchRegistries.model_rebuild()
31
+ FetchRegistriesOrganization.model_rebuild()
32
+ FetchRegistriesOrganizationOrgEntity.model_rebuild()