wandb 0.22.0__py3-none-win_amd64.whl → 0.22.1__py3-none-win_amd64.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)

wandb/proto/v3/wandb_sync_pb2.py

@@ -16,7 +16,7 @@ from wandb.proto import wandb_internal_pb2 as wandb_dot_proto_dot_wandb__interna
 from wandb.proto import wandb_settings_pb2 as wandb_dot_proto_dot_wandb__settings__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cwandb/proto/wandb_sync.proto\x12\x0ewandb_internal\x1a wandb/proto/wandb_internal.proto\x1a wandb/proto/wandb_settings.proto\"Q\n\x15ServerInitSyncRequest\x12\x0c\n\x04path\x18\x01 \x03(\t\x12*\n\x08settings\x18\x02 \x01(\x0b\x32\x18.wandb_internal.Settings\"$\n\x16ServerInitSyncResponse\x12\n\n\x02id\x18\x01 \x01(\t\"4\n\x11ServerSyncRequest\x12\n\n\x02id\x18\x01 \x01(\t\x12\x13\n\x0bparallelism\x18\x02 \x01(\r\"$\n\x12ServerSyncResponse\x12\x0e\n\x06\x65rrors\x18\x02 \x03(\t\"%\n\x17ServerSyncStatusRequest\x12\n\n\x02id\x18\x01 \x01(\t\"]\n\x18ServerSyncStatusResponse\x12-\n\x05stats\x18\x01 \x01(\x0b\x32\x1e.wandb_internal.OperationStats\x12\x12\n\nnew_errors\x18\x02 \x03(\tB\x1bZ\x19\x63ore/pkg/service_go_protob\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cwandb/proto/wandb_sync.proto\x12\x0ewandb_internal\x1a wandb/proto/wandb_internal.proto\x1a wandb/proto/wandb_settings.proto\"Q\n\x15ServerInitSyncRequest\x12\x0c\n\x04path\x18\x01 \x03(\t\x12*\n\x08settings\x18\x02 \x01(\x0b\x32\x18.wandb_internal.Settings\"$\n\x16ServerInitSyncResponse\x12\n\n\x02id\x18\x01 \x01(\t\"4\n\x11ServerSyncRequest\x12\n\n\x02id\x18\x01 \x01(\t\x12\x13\n\x0bparallelism\x18\x02 \x01(\r\"I\n\x12ServerSyncResponse\x12\x33\n\x08messages\x18\x01 \x03(\x0b\x32!.wandb_internal.ServerSyncMessage\"%\n\x17ServerSyncStatusRequest\x12\n\n\x02id\x18\x01 \x01(\t\"\x82\x01\n\x18ServerSyncStatusResponse\x12-\n\x05stats\x18\x01 \x01(\x0b\x32\x1e.wandb_internal.OperationStats\x12\x37\n\x0cnew_messages\x18\x02 \x03(\x0b\x32!.wandb_internal.ServerSyncMessage\"\xaa\x01\n\x11ServerSyncMessage\x12<\n\x08severity\x18\x01 \x01(\x0e\x32*.wandb_internal.ServerSyncMessage.Severity\x12\x0f\n\x07\x63ontent\x18\x02 \x01(\t\"F\n\x08Severity\x12\x13\n\x0fSEVERITY_NOTSET\x10\x00\x12\x11\n\rSEVERITY_INFO\x10\x14\x12\x12\n\x0eSEVERITY_ERROR\x10(B\x1bZ\x19\x63ore/pkg/service_go_protob\x06proto3')
 
 
 
@@ -26,6 +26,8 @@ _SERVERSYNCREQUEST = DESCRIPTOR.message_types_by_name['ServerSyncRequest']
 _SERVERSYNCRESPONSE = DESCRIPTOR.message_types_by_name['ServerSyncResponse']
 _SERVERSYNCSTATUSREQUEST = DESCRIPTOR.message_types_by_name['ServerSyncStatusRequest']
 _SERVERSYNCSTATUSRESPONSE = DESCRIPTOR.message_types_by_name['ServerSyncStatusResponse']
+_SERVERSYNCMESSAGE = DESCRIPTOR.message_types_by_name['ServerSyncMessage']
+_SERVERSYNCMESSAGE_SEVERITY = _SERVERSYNCMESSAGE.enum_types_by_name['Severity']
 ServerInitSyncRequest = _reflection.GeneratedProtocolMessageType('ServerInitSyncRequest', (_message.Message,), {
   'DESCRIPTOR' : _SERVERINITSYNCREQUEST,
   '__module__' : 'wandb.proto.wandb_sync_pb2'
@@ -68,6 +70,13 @@ ServerSyncStatusResponse = _reflection.GeneratedProtocolMessageType('ServerSyncS
   })
 _sym_db.RegisterMessage(ServerSyncStatusResponse)
 
+ServerSyncMessage = _reflection.GeneratedProtocolMessageType('ServerSyncMessage', (_message.Message,), {
+  'DESCRIPTOR' : _SERVERSYNCMESSAGE,
+  '__module__' : 'wandb.proto.wandb_sync_pb2'
+  # @@protoc_insertion_point(class_scope:wandb_internal.ServerSyncMessage)
+  })
+_sym_db.RegisterMessage(ServerSyncMessage)
+
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
@@ -79,9 +88,13 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _SERVERSYNCREQUEST._serialized_start=237
   _SERVERSYNCREQUEST._serialized_end=289
   _SERVERSYNCRESPONSE._serialized_start=291
-  _SERVERSYNCRESPONSE._serialized_end=327
-  _SERVERSYNCSTATUSREQUEST._serialized_start=329
-  _SERVERSYNCSTATUSREQUEST._serialized_end=366
-  _SERVERSYNCSTATUSRESPONSE._serialized_start=368
-  _SERVERSYNCSTATUSRESPONSE._serialized_end=461
+  _SERVERSYNCRESPONSE._serialized_end=364
+  _SERVERSYNCSTATUSREQUEST._serialized_start=366
+  _SERVERSYNCSTATUSREQUEST._serialized_end=403
+  _SERVERSYNCSTATUSRESPONSE._serialized_start=406
+  _SERVERSYNCSTATUSRESPONSE._serialized_end=536
+  _SERVERSYNCMESSAGE._serialized_start=539
+  _SERVERSYNCMESSAGE._serialized_end=709
+  _SERVERSYNCMESSAGE_SEVERITY._serialized_start=639
+  _SERVERSYNCMESSAGE_SEVERITY._serialized_end=709
 # @@protoc_insertion_point(module_scope)

wandb/proto/v4/wandb_sync_pb2.py

@@ -15,7 +15,7 @@ from wandb.proto import wandb_internal_pb2 as wandb_dot_proto_dot_wandb__interna
 from wandb.proto import wandb_settings_pb2 as wandb_dot_proto_dot_wandb__settings__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cwandb/proto/wandb_sync.proto\x12\x0ewandb_internal\x1a wandb/proto/wandb_internal.proto\x1a wandb/proto/wandb_settings.proto\"Q\n\x15ServerInitSyncRequest\x12\x0c\n\x04path\x18\x01 \x03(\t\x12*\n\x08settings\x18\x02 \x01(\x0b\x32\x18.wandb_internal.Settings\"$\n\x16ServerInitSyncResponse\x12\n\n\x02id\x18\x01 \x01(\t\"4\n\x11ServerSyncRequest\x12\n\n\x02id\x18\x01 \x01(\t\x12\x13\n\x0bparallelism\x18\x02 \x01(\r\"$\n\x12ServerSyncResponse\x12\x0e\n\x06\x65rrors\x18\x02 \x03(\t\"%\n\x17ServerSyncStatusRequest\x12\n\n\x02id\x18\x01 \x01(\t\"]\n\x18ServerSyncStatusResponse\x12-\n\x05stats\x18\x01 \x01(\x0b\x32\x1e.wandb_internal.OperationStats\x12\x12\n\nnew_errors\x18\x02 \x03(\tB\x1bZ\x19\x63ore/pkg/service_go_protob\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cwandb/proto/wandb_sync.proto\x12\x0ewandb_internal\x1a wandb/proto/wandb_internal.proto\x1a wandb/proto/wandb_settings.proto\"Q\n\x15ServerInitSyncRequest\x12\x0c\n\x04path\x18\x01 \x03(\t\x12*\n\x08settings\x18\x02 \x01(\x0b\x32\x18.wandb_internal.Settings\"$\n\x16ServerInitSyncResponse\x12\n\n\x02id\x18\x01 \x01(\t\"4\n\x11ServerSyncRequest\x12\n\n\x02id\x18\x01 \x01(\t\x12\x13\n\x0bparallelism\x18\x02 \x01(\r\"I\n\x12ServerSyncResponse\x12\x33\n\x08messages\x18\x01 \x03(\x0b\x32!.wandb_internal.ServerSyncMessage\"%\n\x17ServerSyncStatusRequest\x12\n\n\x02id\x18\x01 \x01(\t\"\x82\x01\n\x18ServerSyncStatusResponse\x12-\n\x05stats\x18\x01 \x01(\x0b\x32\x1e.wandb_internal.OperationStats\x12\x37\n\x0cnew_messages\x18\x02 \x03(\x0b\x32!.wandb_internal.ServerSyncMessage\"\xaa\x01\n\x11ServerSyncMessage\x12<\n\x08severity\x18\x01 \x01(\x0e\x32*.wandb_internal.ServerSyncMessage.Severity\x12\x0f\n\x07\x63ontent\x18\x02 \x01(\t\"F\n\x08Severity\x12\x13\n\x0fSEVERITY_NOTSET\x10\x00\x12\x11\n\rSEVERITY_INFO\x10\x14\x12\x12\n\x0eSEVERITY_ERROR\x10(B\x1bZ\x19\x63ore/pkg/service_go_protob\x06proto3')
 
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'wandb.proto.wandb_sync_pb2', globals())
@@ -30,9 +30,13 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _SERVERSYNCREQUEST._serialized_start=237
   _SERVERSYNCREQUEST._serialized_end=289
   _SERVERSYNCRESPONSE._serialized_start=291
-  _SERVERSYNCRESPONSE._serialized_end=327
-  _SERVERSYNCSTATUSREQUEST._serialized_start=329
-  _SERVERSYNCSTATUSREQUEST._serialized_end=366
-  _SERVERSYNCSTATUSRESPONSE._serialized_start=368
-  _SERVERSYNCSTATUSRESPONSE._serialized_end=461
+  _SERVERSYNCRESPONSE._serialized_end=364
+  _SERVERSYNCSTATUSREQUEST._serialized_start=366
+  _SERVERSYNCSTATUSREQUEST._serialized_end=403
+  _SERVERSYNCSTATUSRESPONSE._serialized_start=406
+  _SERVERSYNCSTATUSRESPONSE._serialized_end=536
+  _SERVERSYNCMESSAGE._serialized_start=539
+  _SERVERSYNCMESSAGE._serialized_end=709
+  _SERVERSYNCMESSAGE_SEVERITY._serialized_start=639
+  _SERVERSYNCMESSAGE_SEVERITY._serialized_end=709
 # @@protoc_insertion_point(module_scope)

wandb/proto/v5/wandb_sync_pb2.py

@@ -16,7 +16,7 @@ from wandb.proto import wandb_internal_pb2 as wandb_dot_proto_dot_wandb__interna
 from wandb.proto import wandb_settings_pb2 as wandb_dot_proto_dot_wandb__settings__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cwandb/proto/wandb_sync.proto\x12\x0ewandb_internal\x1a wandb/proto/wandb_internal.proto\x1a wandb/proto/wandb_settings.proto\"Q\n\x15ServerInitSyncRequest\x12\x0c\n\x04path\x18\x01 \x03(\t\x12*\n\x08settings\x18\x02 \x01(\x0b\x32\x18.wandb_internal.Settings\"$\n\x16ServerInitSyncResponse\x12\n\n\x02id\x18\x01 \x01(\t\"4\n\x11ServerSyncRequest\x12\n\n\x02id\x18\x01 \x01(\t\x12\x13\n\x0bparallelism\x18\x02 \x01(\r\"$\n\x12ServerSyncResponse\x12\x0e\n\x06\x65rrors\x18\x02 \x03(\t\"%\n\x17ServerSyncStatusRequest\x12\n\n\x02id\x18\x01 \x01(\t\"]\n\x18ServerSyncStatusResponse\x12-\n\x05stats\x18\x01 \x01(\x0b\x32\x1e.wandb_internal.OperationStats\x12\x12\n\nnew_errors\x18\x02 \x03(\tB\x1bZ\x19\x63ore/pkg/service_go_protob\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cwandb/proto/wandb_sync.proto\x12\x0ewandb_internal\x1a wandb/proto/wandb_internal.proto\x1a wandb/proto/wandb_settings.proto\"Q\n\x15ServerInitSyncRequest\x12\x0c\n\x04path\x18\x01 \x03(\t\x12*\n\x08settings\x18\x02 \x01(\x0b\x32\x18.wandb_internal.Settings\"$\n\x16ServerInitSyncResponse\x12\n\n\x02id\x18\x01 \x01(\t\"4\n\x11ServerSyncRequest\x12\n\n\x02id\x18\x01 \x01(\t\x12\x13\n\x0bparallelism\x18\x02 \x01(\r\"I\n\x12ServerSyncResponse\x12\x33\n\x08messages\x18\x01 \x03(\x0b\x32!.wandb_internal.ServerSyncMessage\"%\n\x17ServerSyncStatusRequest\x12\n\n\x02id\x18\x01 \x01(\t\"\x82\x01\n\x18ServerSyncStatusResponse\x12-\n\x05stats\x18\x01 \x01(\x0b\x32\x1e.wandb_internal.OperationStats\x12\x37\n\x0cnew_messages\x18\x02 \x03(\x0b\x32!.wandb_internal.ServerSyncMessage\"\xaa\x01\n\x11ServerSyncMessage\x12<\n\x08severity\x18\x01 \x01(\x0e\x32*.wandb_internal.ServerSyncMessage.Severity\x12\x0f\n\x07\x63ontent\x18\x02 \x01(\t\"F\n\x08Severity\x12\x13\n\x0fSEVERITY_NOTSET\x10\x00\x12\x11\n\rSEVERITY_INFO\x10\x14\x12\x12\n\x0eSEVERITY_ERROR\x10(B\x1bZ\x19\x63ore/pkg/service_go_protob\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -31,9 +31,13 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_SERVERSYNCREQUEST']._serialized_start=237
   _globals['_SERVERSYNCREQUEST']._serialized_end=289
   _globals['_SERVERSYNCRESPONSE']._serialized_start=291
-  _globals['_SERVERSYNCRESPONSE']._serialized_end=327
-  _globals['_SERVERSYNCSTATUSREQUEST']._serialized_start=329
-  _globals['_SERVERSYNCSTATUSREQUEST']._serialized_end=366
-  _globals['_SERVERSYNCSTATUSRESPONSE']._serialized_start=368
-  _globals['_SERVERSYNCSTATUSRESPONSE']._serialized_end=461
+  _globals['_SERVERSYNCRESPONSE']._serialized_end=364
+  _globals['_SERVERSYNCSTATUSREQUEST']._serialized_start=366
+  _globals['_SERVERSYNCSTATUSREQUEST']._serialized_end=403
+  _globals['_SERVERSYNCSTATUSRESPONSE']._serialized_start=406
+  _globals['_SERVERSYNCSTATUSRESPONSE']._serialized_end=536
+  _globals['_SERVERSYNCMESSAGE']._serialized_start=539
+  _globals['_SERVERSYNCMESSAGE']._serialized_end=709
+  _globals['_SERVERSYNCMESSAGE_SEVERITY']._serialized_start=639
+  _globals['_SERVERSYNCMESSAGE_SEVERITY']._serialized_end=709
 # @@protoc_insertion_point(module_scope)

wandb/proto/v6/wandb_sync_pb2.py

@@ -26,7 +26,7 @@ from wandb.proto import wandb_internal_pb2 as wandb_dot_proto_dot_wandb__interna
 from wandb.proto import wandb_settings_pb2 as wandb_dot_proto_dot_wandb__settings__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cwandb/proto/wandb_sync.proto\x12\x0ewandb_internal\x1a wandb/proto/wandb_internal.proto\x1a wandb/proto/wandb_settings.proto\"Q\n\x15ServerInitSyncRequest\x12\x0c\n\x04path\x18\x01 \x03(\t\x12*\n\x08settings\x18\x02 \x01(\x0b\x32\x18.wandb_internal.Settings\"$\n\x16ServerInitSyncResponse\x12\n\n\x02id\x18\x01 \x01(\t\"4\n\x11ServerSyncRequest\x12\n\n\x02id\x18\x01 \x01(\t\x12\x13\n\x0bparallelism\x18\x02 \x01(\r\"$\n\x12ServerSyncResponse\x12\x0e\n\x06\x65rrors\x18\x02 \x03(\t\"%\n\x17ServerSyncStatusRequest\x12\n\n\x02id\x18\x01 \x01(\t\"]\n\x18ServerSyncStatusResponse\x12-\n\x05stats\x18\x01 \x01(\x0b\x32\x1e.wandb_internal.OperationStats\x12\x12\n\nnew_errors\x18\x02 \x03(\tB\x1bZ\x19\x63ore/pkg/service_go_protob\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cwandb/proto/wandb_sync.proto\x12\x0ewandb_internal\x1a wandb/proto/wandb_internal.proto\x1a wandb/proto/wandb_settings.proto\"Q\n\x15ServerInitSyncRequest\x12\x0c\n\x04path\x18\x01 \x03(\t\x12*\n\x08settings\x18\x02 \x01(\x0b\x32\x18.wandb_internal.Settings\"$\n\x16ServerInitSyncResponse\x12\n\n\x02id\x18\x01 \x01(\t\"4\n\x11ServerSyncRequest\x12\n\n\x02id\x18\x01 \x01(\t\x12\x13\n\x0bparallelism\x18\x02 \x01(\r\"I\n\x12ServerSyncResponse\x12\x33\n\x08messages\x18\x01 \x03(\x0b\x32!.wandb_internal.ServerSyncMessage\"%\n\x17ServerSyncStatusRequest\x12\n\n\x02id\x18\x01 \x01(\t\"\x82\x01\n\x18ServerSyncStatusResponse\x12-\n\x05stats\x18\x01 \x01(\x0b\x32\x1e.wandb_internal.OperationStats\x12\x37\n\x0cnew_messages\x18\x02 \x03(\x0b\x32!.wandb_internal.ServerSyncMessage\"\xaa\x01\n\x11ServerSyncMessage\x12<\n\x08severity\x18\x01 \x01(\x0e\x32*.wandb_internal.ServerSyncMessage.Severity\x12\x0f\n\x07\x63ontent\x18\x02 \x01(\t\"F\n\x08Severity\x12\x13\n\x0fSEVERITY_NOTSET\x10\x00\x12\x11\n\rSEVERITY_INFO\x10\x14\x12\x12\n\x0eSEVERITY_ERROR\x10(B\x1bZ\x19\x63ore/pkg/service_go_protob\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -41,9 +41,13 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_SERVERSYNCREQUEST']._serialized_start=237
   _globals['_SERVERSYNCREQUEST']._serialized_end=289
   _globals['_SERVERSYNCRESPONSE']._serialized_start=291
-  _globals['_SERVERSYNCRESPONSE']._serialized_end=327
-  _globals['_SERVERSYNCSTATUSREQUEST']._serialized_start=329
-  _globals['_SERVERSYNCSTATUSREQUEST']._serialized_end=366
-  _globals['_SERVERSYNCSTATUSRESPONSE']._serialized_start=368
-  _globals['_SERVERSYNCSTATUSRESPONSE']._serialized_end=461
+  _globals['_SERVERSYNCRESPONSE']._serialized_end=364
+  _globals['_SERVERSYNCSTATUSREQUEST']._serialized_start=366
+  _globals['_SERVERSYNCSTATUSREQUEST']._serialized_end=403
+  _globals['_SERVERSYNCSTATUSRESPONSE']._serialized_start=406
+  _globals['_SERVERSYNCSTATUSRESPONSE']._serialized_end=536
+  _globals['_SERVERSYNCMESSAGE']._serialized_start=539
+  _globals['_SERVERSYNCMESSAGE']._serialized_end=709
+  _globals['_SERVERSYNCMESSAGE_SEVERITY']._serialized_start=639
+  _globals['_SERVERSYNCMESSAGE_SEVERITY']._serialized_end=709
 # @@protoc_insertion_point(module_scope)

wandb/sdk/artifacts/_factories.py

@@ -11,7 +11,12 @@ if TYPE_CHECKING:
     from .storage_policy import StoragePolicy
 
 
-def make_storage_policy() -> StoragePolicy:
+def make_storage_policy(storage_region: str | None = None) -> StoragePolicy:
     """A factory function that returns the default StoragePolicy for the current environment."""
     layout = StorageLayout.V1 if env.get_use_v1_artifacts() else StorageLayout.V2
-    return WandbStoragePolicy.from_config({"storageLayout": layout})
+    config = {"storageLayout": layout}
+    # Only set storage region if is not None for backward compatibility
+    # Validation such as non empty string is done in WandbStoragePolicy.__init__
+    if storage_region is not None:
+        config["storageRegion"] = storage_region
+    return WandbStoragePolicy.from_config(config)