wandb 0.19.7__py3-none-musllinux_1_2_aarch64.whl → 0.19.9__py3-none-musllinux_1_2_aarch64.whl

wandb/apis/public/_generated/base.py

@@ -1,128 +0,0 @@
-# Generated by ariadne-codegen
-
-from __future__ import annotations
-
-from typing import Any, Literal, TypeVar
-
-from pydantic import BaseModel, ConfigDict, Field, Json, ValidationError, WrapValidator
-from pydantic.alias_generators import to_camel
-from pydantic.main import IncEx
-from pydantic_core import to_json
-from pydantic_core.core_schema import ValidatorFunctionWrapHandler
-
-from .typing_compat import Annotated, override
-
-
-# Base class for all automation classes/types.
-# Omitted from docstring to avoid inclusion in generated docs.
-class Base(BaseModel):
-    model_config = ConfigDict(
-        populate_by_name=True,
-        validate_assignment=True,
-        validate_default=True,
-        extra="forbid",
-        alias_generator=to_camel,
-        use_attribute_docstrings=True,
-        from_attributes=True,
-        revalidate_instances="always",
-    )
-
-    @override
-    def model_dump(
-        self,
-        *,
-        mode: Literal["json", "python"] | str = "json",  # NOTE: changed default
-        include: IncEx | None = None,
-        exclude: IncEx | None = None,
-        context: dict[str, Any] | None = None,
-        by_alias: bool = True,  # NOTE: changed default
-        exclude_unset: bool = False,
-        exclude_defaults: bool = False,
-        exclude_none: bool = False,
-        round_trip: bool = True,  # NOTE: changed default
-        warnings: bool | Literal["none", "warn", "error"] = True,
-        serialize_as_any: bool = False,
-    ) -> dict[str, Any]:
-        return super().model_dump(
-            mode=mode,
-            include=include,
-            exclude=exclude,
-            context=context,
-            by_alias=by_alias,
-            exclude_unset=exclude_unset,
-            exclude_defaults=exclude_defaults,
-            exclude_none=exclude_none,
-            round_trip=round_trip,
-            warnings=warnings,
-            serialize_as_any=serialize_as_any,
-        )
-
-    @override
-    def model_dump_json(
-        self,
-        *,
-        indent: int | None = None,
-        include: IncEx | None = None,
-        exclude: IncEx | None = None,
-        context: dict[str, Any] | None = None,
-        by_alias: bool = True,  # NOTE: changed default
-        exclude_unset: bool = False,
-        exclude_defaults: bool = False,
-        exclude_none: bool = False,
-        round_trip: bool = True,  # NOTE: changed default
-        warnings: bool | Literal["none", "warn", "error"] = True,
-        serialize_as_any: bool = False,
-    ) -> str:
-        return super().model_dump_json(
-            indent=indent,
-            include=include,
-            exclude=exclude,
-            context=context,
-            by_alias=by_alias,
-            exclude_unset=exclude_unset,
-            exclude_defaults=exclude_defaults,
-            exclude_none=exclude_none,
-            round_trip=round_trip,
-            warnings=warnings,
-            serialize_as_any=serialize_as_any,
-        )
-
-
-# Base class with extra customization for GQL generated types.
-# Omitted from docstring to avoid inclusion in generated docs.
-class GQLBase(Base):
-    model_config = ConfigDict(
-        extra="ignore",
-        protected_namespaces=(),
-    )
-
-
-# ------------------------------------------------------------------------------
-# Reusable annotations for field types
-T = TypeVar("T")
-
-GQLId = Annotated[
-    str,
-    Field(repr=False, strict=True, frozen=True),
-]
-
-Typename = Annotated[
-    T,
-    Field(repr=False, alias="__typename", frozen=True),
-]
-
-
-def validate_maybe_json(v: Any, handler: ValidatorFunctionWrapHandler) -> Any:
-    """Wraps default Json[...] field validator to allow instantiation with an already-decoded value."""
-    try:
-        return handler(v)
-    except ValidationError:
-        # Try revalidating after properly jsonifying the value
-        return handler(to_json(v, by_alias=True, round_trip=True))
-
-
-SerializedToJson = Annotated[
-    Json[T],
-    # Allow lenient instantiation/validation: incoming data may already be deserialized.
-    WrapValidator(validate_maybe_json),
-]

wandb/sdk/interface/message_future.py

@@ -1,27 +0,0 @@
-"""MessageFuture - represents a message result of an asynchronous operation.
-
-Base class MessageFuture for MessageFutureObject and MessageFuturePoll
-
-"""
-
-import threading
-from abc import abstractmethod
-from typing import Optional
-
-from wandb.proto import wandb_internal_pb2 as pb
-
-
-class MessageFuture:
-    _object: Optional[pb.Result]
-
-    def __init__(self) -> None:
-        self._object = None
-        self._object_ready = threading.Event()
-
-    def _set_object(self, obj: pb.Result) -> None:
-        self._object = obj
-        self._object_ready.set()
-
-    @abstractmethod
-    def get(self, timeout: Optional[int] = None) -> Optional[pb.Result]:
-        raise NotImplementedError

wandb/sdk/interface/message_future_poll.py

@@ -1,50 +0,0 @@
-"""MessageFuturePoll - Derived from MessageFuture but implementing polling loop.
-
-MessageFuture represents a message result of an asynchronous operation.
-
-MessageFuturePoll implements a polling loop to periodically query for a
-completed async operation.
-
-"""
-
-import time
-from typing import Any, Optional
-
-from wandb.proto import wandb_internal_pb2 as pb
-
-from .message_future import MessageFuture
-
-
-class MessageFuturePoll(MessageFuture):
-    _fn: Any
-    _xid: str
-
-    def __init__(self, fn: Any, xid: str) -> None:
-        super().__init__()
-        self._fn = fn
-        self._xid = xid
-
-    def get(self, timeout: Optional[int] = None) -> Optional[pb.Result]:
-        self._poll(timeout=timeout)
-        if self._object_ready.is_set():
-            return self._object
-        return None
-
-    def _poll(self, timeout: Optional[int] = None) -> None:
-        if self._object_ready.is_set():
-            return
-        done = False
-        start_time = time.time()
-        sleep_time = 0.5
-        while not done:
-            result = self._fn(xid=self._xid)
-            if result:
-                self._set_object(result)
-                done = True
-                continue
-            now_time = time.time()
-            if timeout and start_time - now_time > timeout:
-                done = True
-                continue
-            time.sleep(sleep_time)
-            sleep_time = min(sleep_time * 2, 5)