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.
- wandb/__init__.py +1 -1
- wandb/__init__.pyi +1 -1
- wandb/apis/public/api.py +1 -2
- wandb/apis/public/artifacts.py +3 -5
- wandb/apis/public/registries/_utils.py +14 -16
- wandb/apis/public/registries/registries_search.py +176 -289
- wandb/apis/public/reports.py +13 -10
- wandb/automations/_generated/delete_automation.py +1 -3
- wandb/automations/_generated/enums.py +13 -11
- wandb/bin/gpu_stats.exe +0 -0
- wandb/bin/wandb-core +0 -0
- wandb/cli/cli.py +47 -2
- wandb/integration/metaflow/data_pandas.py +2 -2
- wandb/integration/metaflow/data_pytorch.py +75 -0
- wandb/integration/metaflow/data_sklearn.py +76 -0
- wandb/integration/metaflow/metaflow.py +16 -87
- wandb/integration/weave/__init__.py +6 -0
- wandb/integration/weave/interface.py +49 -0
- wandb/integration/weave/weave.py +63 -0
- wandb/proto/v3/wandb_internal_pb2.py +3 -2
- wandb/proto/v4/wandb_internal_pb2.py +2 -2
- wandb/proto/v5/wandb_internal_pb2.py +2 -2
- wandb/proto/v6/wandb_internal_pb2.py +2 -2
- wandb/sdk/artifacts/_factories.py +17 -0
- wandb/sdk/artifacts/_generated/__init__.py +221 -13
- wandb/sdk/artifacts/_generated/artifact_by_id.py +17 -0
- wandb/sdk/artifacts/_generated/artifact_by_name.py +22 -0
- wandb/sdk/artifacts/_generated/artifact_collection_membership_file_urls.py +43 -0
- wandb/sdk/artifacts/_generated/artifact_created_by.py +47 -0
- wandb/sdk/artifacts/_generated/artifact_file_urls.py +22 -0
- wandb/sdk/artifacts/_generated/artifact_type.py +31 -0
- wandb/sdk/artifacts/_generated/artifact_used_by.py +43 -0
- wandb/sdk/artifacts/_generated/artifact_via_membership_by_name.py +26 -0
- wandb/sdk/artifacts/_generated/delete_artifact.py +28 -0
- wandb/sdk/artifacts/_generated/enums.py +5 -0
- wandb/sdk/artifacts/_generated/fetch_artifact_manifest.py +38 -0
- wandb/sdk/artifacts/_generated/fetch_registries.py +32 -0
- wandb/sdk/artifacts/_generated/fragments.py +279 -41
- wandb/sdk/artifacts/_generated/link_artifact.py +6 -0
- wandb/sdk/artifacts/_generated/operations.py +654 -51
- wandb/sdk/artifacts/_generated/registry_collections.py +34 -0
- wandb/sdk/artifacts/_generated/registry_versions.py +34 -0
- wandb/sdk/artifacts/_generated/unlink_artifact.py +25 -0
- wandb/sdk/artifacts/_graphql_fragments.py +3 -86
- wandb/sdk/artifacts/_validators.py +6 -4
- wandb/sdk/artifacts/artifact.py +410 -547
- wandb/sdk/artifacts/artifact_file_cache.py +11 -7
- wandb/sdk/artifacts/artifact_manifest.py +10 -9
- wandb/sdk/artifacts/artifact_manifest_entry.py +15 -18
- wandb/sdk/artifacts/artifact_manifests/artifact_manifest_v1.py +5 -3
- wandb/sdk/artifacts/storage_handlers/gcs_handler.py +1 -1
- wandb/sdk/artifacts/storage_handlers/http_handler.py +1 -1
- wandb/sdk/artifacts/storage_handlers/s3_handler.py +1 -1
- wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py +1 -1
- wandb/sdk/data_types/video.py +2 -2
- wandb/sdk/interface/interface_queue.py +1 -4
- wandb/sdk/interface/interface_shared.py +26 -37
- wandb/sdk/interface/interface_sock.py +24 -14
- wandb/sdk/internal/settings_static.py +2 -3
- wandb/sdk/launch/create_job.py +12 -1
- wandb/sdk/launch/inputs/internal.py +25 -24
- wandb/sdk/launch/inputs/schema.py +31 -1
- wandb/sdk/launch/runner/kubernetes_runner.py +24 -29
- wandb/sdk/lib/asyncio_compat.py +16 -16
- wandb/sdk/lib/asyncio_manager.py +252 -0
- wandb/sdk/lib/hashutil.py +13 -4
- wandb/sdk/lib/paths.py +23 -21
- wandb/sdk/lib/printer.py +2 -2
- wandb/sdk/lib/printer_asyncio.py +3 -1
- wandb/sdk/lib/retry.py +185 -78
- wandb/sdk/lib/service/service_client.py +106 -0
- wandb/sdk/lib/service/service_connection.py +20 -26
- wandb/sdk/lib/service/service_token.py +30 -13
- wandb/sdk/mailbox/mailbox.py +13 -5
- wandb/sdk/mailbox/mailbox_handle.py +22 -13
- wandb/sdk/mailbox/response_handle.py +42 -106
- wandb/sdk/mailbox/wait_with_progress.py +7 -42
- wandb/sdk/wandb_init.py +11 -25
- wandb/sdk/wandb_login.py +1 -1
- wandb/sdk/wandb_run.py +92 -56
- wandb/sdk/wandb_settings.py +45 -32
- wandb/sdk/wandb_setup.py +176 -96
- wandb/util.py +1 -1
- {wandb-0.21.1.dist-info → wandb-0.21.3.dist-info}/METADATA +2 -2
- {wandb-0.21.1.dist-info → wandb-0.21.3.dist-info}/RECORD +88 -72
- wandb/sdk/interface/interface_relay.py +0 -38
- wandb/sdk/interface/router.py +0 -89
- wandb/sdk/interface/router_queue.py +0 -43
- wandb/sdk/interface/router_relay.py +0 -50
- wandb/sdk/interface/router_sock.py +0 -32
- wandb/sdk/lib/sock_client.py +0 -232
- {wandb-0.21.1.dist-info → wandb-0.21.3.dist-info}/WHEEL +0 -0
- {wandb-0.21.1.dist-info → wandb-0.21.3.dist-info}/entry_points.txt +0 -0
- {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()
|
@@ -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()
|