wandb 0.22.0__py3-none-macosx_12_0_x86_64.whl → 0.22.2__py3-none-macosx_12_0_x86_64.whl

wandb/__init__.py

@@ -10,7 +10,7 @@ For reference documentation, see https://docs.wandb.com/ref/python.
 """
 from __future__ import annotations
 
-__version__ = "0.22.0"
+__version__ = "0.22.2"
 
 
 from wandb.errors import Error

wandb/__init__.pyi

@@ -107,7 +107,7 @@ if TYPE_CHECKING:
     import wandb
     from wandb.plot import CustomChart
 
-__version__: str = "0.22.0"
+__version__: str = "0.22.2"
 
 run: Run | None
 config: wandb_config.Config
@@ -388,8 +388,8 @@ def init(
             enable on your settings page.
         tensorboard: Deprecated. Use `sync_tensorboard` instead.
         sync_tensorboard: Enables automatic syncing of W&B logs from TensorBoard
-            or TensorBoardX, saving relevant event files for viewing in the W&B UI.
-            saving relevant event files for viewing in the W&B UI. (Default: `False`)
+            or TensorBoardX, saving relevant event files for viewing in
+            the W&B UI.
         monitor_gym: Enables automatic logging of videos of the environment when
             using OpenAI Gym.
         settings: Specifies a dictionary or `wandb.Settings` object with advanced
@@ -754,6 +754,9 @@ def save(
     When given an absolute path or glob and no `base_path`, one
     directory level is preserved as in the example above.
 
+    Files are automatically deduplicated: calling `save()` multiple times
+    on the same file without modifications will not re-upload it.
+
     Args:
         glob_str: A relative or absolute path or Unix glob.
         base_path: A path to use to infer a directory structure; see examples.
@@ -778,10 +781,10 @@ def save(
     run.save("these/are/myfiles/*", base_path="these")
     # => Saves files in an "are/myfiles/" folder in the run.
 
-    run.save("/User/username/Documents/run123/*.txt")
+    run.save("/Users/username/Documents/run123/*.txt")
     # => Saves files in a "run123/" folder in the run. See note below.
 
-    run.save("/User/username/Documents/run123/*.txt", base_path="/User")
+    run.save("/Users/username/Documents/run123/*.txt", base_path="/Users")
     # => Saves files in a "username/Documents/run123/" folder in the run.
 
     run.save("files/*/saveme.txt")

wandb/_pydantic/__init__.py

@@ -1,19 +1,9 @@
 """Internal utilities for working with pydantic."""
 
-from .base import CompatBaseModel, GQLBase
-from .field_types import GQLId, Typename
-from .utils import IS_PYDANTIC_V2, from_json, gql_typename, pydantic_isinstance, to_json
-from .v1_compat import (
-    AliasChoices,
-    computed_field,
-    field_validator,
-    model_validator,
-    to_camel,
-)
-
 __all__ = [
     "IS_PYDANTIC_V2",
     "CompatBaseModel",
+    "JsonableModel",
     "GQLBase",
     "Typename",
     "GQLId",
@@ -27,3 +17,14 @@ __all__ = [
     "from_json",
     "gql_typename",
 ]
+
+from .base import CompatBaseModel, GQLBase, JsonableModel
+from .field_types import GQLId, Typename
+from .utils import IS_PYDANTIC_V2, from_json, gql_typename, pydantic_isinstance, to_json
+from .v1_compat import (
+    AliasChoices,
+    computed_field,
+    field_validator,
+    model_validator,
+    to_camel,
+)

wandb/_pydantic/base.py

@@ -2,7 +2,8 @@
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, Callable, Literal
+from abc import ABC
+from typing import TYPE_CHECKING, Any, Callable, ClassVar, Literal
 
 from pydantic import BaseModel, ConfigDict
 from typing_extensions import TypedDict, Unpack, override
@@ -29,34 +30,54 @@ class ModelDumpKwargs(TypedDict, total=False):
     serialize_as_any: bool
 
 
-#: Custom overrides of default kwargs for `BaseModel.model_{dump,dump_json}`.
-MODEL_DUMP_DEFAULTS = ModelDumpKwargs(
-    by_alias=True,  # Always serialize with aliases (e.g. camelCase names)
-    round_trip=True,  # Ensure serialized values remain valid inputs for deserialization
-)
+# ---------------------------------------------------------------------------
+# Base models and mixin classes.
+#
+# Extra info is provided for devs in inline comments, NOT docstrings.  This
+# prevents it from showing up in generated docs for subclasses.
 
 
-# v1-compatible base class for pydantic types.
+# FOR INTERNAL USE ONLY: v1-compatible drop-in replacement for `pydantic.BaseModel`.
+# If pydantic v2 is detected, this is just `pydantic.BaseModel`.
+#
+# Deliberately inherits ALL default configuration from `pydantic.BaseModel`.
 class CompatBaseModel(PydanticCompatMixin, BaseModel):
     __doc__ = None  # Prevent subclasses from inheriting the BaseModel docstring
 
 
-# Base class for all GraphQL-generated types.
-# Omitted from docstring to avoid inclusion in generated docs.
-class GQLBase(CompatBaseModel):
+class JsonableModel(CompatBaseModel, ABC):
+    # Base class with sensible default behavior for classes that need to convert to/from JSON.
+    #
+    # Automatically parse/serialize "raw" API data (e.g. automatically convert to/from camelCase keys):
+    # - `.model_{dump,dump_json}()` should return "JSON-ready" dicts or JSON strings
+    # - `.model_{validate,validate_json}()` should accept "JSON-ready" dicts or JSON strings
+    #
+    # Ensure round-trip serialization <-> deserialization between:
+    # - `model_dump()` <-> `model_validate()`
+    # - `model_dump_json()` <-> `model_validate_json()`
+    #
+    # These behaviors are useful for models that need to predictably handle e.g. GraphQL request/response data.
+
     model_config = ConfigDict(
-        populate_by_name=True,  # Discouraged in pydantic v2.11+, will be deprecated in v3
-        validate_by_name=True,  # Introduced in pydantic v2.11
-        validate_by_alias=True,  # Introduced in pydantic v2.11
-        serialize_by_alias=True,  # Introduced in pydantic v2.11
+        # ---------------------------------------------------------------------------
+        # Discouraged in v2.11+, deprecated in v3. Kept here for compatibility.
+        populate_by_name=True,
+        # ---------------------------------------------------------------------------
+        # Introduced in v2.11, ignored in earlier versions
+        validate_by_name=True,
+        validate_by_alias=True,
+        serialize_by_alias=True,
+        # ---------------------------------------------------------------------------
         validate_assignment=True,
-        validate_default=True,
         use_attribute_docstrings=True,
         from_attributes=True,
-        revalidate_instances="always",
-        protected_namespaces=(),  # Some GraphQL fields may begin with "model_"
     )
 
+    # Custom defaults keyword args for `JsonableModel.model_{dump,dump_json}`:
+    # - by_alias: Convert keys to JSON-ready names and objects to JSON-ready dicts.
+    # - round_trip: Ensure round-trippable result
+    __DUMP_DEFAULTS: ClassVar[ModelDumpKwargs] = dict(by_alias=True, round_trip=True)
+
     @override
     def model_dump(
         self,
@@ -64,7 +85,7 @@ class GQLBase(CompatBaseModel):
         mode: Literal["json", "python"] | str = "json",  # NOTE: changed default
         **kwargs: Unpack[ModelDumpKwargs],
     ) -> dict[str, Any]:
-        kwargs = {**MODEL_DUMP_DEFAULTS, **kwargs}
+        kwargs = {**self.__DUMP_DEFAULTS, **kwargs}  # allows overrides, if needed
         return super().model_dump(mode=mode, **kwargs)
 
     @override
@@ -74,5 +95,14 @@ class GQLBase(CompatBaseModel):
         indent: int | None = None,
         **kwargs: Unpack[ModelDumpKwargs],
     ) -> str:
-        kwargs = {**MODEL_DUMP_DEFAULTS, **kwargs}
+        kwargs = {**self.__DUMP_DEFAULTS, **kwargs}  # allows overrides, if needed
         return super().model_dump_json(indent=indent, **kwargs)
+
+
+# Base class for all GraphQL-generated types.
+class GQLBase(JsonableModel, ABC):
+    model_config = ConfigDict(
+        validate_default=True,
+        revalidate_instances="always",
+        protected_namespaces=(),  # Some GraphQL fields may begin with "model_"
+    )

wandb/apis/__init__.py

@@ -1,5 +1,7 @@
 """api."""
 
+from __future__ import annotations
+
 from typing import Callable
 
 import requests

wandb/apis/attrs.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Any, MutableMapping
 
 import wandb

wandb/apis/importers/internals/internal.py

@@ -8,12 +8,10 @@ from pathlib import Path
 from typing import Any, Dict, Iterable, Optional
 
 import numpy as np
-from google.protobuf.json_format import ParseDict
 from tenacity import retry, stop_after_attempt, wait_random_exponential
 
 from wandb import Artifact
 from wandb.proto import wandb_internal_pb2 as pb
-from wandb.proto import wandb_settings_pb2
 from wandb.proto import wandb_telemetry_pb2 as telem_pb
 from wandb.sdk.interface.interface import file_policy_to_enum
 from wandb.sdk.interface.interface_queue import InterfaceQueue
@@ -310,27 +308,22 @@ def _make_settings(
 ) -> SettingsStatic:
     _settings_override = coalesce(settings_override, {})
 
-    default_settings: Dict[str, Any] = {
-        "x_files_dir": os.path.join(root_dir, "files"),
-        "root_dir": root_dir,
-        "sync_file": os.path.join(root_dir, "txlog.wandb"),
-        "resume": "false",
-        "program": None,
-        "ignore_globs": [],
-        "disable_job_creation": True,
-        "x_start_time": 0,
-        "_offline": None,
-        "x_sync": True,
-        "x_live_policy_rate_limit": 15,  # matches dir_watcher
-        "x_live_policy_wait_time": 600,  # matches dir_watcher
-        "x_file_stream_timeout_seconds": 60,
-    }
-
-    combined_settings = {**default_settings, **_settings_override}
-    settings_message = wandb_settings_pb2.Settings()
-    ParseDict(combined_settings, settings_message)
-
-    return SettingsStatic(settings_message)
+    return SettingsStatic(
+        {
+            "x_files_dir": os.path.join(root_dir, "files"),
+            "root_dir": root_dir,
+            "resume": "never",
+            "program": None,
+            "ignore_globs": [],
+            "disable_job_creation": True,
+            "x_start_time": 0,
+            "x_sync": True,
+            "x_live_policy_rate_limit": 15,  # matches dir_watcher
+            "x_live_policy_wait_time": 600,  # matches dir_watcher
+            "x_file_stream_timeout_seconds": 60,
+            **_settings_override,
+        }
+    )
 
 
 def send_run(

wandb/apis/internal.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Any
 
 from wandb.sdk.internal.internal_api import Api as InternalApi

wandb/apis/normalize.py

@@ -1,5 +1,7 @@
 """normalize."""
 
+from __future__ import annotations
+
 import ast
 import sys
 from functools import wraps

wandb/apis/public/__init__.py

@@ -10,7 +10,7 @@ __all__ = (
     "ArtifactTypes",
     "RunArtifacts",
     "Automations",
-    "File",  # doc:exclude
+    "File",
     "Files",
     "HistoryScan",  # doc:exclude
     "SampledHistoryScan",  # doc:exclude
@@ -27,6 +27,7 @@ __all__ = (
     "Sweeps",
     "QueryGenerator",  # doc:exclude
     "Registry",
+    "Registries",  # doc:exclude
     "BetaReport",
     "PanelMetricsHelper",  # doc:exclude
     "PythonMongoishQueryGenerator",  # doc:exclude
@@ -64,7 +65,7 @@ from wandb.apis.public.jobs import (
 )
 from wandb.apis.public.projects import PROJECT_FRAGMENT, Project, Projects, Sweeps
 from wandb.apis.public.query_generator import QueryGenerator
-from wandb.apis.public.registries.registry import Registry
+from wandb.apis.public.registries import Registries, Registry
 from wandb.apis.public.reports import (
     BetaReport,
     PanelMetricsHelper,