wandb 0.21.4__py3-none-musllinux_1_2_aarch64.whl → 0.22.1__py3-none-musllinux_1_2_aarch64.whl

wandb/automations/_generated/__init__.py

@@ -1,99 +1,5 @@
 # Generated by ariadne-codegen
 
-from .create_automation import CreateAutomation
-from .create_generic_webhook_integration import (
-    CreateGenericWebhookIntegration,
-    CreateGenericWebhookIntegrationCreateGenericWebhookIntegration,
-    CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationGenericWebhookIntegration,
-    CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationIntegration,
-)
-from .delete_automation import DeleteAutomation
-from .enums import (
-    AlertSeverity,
-    EventTriggeringConditionType,
-    TriggeredActionType,
-    TriggerScopeType,
-)
-from .fragments import (
-    ArtifactPortfolioScopeFields,
-    ArtifactSequenceScopeFields,
-    FilterEventFields,
-    GenericWebhookActionFields,
-    GenericWebhookActionFieldsIntegrationGenericWebhookIntegration,
-    GenericWebhookActionFieldsIntegrationIntegration,
-    GenericWebhookIntegrationConnectionFields,
-    GenericWebhookIntegrationConnectionFieldsEdges,
-    GenericWebhookIntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration,
-    GenericWebhookIntegrationConnectionFieldsEdgesNodeIntegration,
-    GenericWebhookIntegrationFields,
-    IntegrationConnectionFields,
-    IntegrationConnectionFieldsEdges,
-    IntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration,
-    IntegrationConnectionFieldsEdgesNodeIntegration,
-    IntegrationConnectionFieldsEdgesNodeSlackIntegration,
-    NoOpActionFields,
-    NotificationActionFields,
-    NotificationActionFieldsIntegrationIntegration,
-    NotificationActionFieldsIntegrationSlackIntegration,
-    PageInfoFields,
-    ProjectConnectionFields,
-    ProjectConnectionFieldsEdges,
-    ProjectConnectionFieldsEdgesNode,
-    ProjectScopeFields,
-    QueueJobActionFields,
-    QueueJobActionFieldsQueue,
-    SlackIntegrationConnectionFields,
-    SlackIntegrationConnectionFieldsEdges,
-    SlackIntegrationConnectionFieldsEdgesNodeIntegration,
-    SlackIntegrationConnectionFieldsEdgesNodeSlackIntegration,
-    SlackIntegrationFields,
-    TriggerFields,
-    TriggerFieldsActionGenericWebhookTriggeredAction,
-    TriggerFieldsActionNoOpTriggeredAction,
-    TriggerFieldsActionNotificationTriggeredAction,
-    TriggerFieldsActionQueueJobTriggeredAction,
-    TriggerFieldsEventFilterEventTriggeringCondition,
-    TriggerFieldsScopeArtifactPortfolio,
-    TriggerFieldsScopeArtifactSequence,
-    TriggerFieldsScopeProject,
-)
-from .generic_webhook_integrations_by_entity import (
-    GenericWebhookIntegrationsByEntity,
-    GenericWebhookIntegrationsByEntityEntity,
-)
-from .get_automations import GetAutomations, GetAutomationsSearchScope
-from .get_automations_by_entity import (
-    GetAutomationsByEntity,
-    GetAutomationsByEntitySearchScope,
-)
-from .input_types import (
-    CreateFilterTriggerInput,
-    CreateGenericWebhookIntegrationInput,
-    GenericWebhookActionInput,
-    NoOpTriggeredActionInput,
-    NotificationActionInput,
-    QueueJobActionInput,
-    TriggeredActionConfig,
-    UpdateFilterTriggerInput,
-)
-from .integrations_by_entity import IntegrationsByEntity, IntegrationsByEntityEntity
-from .operations import (
-    CREATE_AUTOMATION_GQL,
-    CREATE_GENERIC_WEBHOOK_INTEGRATION_GQL,
-    DELETE_AUTOMATION_GQL,
-    GENERIC_WEBHOOK_INTEGRATIONS_BY_ENTITY_GQL,
-    GET_AUTOMATIONS_BY_ENTITY_GQL,
-    GET_AUTOMATIONS_GQL,
-    INTEGRATIONS_BY_ENTITY_GQL,
-    SLACK_INTEGRATIONS_BY_ENTITY_GQL,
-    UPDATE_AUTOMATION_GQL,
-)
-from .slack_integrations_by_entity import (
-    SlackIntegrationsByEntity,
-    SlackIntegrationsByEntityEntity,
-)
-from .update_automation import UpdateAutomation
-
 __all__ = [
     "CREATE_AUTOMATION_GQL",
     "CREATE_GENERIC_WEBHOOK_INTEGRATION_GQL",
@@ -105,22 +11,14 @@ __all__ = [
     "SLACK_INTEGRATIONS_BY_ENTITY_GQL",
     "UPDATE_AUTOMATION_GQL",
     "GetAutomations",
-    "GetAutomationsSearchScope",
     "GetAutomationsByEntity",
-    "GetAutomationsByEntitySearchScope",
     "CreateAutomation",
     "UpdateAutomation",
     "DeleteAutomation",
     "IntegrationsByEntity",
-    "IntegrationsByEntityEntity",
     "SlackIntegrationsByEntity",
-    "SlackIntegrationsByEntityEntity",
     "GenericWebhookIntegrationsByEntity",
-    "GenericWebhookIntegrationsByEntityEntity",
     "CreateGenericWebhookIntegration",
-    "CreateGenericWebhookIntegrationCreateGenericWebhookIntegration",
-    "CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationGenericWebhookIntegration",
-    "CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationIntegration",
     "CreateFilterTriggerInput",
     "CreateGenericWebhookIntegrationInput",
     "GenericWebhookActionInput",
@@ -133,45 +31,74 @@ __all__ = [
     "ArtifactSequenceScopeFields",
     "FilterEventFields",
     "GenericWebhookActionFields",
-    "GenericWebhookActionFieldsIntegrationGenericWebhookIntegration",
-    "GenericWebhookActionFieldsIntegrationIntegration",
     "GenericWebhookIntegrationConnectionFields",
-    "GenericWebhookIntegrationConnectionFieldsEdges",
-    "GenericWebhookIntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration",
-    "GenericWebhookIntegrationConnectionFieldsEdgesNodeIntegration",
     "GenericWebhookIntegrationFields",
     "IntegrationConnectionFields",
-    "IntegrationConnectionFieldsEdges",
-    "IntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration",
-    "IntegrationConnectionFieldsEdgesNodeIntegration",
-    "IntegrationConnectionFieldsEdgesNodeSlackIntegration",
     "NoOpActionFields",
     "NotificationActionFields",
-    "NotificationActionFieldsIntegrationIntegration",
-    "NotificationActionFieldsIntegrationSlackIntegration",
     "PageInfoFields",
     "ProjectConnectionFields",
-    "ProjectConnectionFieldsEdges",
-    "ProjectConnectionFieldsEdgesNode",
     "ProjectScopeFields",
     "QueueJobActionFields",
-    "QueueJobActionFieldsQueue",
     "SlackIntegrationConnectionFields",
-    "SlackIntegrationConnectionFieldsEdges",
-    "SlackIntegrationConnectionFieldsEdgesNodeIntegration",
-    "SlackIntegrationConnectionFieldsEdgesNodeSlackIntegration",
     "SlackIntegrationFields",
     "TriggerFields",
-    "TriggerFieldsActionGenericWebhookTriggeredAction",
-    "TriggerFieldsActionNoOpTriggeredAction",
-    "TriggerFieldsActionNotificationTriggeredAction",
-    "TriggerFieldsActionQueueJobTriggeredAction",
-    "TriggerFieldsEventFilterEventTriggeringCondition",
-    "TriggerFieldsScopeArtifactPortfolio",
-    "TriggerFieldsScopeArtifactSequence",
-    "TriggerFieldsScopeProject",
     "AlertSeverity",
     "EventTriggeringConditionType",
     "TriggerScopeType",
     "TriggeredActionType",
 ]
+from .create_automation import CreateAutomation
+from .create_generic_webhook_integration import CreateGenericWebhookIntegration
+from .delete_automation import DeleteAutomation
+from .enums import (
+    AlertSeverity,
+    EventTriggeringConditionType,
+    TriggeredActionType,
+    TriggerScopeType,
+)
+from .fragments import (
+    ArtifactPortfolioScopeFields,
+    ArtifactSequenceScopeFields,
+    FilterEventFields,
+    GenericWebhookActionFields,
+    GenericWebhookIntegrationConnectionFields,
+    GenericWebhookIntegrationFields,
+    IntegrationConnectionFields,
+    NoOpActionFields,
+    NotificationActionFields,
+    PageInfoFields,
+    ProjectConnectionFields,
+    ProjectScopeFields,
+    QueueJobActionFields,
+    SlackIntegrationConnectionFields,
+    SlackIntegrationFields,
+    TriggerFields,
+)
+from .generic_webhook_integrations_by_entity import GenericWebhookIntegrationsByEntity
+from .get_automations import GetAutomations
+from .get_automations_by_entity import GetAutomationsByEntity
+from .input_types import (
+    CreateFilterTriggerInput,
+    CreateGenericWebhookIntegrationInput,
+    GenericWebhookActionInput,
+    NoOpTriggeredActionInput,
+    NotificationActionInput,
+    QueueJobActionInput,
+    TriggeredActionConfig,
+    UpdateFilterTriggerInput,
+)
+from .integrations_by_entity import IntegrationsByEntity
+from .operations import (
+    CREATE_AUTOMATION_GQL,
+    CREATE_GENERIC_WEBHOOK_INTEGRATION_GQL,
+    DELETE_AUTOMATION_GQL,
+    GENERIC_WEBHOOK_INTEGRATIONS_BY_ENTITY_GQL,
+    GET_AUTOMATIONS_BY_ENTITY_GQL,
+    GET_AUTOMATIONS_GQL,
+    INTEGRATIONS_BY_ENTITY_GQL,
+    SLACK_INTEGRATIONS_BY_ENTITY_GQL,
+    UPDATE_AUTOMATION_GQL,
+)
+from .slack_integrations_by_entity import SlackIntegrationsByEntity
+from .update_automation import UpdateAutomation

wandb/automations/_generated/create_generic_webhook_integration.py

@@ -21,7 +21,7 @@ class CreateGenericWebhookIntegration(GQLBase):
 class CreateGenericWebhookIntegrationCreateGenericWebhookIntegration(GQLBase):
     integration: Union[
         CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationIntegration,
-        CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationGenericWebhookIntegration,
+        GenericWebhookIntegrationFields,
     ] = Field(discriminator="typename__")
 
 
@@ -33,11 +33,5 @@ class CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationI
     ]
 
 
-class CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationGenericWebhookIntegration(
-    GenericWebhookIntegrationFields
-):
-    typename__: Typename[Literal["GenericWebhookIntegration"]]
-
-
 CreateGenericWebhookIntegration.model_rebuild()
 CreateGenericWebhookIntegrationCreateGenericWebhookIntegration.model_rebuild()

wandb/automations/_generated/fragments.py

@@ -52,7 +52,7 @@ class GenericWebhookActionFields(GQLBase):
     )
     integration: Union[
         GenericWebhookActionFieldsIntegrationIntegration,
-        GenericWebhookActionFieldsIntegrationGenericWebhookIntegration,
+        GenericWebhookIntegrationFields,
     ] = Field(discriminator="typename__")
     request_payload: Optional[str] = Field(alias="requestPayload")
 
@@ -75,7 +75,7 @@ class GenericWebhookIntegrationConnectionFieldsEdges(GQLBase):
         Annotated[
             Union[
                 GenericWebhookIntegrationConnectionFieldsEdgesNodeIntegration,
-                GenericWebhookIntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration,
+                GenericWebhookIntegrationFields,
             ],
             Field(discriminator="typename__"),
         ]
@@ -109,8 +109,8 @@ class IntegrationConnectionFieldsEdges(GQLBase):
         Annotated[
             Union[
                 IntegrationConnectionFieldsEdgesNodeIntegration,
-                IntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration,
-                IntegrationConnectionFieldsEdgesNodeSlackIntegration,
+                GenericWebhookIntegrationFields,
+                SlackIntegrationFields,
             ],
             Field(discriminator="typename__"),
         ]
@@ -131,8 +131,7 @@ class NotificationActionFields(GQLBase):
         "NotificationTriggeredAction"
     )
     integration: Union[
-        NotificationActionFieldsIntegrationIntegration,
-        NotificationActionFieldsIntegrationSlackIntegration,
+        NotificationActionFieldsIntegrationIntegration, SlackIntegrationFields
     ] = Field(discriminator="typename__")
     title: Optional[str]
     message: Optional[str]
@@ -194,7 +193,7 @@ class SlackIntegrationConnectionFieldsEdges(GQLBase):
         Annotated[
             Union[
                 SlackIntegrationConnectionFieldsEdgesNodeIntegration,
-                SlackIntegrationConnectionFieldsEdgesNodeSlackIntegration,
+                SlackIntegrationFields,
             ],
             Field(discriminator="typename__"),
         ]
@@ -223,16 +222,14 @@ class TriggerFields(GQLBase):
     description: Optional[str]
     enabled: bool
     scope: Union[
-        TriggerFieldsScopeProject,
-        TriggerFieldsScopeArtifactSequence,
-        TriggerFieldsScopeArtifactPortfolio,
+        ProjectScopeFields, ArtifactSequenceScopeFields, ArtifactPortfolioScopeFields
     ] = Field(discriminator="typename__")
-    event: TriggerFieldsEventFilterEventTriggeringCondition
+    event: FilterEventFields
     action: Union[
-        TriggerFieldsActionQueueJobTriggeredAction,
-        TriggerFieldsActionNotificationTriggeredAction,
-        TriggerFieldsActionGenericWebhookTriggeredAction,
-        TriggerFieldsActionNoOpTriggeredAction,
+        QueueJobActionFields,
+        NotificationActionFields,
+        GenericWebhookActionFields,
+        NoOpActionFields,
     ] = Field(discriminator="typename__")
 
 
@@ -243,68 +240,6 @@ class UpdateAutomationResult(GQLBase):
     trigger: Optional[TriggerFields]
 
 
-class TriggerFieldsScopeArtifactPortfolio(ArtifactPortfolioScopeFields):
-    typename__: Typename[Literal["ArtifactPortfolio"]]
-
-
-class TriggerFieldsScopeArtifactSequence(ArtifactSequenceScopeFields):
-    typename__: Typename[Literal["ArtifactSequence"]]
-
-
-class TriggerFieldsEventFilterEventTriggeringCondition(FilterEventFields):
-    typename__: Typename[Literal["FilterEventTriggeringCondition"]]
-
-
-class TriggerFieldsActionGenericWebhookTriggeredAction(GenericWebhookActionFields):
-    typename__: Typename[Literal["GenericWebhookTriggeredAction"]]
-
-
-class GenericWebhookActionFieldsIntegrationGenericWebhookIntegration(
-    GenericWebhookIntegrationFields
-):
-    typename__: Typename[Literal["GenericWebhookIntegration"]]
-
-
-class GenericWebhookIntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration(
-    GenericWebhookIntegrationFields
-):
-    typename__: Typename[Literal["GenericWebhookIntegration"]]
-
-
-class IntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration(
-    GenericWebhookIntegrationFields
-):
-    typename__: Typename[Literal["GenericWebhookIntegration"]]
-
-
-class TriggerFieldsActionNoOpTriggeredAction(NoOpActionFields):
-    typename__: Typename[Literal["NoOpTriggeredAction"]]
-
-
-class TriggerFieldsActionNotificationTriggeredAction(NotificationActionFields):
-    typename__: Typename[Literal["NotificationTriggeredAction"]]
-
-
-class TriggerFieldsScopeProject(ProjectScopeFields):
-    typename__: Typename[Literal["Project"]]
-
-
-class TriggerFieldsActionQueueJobTriggeredAction(QueueJobActionFields):
-    typename__: Typename[Literal["QueueJobTriggeredAction"]]
-
-
-class IntegrationConnectionFieldsEdgesNodeSlackIntegration(SlackIntegrationFields):
-    typename__: Typename[Literal["SlackIntegration"]]
-
-
-class NotificationActionFieldsIntegrationSlackIntegration(SlackIntegrationFields):
-    typename__: Typename[Literal["SlackIntegration"]]
-
-
-class SlackIntegrationConnectionFieldsEdgesNodeSlackIntegration(SlackIntegrationFields):
-    typename__: Typename[Literal["SlackIntegration"]]
-
-
 ArtifactPortfolioScopeFields.model_rebuild()
 ArtifactSequenceScopeFields.model_rebuild()
 CreateAutomationResult.model_rebuild()
@@ -335,24 +270,24 @@ SlackIntegrationConnectionFieldsEdgesNodeIntegration.model_rebuild()
 SlackIntegrationFields.model_rebuild()
 TriggerFields.model_rebuild()
 UpdateAutomationResult.model_rebuild()
-TriggerFieldsScopeArtifactPortfolio.model_rebuild()
-TriggerFieldsScopeArtifactSequence.model_rebuild()
-TriggerFieldsEventFilterEventTriggeringCondition.model_rebuild()
-TriggerFieldsActionGenericWebhookTriggeredAction.model_rebuild()
-GenericWebhookActionFieldsIntegrationGenericWebhookIntegration.model_rebuild()
-GenericWebhookIntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration.model_rebuild()
-IntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration.model_rebuild()
-TriggerFieldsActionNoOpTriggeredAction.model_rebuild()
-TriggerFieldsActionNotificationTriggeredAction.model_rebuild()
+ArtifactPortfolioScopeFields.model_rebuild()
+ArtifactSequenceScopeFields.model_rebuild()
+FilterEventFields.model_rebuild()
+GenericWebhookActionFields.model_rebuild()
+GenericWebhookIntegrationFields.model_rebuild()
+GenericWebhookIntegrationFields.model_rebuild()
+GenericWebhookIntegrationFields.model_rebuild()
+NoOpActionFields.model_rebuild()
+NotificationActionFields.model_rebuild()
 PageInfoFields.model_rebuild()
 PageInfoFields.model_rebuild()
 PageInfoFields.model_rebuild()
 PageInfoFields.model_rebuild()
-TriggerFieldsScopeProject.model_rebuild()
-TriggerFieldsActionQueueJobTriggeredAction.model_rebuild()
-IntegrationConnectionFieldsEdgesNodeSlackIntegration.model_rebuild()
-NotificationActionFieldsIntegrationSlackIntegration.model_rebuild()
-SlackIntegrationConnectionFieldsEdgesNodeSlackIntegration.model_rebuild()
+ProjectScopeFields.model_rebuild()
+QueueJobActionFields.model_rebuild()
+SlackIntegrationFields.model_rebuild()
+SlackIntegrationFields.model_rebuild()
+SlackIntegrationFields.model_rebuild()
 TriggerFields.model_rebuild()
 TriggerFields.model_rebuild()
 TriggerFields.model_rebuild()

wandb/cli/beta_sync.py

@@ -14,7 +14,7 @@ import wandb
 from wandb.proto.wandb_sync_pb2 import ServerSyncResponse
 from wandb.sdk import wandb_setup
 from wandb.sdk.lib import asyncio_compat
-from wandb.sdk.lib.printer import ERROR, Printer, new_printer
+from wandb.sdk.lib.printer import Printer, new_printer
 from wandb.sdk.lib.progress import progress_printer
 from wandb.sdk.lib.service.service_connection import ServiceConnection
 from wandb.sdk.mailbox.mailbox_handle import MailboxHandle
@@ -86,12 +86,10 @@ async def _do_sync(
 
     This is factored out to make the progress animation testable.
     """
-    init_result = await service.init_sync(
-        wandb_files,
-        settings,
-    ).wait_async(timeout=5)
+    init_handle = await service.init_sync(wandb_files, settings)
+    init_result = await init_handle.wait_async(timeout=5)
 
-    sync_handle = service.sync(init_result.id, parallelism=parallelism)
+    sync_handle = await service.sync(init_result.id, parallelism=parallelism)
 
     await _SyncStatusLoop(
         init_result.id,
@@ -130,19 +128,19 @@ class _SyncStatusLoop:
         handle: MailboxHandle[ServerSyncResponse],
     ) -> None:
         response = await handle.wait_async(timeout=None)
-        if messages := list(response.errors):
-            self._printer.display(messages, level=ERROR)
+        for msg in response.messages:
+            self._printer.display(msg.content, level=msg.severity)
         self._done.set()
 
     async def _show_progress_until_done(self) -> None:
         """Show rate-limited status updates until _done is set."""
         with progress_printer(self._printer, "Syncing...") as progress:
             while not await self._rate_limit_check_done():
-                handle = self._service.sync_status(self._id)
+                handle = await self._service.sync_status(self._id)
                 response = await handle.wait_async(timeout=None)
 
-                if messages := list(response.new_errors):
-                    self._printer.display(messages, level=ERROR)
+                for msg in response.new_messages:
+                    self._printer.display(msg.content, level=msg.severity)
                 progress.update(response.stats)
 
     async def _rate_limit_check_done(self) -> bool:

wandb/errors/errors.py

@@ -1,4 +1,4 @@
-from typing import Optional
+from __future__ import annotations
 
 
 class Error(Exception):
@@ -7,7 +7,7 @@ class Error(Exception):
     <!-- lazydoc-ignore-class: internal -->
     """
 
-    def __init__(self, message, context: Optional[dict] = None) -> None:
+    def __init__(self, message: str, context: dict | None = None) -> None:
         super().__init__(message)
         self.message = message
         # sentry context capture
@@ -18,7 +18,7 @@ class Error(Exception):
 class CommError(Error):
     """Error communicating with W&B servers."""
 
-    def __init__(self, msg, exc=None) -> None:
+    def __init__(self, msg: str, exc: Exception | None = None) -> None:
         self.exc = exc
         self.message = msg
         super().__init__(self.message)