wandb 0.19.8__py3-none-win_amd64.whl → 0.19.10__py3-none-win_amd64.whl

wandb/sdk/wandb_settings.py

@@ -11,30 +11,28 @@ import re
 import shutil
 import socket
 import sys
-import tempfile
 from datetime import datetime
-from typing import Any, Literal, Sequence
-from urllib.parse import quote, unquote, urlencode
 
-if sys.version_info >= (3, 11):
-    from typing import Self
-else:
-    from typing_extensions import Self
+# Optional and Union are used for type hinting instead of | because
+# the latter is not supported in pydantic<2.6 and Python<3.10.
+# Dict, List, and Tuple are used for backwards compatibility
+# with pydantic v1 and Python<3.9.
+from typing import Any, Callable, Dict, List, Literal, Optional, Sequence, Tuple, Union
+from urllib.parse import quote, unquote, urlencode
 
 from google.protobuf.wrappers_pb2 import BoolValue, DoubleValue, Int32Value, StringValue
-from pydantic import (
+from pydantic import BaseModel, ConfigDict, Field
+from typing_extensions import Self
+
+import wandb
+from wandb import env, termwarn, util
+from wandb._pydantic import (
+    IS_PYDANTIC_V2,
     AliasChoices,
-    BaseModel,
-    ConfigDict,
-    Field,
     computed_field,
     field_validator,
     model_validator,
 )
-from pydantic_core import SchemaValidator, core_schema
-
-import wandb
-from wandb import env, termwarn, util
 from wandb.errors import UsageError
 from wandb.proto import wandb_settings_pb2
 
@@ -42,6 +40,117 @@ from .lib import apikey, credentials, ipython
 from .lib.gitlib import GitRepo
 from .lib.run_moment import RunMoment
 
+validate_url: Callable[[str], None]
+
+if IS_PYDANTIC_V2:
+    from pydantic_core import SchemaValidator, core_schema
+
+    def validate_url(url: str) -> None:
+        """Validate a URL string."""
+        url_validator = SchemaValidator(
+            core_schema.url_schema(
+                allowed_schemes=["http", "https"],
+                strict=True,
+            )
+        )
+        url_validator.validate_python(url)
+else:
+    from pydantic import root_validator
+
+    def validate_url(url: str) -> None:
+        """Validate the base url of the wandb server.
+
+        param value: URL to validate
+
+        Based on the Django URLValidator, but with a few additional checks.
+
+        Copyright (c) Django Software Foundation and individual contributors.
+        All rights reserved.
+
+        Redistribution and use in source and binary forms, with or without modification,
+        are permitted provided that the following conditions are met:
+
+            1. Redistributions of source code must retain the above copyright notice,
+               this list of conditions and the following disclaimer.
+
+            2. Redistributions in binary form must reproduce the above copyright
+               notice, this list of conditions and the following disclaimer in the
+               documentation and/or other materials provided with the distribution.
+
+            3. Neither the name of Django nor the names of its contributors may be used
+               to endorse or promote products derived from this software without
+               specific prior written permission.
+
+        THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
+        ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+        WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+        DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
+        ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+        (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+        LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
+        ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+        (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+        SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+        """
+        from urllib.parse import urlparse, urlsplit
+
+        if url is None:
+            return
+
+        ul = "\u00a1-\uffff"  # Unicode letters range (must not be a raw string).
+
+        # IP patterns
+        ipv4_re = (
+            r"(?:0|25[0-5]|2[0-4][0-9]|1[0-9]?[0-9]?|[1-9][0-9]?)"
+            r"(?:\.(?:0|25[0-5]|2[0-4][0-9]|1[0-9]?[0-9]?|[1-9][0-9]?)){3}"
+        )
+        ipv6_re = r"\[[0-9a-f:.]+\]"  # (simple regex, validated later)
+
+        # Host patterns
+        hostname_re = (
+            r"[a-z" + ul + r"0-9](?:[a-z" + ul + r"0-9-]{0,61}[a-z" + ul + r"0-9])?"
+        )
+        # Max length for domain name labels is 63 characters per RFC 1034 sec. 3.1
+        domain_re = r"(?:\.(?!-)[a-z" + ul + r"0-9-]{1,63}(?<!-))*"
+        tld_re = (
+            r"\."  # dot
+            r"(?!-)"  # can't start with a dash
+            r"(?:[a-z" + ul + "-]{2,63}"  # domain label
+            r"|xn--[a-z0-9]{1,59})"  # or punycode label
+            r"(?<!-)"  # can't end with a dash
+            r"\.?"  # may have a trailing dot
+        )
+        # host_re = "(" + hostname_re + domain_re + tld_re + "|localhost)"
+        # todo?: allow hostname to be just a hostname (no tld)?
+        host_re = "(" + hostname_re + domain_re + f"({tld_re})?" + "|localhost)"
+
+        regex = re.compile(
+            r"^(?:[a-z0-9.+-]*)://"  # scheme is validated separately
+            r"(?:[^\s:@/]+(?::[^\s:@/]*)?@)?"  # user:pass authentication
+            r"(?:" + ipv4_re + "|" + ipv6_re + "|" + host_re + ")"
+            r"(?::[0-9]{1,5})?"  # port
+            r"(?:[/?#][^\s]*)?"  # resource path
+            r"\Z",
+            re.IGNORECASE,
+        )
+        schemes = {"http", "https"}
+        unsafe_chars = frozenset("\t\r\n")
+
+        scheme = url.split("://")[0].lower()
+        split_url = urlsplit(url)
+        parsed_url = urlparse(url)
+
+        if parsed_url.netloc == "":
+            raise ValueError(f"Invalid URL: {url}")
+        elif unsafe_chars.intersection(url):
+            raise ValueError("URL cannot contain unsafe characters")
+        elif scheme not in schemes:
+            raise ValueError("URL must start with `http(s)://`")
+        elif not regex.search(url):
+            raise ValueError(f"{url} is not a valid server address")
+        elif split_url.hostname is None or len(split_url.hostname) > 253:
+            raise ValueError("hostname is invalid")
+
 
 def _path_convert(*args: str) -> str:
     """Join path and apply os.path.expanduser to it."""
@@ -86,7 +195,7 @@ class Settings(BaseModel, validate_assignment=True):
     allow_val_change: bool = False
     """Flag to allow modification of `Config` values after they've been set."""
 
-    anonymous: Literal["allow", "must", "never"] | None = None
+    anonymous: Optional[Literal["allow", "must", "never"]] = None
     """Controls anonymous data logging.
 
     Possible values are:
@@ -100,19 +209,19 @@ class Settings(BaseModel, validate_assignment=True):
        signed-up user account.
     """
 
-    api_key: str | None = None
+    api_key: Optional[str] = None
     """The W&B API key."""
 
-    azure_account_url_to_access_key: dict[str, str] | None = None
+    azure_account_url_to_access_key: Optional[Dict[str, str]] = None
     """Mapping of Azure account URLs to their corresponding access keys for Azure integration."""
 
     base_url: str = "https://api.wandb.ai"
     """The URL of the W&B backend for data synchronization."""
 
-    code_dir: str | None = None
+    code_dir: Optional[str] = None
     """Directory containing the code to be tracked by W&B."""
 
-    config_paths: Sequence[str] | None = None
+    config_paths: Optional[Sequence[str]] = None
     """Paths to files to load configuration from into the `Config` object."""
 
     console: Literal["auto", "off", "wrap", "redirect", "wrap_raw", "wrap_emu"] = Field(
@@ -137,7 +246,7 @@ class Settings(BaseModel, validate_assignment=True):
 
       "wrap_emu" - Same as "wrap" but captures output through an emulator.
       Derived from the `wrap` setting and should not be set manually.
-      """
+    """
 
     console_multipart: bool = False
     """Whether to produce multipart console log files."""
@@ -156,57 +265,57 @@ class Settings(BaseModel, validate_assignment=True):
     disable_job_creation: bool = True
     """Whether to disable the creation of a job artifact for W&B Launch."""
 
-    docker: str | None = None
+    docker: Optional[str] = None
     """The Docker image used to execute the script."""
 
-    email: str | None = None
+    email: Optional[str] = None
     """The email address of the user."""
 
-    entity: str | None = None
+    entity: Optional[str] = None
     """The W&B entity, such as a user or a team."""
 
-    organization: str | None = None
+    organization: Optional[str] = None
     """The W&B organization."""
 
     force: bool = False
     """Whether to pass the `force` flag to `wandb.login()`."""
 
-    fork_from: RunMoment | None = None
+    fork_from: Optional[RunMoment] = None
     """Specifies a point in a previous execution of a run to fork from.
 
     The point is defined by the run ID, a metric, and its value.
     Currently, only the metric '_step' is supported.
     """
 
-    git_commit: str | None = None
+    git_commit: Optional[str] = None
     """The git commit hash to associate with the run."""
 
     git_remote: str = "origin"
     """The git remote to associate with the run."""
 
-    git_remote_url: str | None = None
+    git_remote_url: Optional[str] = None
     """The URL of the git remote repository."""
 
-    git_root: str | None = None
+    git_root: Optional[str] = None
     """Root directory of the git repository."""
 
     heartbeat_seconds: int = 30
     """Interval in seconds between heartbeat signals sent to the W&B servers."""
 
-    host: str | None = None
+    host: Optional[str] = None
     """Hostname of the machine running the script."""
 
-    http_proxy: str | None = None
+    http_proxy: Optional[str] = None
     """Custom proxy servers for http requests to W&B."""
 
-    https_proxy: str | None = None
+    https_proxy: Optional[str] = None
     """Custom proxy servers for https requests to W&B."""
 
     # Path to file containing an identity token (JWT) for authentication.
-    identity_token_file: str | None = None
+    identity_token_file: Optional[str] = None
     """Path to file containing an identity token (JWT) for authentication."""
 
-    ignore_globs: tuple[str, ...] = ()
+    ignore_globs: Sequence[str] = ()
     """Unix glob patterns relative to `files_dir` specifying files to exclude from upload."""
 
     init_timeout: float = 90.0
@@ -215,10 +324,10 @@ class Settings(BaseModel, validate_assignment=True):
     insecure_disable_ssl: bool = False
     """Whether to insecurely disable SSL verification."""
 
-    job_name: str | None = None
+    job_name: Optional[str] = None
     """Name of the Launch job running the script."""
 
-    job_source: Literal["repo", "artifact", "image"] | None = None
+    job_source: Optional[Literal["repo", "artifact", "image"]] = None
     """Source type for Launch."""
 
     label_disable: bool = False
@@ -227,10 +336,10 @@ class Settings(BaseModel, validate_assignment=True):
     launch: bool = False
     """Flag to indicate if the run is being launched through W&B Launch."""
 
-    launch_config_path: str | None = None
+    launch_config_path: Optional[str] = None
     """Path to the launch configuration file."""
 
-    login_timeout: float | None = None
+    login_timeout: Optional[float] = None
     """Time in seconds to wait for login operations before timing out."""
 
     mode: Literal["online", "offline", "dryrun", "disabled", "run", "shared"] = Field(
@@ -239,13 +348,13 @@ class Settings(BaseModel, validate_assignment=True):
     )
     """The operating mode for W&B logging and synchronization."""
 
-    notebook_name: str | None = None
+    notebook_name: Optional[str] = None
     """Name of the notebook if running in a Jupyter-like environment."""
 
-    program: str | None = None
+    program: Optional[str] = None
     """Path to the script that created the run, if available."""
 
-    program_abspath: str | None = None
+    program_abspath: Optional[str] = None
     """The absolute path from the root repository directory to the script that
     created the run.
 
@@ -253,26 +362,46 @@ class Settings(BaseModel, validate_assignment=True):
     .git directory, if it exists. Otherwise, it's the current working directory.
     """
 
-    program_relpath: str | None = None
+    program_relpath: Optional[str] = None
     """The relative path to the script that created the run."""
 
-    project: str | None = None
+    project: Optional[str] = None
     """The W&B project ID."""
 
     quiet: bool = False
     """Flag to suppress non-essential output."""
 
-    reinit: bool = False
-    """Flag to allow reinitialization of a run.
-
-    If not set, when an active run exists, calling `wandb.init()` returns the existing run
-    instead of creating a new one.
+    reinit: Union[
+        Literal[
+            "default",
+            "return_previous",
+            "finish_previous",
+            "create_new",
+        ],
+        bool,
+    ] = "default"
+    """What to do when `wandb.init()` is called while a run is active.
+
+    Options:
+    - "default": Use "finish_previous" in notebooks and "return_previous"
+        otherwise.
+    - "return_previous": Return the most recently created run
+        that is not yet finished. This does not update `wandb.run`; see
+        the "create_new" option.
+    - "finish_previous": Finish all active runs, then return a new run.
+    - "create_new": Create a new run without modifying other active runs.
+        Does not update `wandb.run` and top-level functions like `wandb.log`.
+        Because of this, some older integrations that rely on the global run
+        will not work.
+
+    Can also be a boolean, but this is deprecated. False is the same as
+    "return_previous", and True is the same as "finish_previous".
     """
 
     relogin: bool = False
     """Flag to force a new login attempt."""
 
-    resume: Literal["allow", "must", "never", "auto"] | None = None
+    resume: Optional[Literal["allow", "must", "never", "auto"]] = None
     """Specifies the resume behavior for the run.
 
     The available options are:
@@ -290,7 +419,7 @@ class Settings(BaseModel, validate_assignment=True):
       machine.
     """
 
-    resume_from: RunMoment | None = None
+    resume_from: Optional[RunMoment] = None
     """Specifies a point in a previous execution of a run to resume from.
 
     The point is defined by the run ID, a metric, and its value.
@@ -309,31 +438,31 @@ class Settings(BaseModel, validate_assignment=True):
     In particular, this is used to derive the wandb directory and the run directory.
     """
 
-    run_group: str | None = None
+    run_group: Optional[str] = None
     """Group identifier for related runs.
 
     Used for grouping runs in the UI.
     """
 
-    run_id: str | None = None
+    run_id: Optional[str] = None
     """The ID of the run."""
 
-    run_job_type: str | None = None
+    run_job_type: Optional[str] = None
     """Type of job being run (e.g., training, evaluation)."""
 
-    run_name: str | None = None
+    run_name: Optional[str] = None
     """Human-readable name for the run."""
 
-    run_notes: str | None = None
+    run_notes: Optional[str] = None
     """Additional notes or description for the run."""
 
-    run_tags: tuple[str, ...] | None = None
+    run_tags: Optional[Tuple[str, ...]] = None
     """Tags to associate with the run for organization and filtering."""
 
     sagemaker_disable: bool = False
     """Flag to disable SageMaker-specific functionality."""
 
-    save_code: bool | None = None
+    save_code: Optional[bool] = None
     """Whether to save the code associated with the run."""
 
     settings_system: str = Field(
@@ -343,10 +472,10 @@ class Settings(BaseModel, validate_assignment=True):
     )
     """Path to the system-wide settings file."""
 
-    show_colors: bool | None = None
+    show_colors: Optional[bool] = None
     """Whether to use colored output in the console."""
 
-    show_emoji: bool | None = None
+    show_emoji: Optional[bool] = None
     """Whether to show emoji in the console output."""
 
     show_errors: bool = True
@@ -361,10 +490,10 @@ class Settings(BaseModel, validate_assignment=True):
     silent: bool = False
     """Flag to suppress all output."""
 
-    start_method: str | None = None
+    start_method: Optional[str] = None
     """Method to use for starting subprocesses."""
 
-    strict: bool | None = None
+    strict: Optional[bool] = None
     """Whether to enable strict mode for validation and error checking."""
 
     summary_timeout: int = 60
@@ -373,10 +502,10 @@ class Settings(BaseModel, validate_assignment=True):
     summary_warnings: int = 5  # TODO: kill this with fire
     """Maximum number of summary warnings to display."""
 
-    sweep_id: str | None = None
+    sweep_id: Optional[str] = None
     """Identifier of the sweep this run belongs to."""
 
-    sweep_param_path: str | None = None
+    sweep_param_path: Optional[str] = None
     """Path to the sweep parameters configuration."""
 
     symlink: bool = Field(
@@ -384,13 +513,13 @@ class Settings(BaseModel, validate_assignment=True):
     )
     """Whether to use symlinks (True by default except on Windows)."""
 
-    sync_tensorboard: bool | None = None
+    sync_tensorboard: Optional[bool] = None
     """Whether to synchronize TensorBoard logs with W&B."""
 
     table_raise_on_max_row_limit_exceeded: bool = False
     """Whether to raise an exception when table row limits are exceeded."""
 
-    username: str | None = None
+    username: Optional[str] = None
     """Username."""
 
     # Internal settings.
@@ -424,13 +553,13 @@ class Settings(BaseModel, validate_assignment=True):
     x_disable_machine_info: bool = False
     """Flag to disable automatic machine info collection."""
 
-    x_executable: str | None = None
+    x_executable: Optional[str] = None
     """Path to the Python executable."""
 
-    x_extra_http_headers: dict[str, str] | None = None
+    x_extra_http_headers: Optional[Dict[str, str]] = None
     """Additional headers to add to all outgoing HTTP requests."""
 
-    x_file_stream_max_bytes: int | None = None
+    x_file_stream_max_bytes: Optional[int] = None
     """An approximate maximum request size for the filestream API.
 
     Its purpose is to prevent HTTP requests from failing due to
@@ -438,50 +567,50 @@ class Settings(BaseModel, validate_assignment=True):
     requests will be slightly larger.
     """
 
-    x_file_stream_max_line_bytes: int | None = None
+    x_file_stream_max_line_bytes: Optional[int] = None
     """Maximum line length for filestream JSONL files."""
 
-    x_file_stream_transmit_interval: float | None = None
+    x_file_stream_transmit_interval: Optional[float] = None
     """Interval in seconds between filestream transmissions."""
 
     # Filestream retry client configuration.
 
-    x_file_stream_retry_max: int | None = None
+    x_file_stream_retry_max: Optional[int] = None
     """Max number of retries for filestream operations."""
 
-    x_file_stream_retry_wait_min_seconds: float | None = None
+    x_file_stream_retry_wait_min_seconds: Optional[float] = None
     """Minimum wait time between retries for filestream operations."""
 
-    x_file_stream_retry_wait_max_seconds: float | None = None
+    x_file_stream_retry_wait_max_seconds: Optional[float] = None
     """Maximum wait time between retries for filestream operations."""
 
-    x_file_stream_timeout_seconds: float | None = None
+    x_file_stream_timeout_seconds: Optional[float] = None
     """Timeout in seconds for individual filestream HTTP requests."""
 
     # file transfer retry client configuration
 
-    x_file_transfer_retry_max: int | None = None
+    x_file_transfer_retry_max: Optional[int] = None
     """Max number of retries for file transfer operations."""
 
-    x_file_transfer_retry_wait_min_seconds: float | None = None
+    x_file_transfer_retry_wait_min_seconds: Optional[float] = None
     """Minimum wait time between retries for file transfer operations."""
 
-    x_file_transfer_retry_wait_max_seconds: float | None = None
+    x_file_transfer_retry_wait_max_seconds: Optional[float] = None
     """Maximum wait time between retries for file transfer operations."""
 
-    x_file_transfer_timeout_seconds: float | None = None
+    x_file_transfer_timeout_seconds: Optional[float] = None
     """Timeout in seconds for individual file transfer HTTP requests."""
 
-    x_files_dir: str | None = None
+    x_files_dir: Optional[str] = None
     """Override setting for the computed files_dir.."""
 
-    x_flow_control_custom: bool | None = None
+    x_flow_control_custom: Optional[bool] = None
     """Flag indicating custom flow control for filestream.
 
     TODO: Not implemented in wandb-core.
     """
 
-    x_flow_control_disabled: bool | None = None
+    x_flow_control_disabled: Optional[bool] = None
     """Flag indicating flow control is disabled for filestream.
 
     TODO: Not implemented in wandb-core.
@@ -489,47 +618,47 @@ class Settings(BaseModel, validate_assignment=True):
 
     # graphql retry client configuration
 
-    x_graphql_retry_max: int | None = None
+    x_graphql_retry_max: Optional[int] = None
     """Max number of retries for GraphQL operations."""
 
-    x_graphql_retry_wait_min_seconds: float | None = None
+    x_graphql_retry_wait_min_seconds: Optional[float] = None
     """Minimum wait time between retries for GraphQL operations."""
 
-    x_graphql_retry_wait_max_seconds: float | None = None
+    x_graphql_retry_wait_max_seconds: Optional[float] = None
     """Maximum wait time between retries for GraphQL operations."""
 
-    x_graphql_timeout_seconds: float | None = None
+    x_graphql_timeout_seconds: Optional[float] = None
     """Timeout in seconds for individual GraphQL requests."""
 
     x_internal_check_process: float = 8.0
     """Interval for internal process health checks in seconds."""
 
-    x_jupyter_name: str | None = None
+    x_jupyter_name: Optional[str] = None
     """Name of the Jupyter notebook."""
 
-    x_jupyter_path: str | None = None
+    x_jupyter_path: Optional[str] = None
     """Path to the Jupyter notebook."""
 
-    x_jupyter_root: str | None = None
+    x_jupyter_root: Optional[str] = None
     """Root directory of the Jupyter notebook."""
 
-    x_label: str | None = None
+    x_label: Optional[str] = None
     """Label to assign to system metrics and console logs collected for the run.
 
     This is used to group data by on the frontend and can be used to distinguish data
     from different processes in a distributed training job.
     """
 
-    x_live_policy_rate_limit: int | None = None
+    x_live_policy_rate_limit: Optional[int] = None
     """Rate limit for live policy updates in seconds."""
 
-    x_live_policy_wait_time: int | None = None
+    x_live_policy_wait_time: Optional[int] = None
     """Wait time between live policy updates in seconds."""
 
     x_log_level: int = logging.INFO
     """Logging level for internal operations."""
 
-    x_network_buffer: int | None = None
+    x_network_buffer: Optional[int] = None
     """Size of the network buffer used in flow control.
 
     TODO: Not implemented in wandb-core.
@@ -545,14 +674,14 @@ class Settings(BaseModel, validate_assignment=True):
     as the primary process handles the main logging.
     """
 
-    x_proxies: dict[str, str] | None = None
+    x_proxies: Optional[Dict[str, str]] = None
     """Custom proxy servers for requests to W&B.
 
     This is deprecated and will be removed in future versions.
     Please use `http_proxy` and `https_proxy` instead.
     """
 
-    x_runqueue_item_id: str | None = None
+    x_runqueue_item_id: Optional[str] = None
     """ID of the Launch run queue item being processed."""
 
     x_require_legacy_service: bool = False
@@ -567,13 +696,19 @@ class Settings(BaseModel, validate_assignment=True):
     This does not disable user-provided summary updates.
     """
 
-    x_service_transport: str | None = None
+    x_server_side_expand_glob_metrics: bool = False
+    """Flag to delegate glob matching of metrics in define_metric to the server.
+
+    If the server does not support this, the client will perform the glob matching.
+    """
+
+    x_service_transport: Optional[str] = None
     """Transport method for communication with the wandb service."""
 
     x_service_wait: float = 30.0
     """Time in seconds to wait for the wandb-core internal service to start."""
 
-    x_start_time: float | None = None
+    x_start_time: Optional[float] = None
     """The start time of the run in seconds since the Unix epoch."""
 
     x_stats_pid: int = os.getpid()
@@ -582,13 +717,13 @@ class Settings(BaseModel, validate_assignment=True):
     x_stats_sampling_interval: float = Field(default=15.0)
     """Sampling interval for the system monitor in seconds."""
 
-    x_stats_neuron_monitor_config_path: str | None = None
+    x_stats_neuron_monitor_config_path: Optional[str] = None
     """Path to the default config file for the neuron-monitor tool.
 
     This is used to monitor AWS Trainium devices.
     """
 
-    x_stats_dcgm_exporter: str | None = None
+    x_stats_dcgm_exporter: Optional[str] = None
     """Endpoint to extract Nvidia DCGM metrics from.
 
     Two options are supported:
@@ -602,12 +737,12 @@ class Settings(BaseModel, validate_assignment=True):
     - TODO: `http://192.168.0.1:9400/metrics`.
     """
 
-    x_stats_open_metrics_endpoints: dict[str, str] | None = None
+    x_stats_open_metrics_endpoints: Optional[Dict[str, str]] = None
     """OpenMetrics `/metrics` endpoints to monitor for system metrics."""
 
-    x_stats_open_metrics_filters: dict[str, dict[str, str]] | Sequence[str] | None = (
-        None
-    )
+    x_stats_open_metrics_filters: Union[
+        Dict[str, Dict[str, str]], Sequence[str], None
+    ] = None
     """Filter to apply to metrics collected from OpenMetrics `/metrics` endpoints.
 
     Supports two formats:
@@ -615,17 +750,17 @@ class Settings(BaseModel, validate_assignment=True):
     - ("metric regex pattern 1", "metric regex pattern 2", ...)
     """
 
-    x_stats_open_metrics_http_headers: dict[str, str] | None = None
+    x_stats_open_metrics_http_headers: Optional[Dict[str, str]] = None
     """HTTP headers to add to OpenMetrics requests."""
 
-    x_stats_disk_paths: Sequence[str] | None = Field(
+    x_stats_disk_paths: Optional[Sequence[str]] = Field(
         default_factory=lambda: ("/", "/System/Volumes/Data")
         if platform.system() == "Darwin"
         else ("/",)
     )
     """System paths to monitor for disk usage."""
 
-    x_stats_gpu_device_ids: Sequence[int] | None = None
+    x_stats_gpu_device_ids: Optional[Sequence[int]] = None
     """GPU device indices to monitor.
 
     If not set, captures metrics for all GPUs.
@@ -665,24 +800,39 @@ class Settings(BaseModel, validate_assignment=True):
                 new_values[key] = values[key]
         return new_values
 
-    @model_validator(mode="after")
-    def validate_mutual_exclusion_of_branching_args(self) -> Self:
-        if (
-            sum(
-                o is not None
-                for o in [
-                    self.fork_from,
-                    self.resume,
-                    self.resume_from,
-                ]
-            )
-            > 1
-        ):
-            raise ValueError(
-                "`fork_from`, `resume`, or `resume_from` are mutually exclusive. "
-                "Please specify only one of them."
-            )
-        return self
+    if IS_PYDANTIC_V2:
+
+        @model_validator(mode="after")
+        def validate_mutual_exclusion_of_branching_args(self) -> Self:
+            if (
+                sum(
+                    o is not None
+                    for o in [self.fork_from, self.resume, self.resume_from]
+                )
+                > 1
+            ):
+                raise ValueError(
+                    "`fork_from`, `resume`, or `resume_from` are mutually exclusive. "
+                    "Please specify only one of them."
+                )
+            return self
+    else:
+
+        @root_validator(pre=False)  # type: ignore [call-overload]
+        @classmethod
+        def validate_mutual_exclusion_of_branching_args(cls, values):
+            if (
+                sum(
+                    values.get(o) is not None
+                    for o in ["fork_from", "resume", "resume_from"]
+                )
+                > 1
+            ):
+                raise ValueError(
+                    "`fork_from`, `resume`, or `resume_from` are mutually exclusive. "
+                    "Please specify only one of them."
+                )
+            return values
 
     # Field validators.
 
@@ -707,7 +857,7 @@ class Settings(BaseModel, validate_assignment=True):
     @field_validator("base_url", mode="after")
     @classmethod
     def validate_base_url(cls, value):
-        cls.validate_url(value)
+        validate_url(value)
         # wandb.ai-specific checks
         if re.match(r".*wandb\.ai[^\.]*$", value) and "api." not in value:
             # user might guess app.wandb.ai or wandb.ai is the default cloud server
@@ -728,13 +878,21 @@ class Settings(BaseModel, validate_assignment=True):
 
     @field_validator("console", mode="after")
     @classmethod
-    def validate_console(cls, value, info):
+    def validate_console(cls, value, values):
         if value != "auto":
             return value
+
+        if hasattr(values, "data"):
+            # pydantic v2
+            values = values.data
+        else:
+            # pydantic v1
+            values = values
+
         if (
             ipython.in_jupyter()
-            or (info.data.get("start_method") == "thread")
-            or not info.data.get("disable_service")
+            or (values.get("start_method") == "thread")
+            or not values.get("x_disable_service")
             or platform.system() == "Windows"
         ):
             value = "wrap"
@@ -767,12 +925,20 @@ class Settings(BaseModel, validate_assignment=True):
 
     @field_validator("fork_from", mode="before")
     @classmethod
-    def validate_fork_from(cls, value, info) -> RunMoment | None:
+    def validate_fork_from(cls, value, values) -> Optional[RunMoment]:
         run_moment = cls._runmoment_preprocessor(value)
+
+        if hasattr(values, "data"):
+            # pydantic v2
+            values = values.data
+        else:
+            # pydantic v1
+            values = values
+
         if (
             run_moment
-            and info.data.get("run_id") is not None
-            and info.data.get("run_id") == run_moment.run
+            and values.get("run_id") is not None
+            and values.get("run_id") == run_moment.run
         ):
             raise ValueError(
                 "Provided `run_id` is the same as the run to `fork_from`. "
@@ -786,7 +952,7 @@ class Settings(BaseModel, validate_assignment=True):
     def validate_http_proxy(cls, value):
         if value is None:
             return None
-        cls.validate_url(value)
+        validate_url(value)
         return value.rstrip("/")
 
     @field_validator("https_proxy", mode="after")
@@ -794,7 +960,7 @@ class Settings(BaseModel, validate_assignment=True):
     def validate_https_proxy(cls, value):
         if value is None:
             return None
-        cls.validate_url(value)
+        validate_url(value)
         return value.rstrip("/")
 
     @field_validator("ignore_globs", mode="after")
@@ -828,7 +994,7 @@ class Settings(BaseModel, validate_assignment=True):
 
     @field_validator("project", mode="after")
     @classmethod
-    def validate_project(cls, value, info):
+    def validate_project(cls, value, values):
         if value is None:
             return None
         invalid_chars_list = list("/\\#?%:")
@@ -854,12 +1020,20 @@ class Settings(BaseModel, validate_assignment=True):
 
     @field_validator("resume_from", mode="before")
     @classmethod
-    def validate_resume_from(cls, value, info) -> RunMoment | None:
+    def validate_resume_from(cls, value, values) -> Optional[RunMoment]:
         run_moment = cls._runmoment_preprocessor(value)
+
+        if hasattr(values, "data"):
+            # pydantic v2
+            values = values.data
+        else:
+            # pydantic v1
+            values = values
+
         if (
             run_moment
-            and info.data.get("run_id") is not None
-            and info.data.get("run_id") != run_moment.run
+            and values.get("run_id") is not None
+            and values.get("run_id") != run_moment.run
         ):
             raise ValueError(
                 "Both `run_id` and `resume_from` have been specified with different ids."
@@ -876,7 +1050,7 @@ class Settings(BaseModel, validate_assignment=True):
 
     @field_validator("run_id", mode="after")
     @classmethod
-    def validate_run_id(cls, value, info):
+    def validate_run_id(cls, value, values):
         if value is None:
             return None
 
@@ -902,7 +1076,7 @@ class Settings(BaseModel, validate_assignment=True):
             raise UsageError("Service wait time cannot be negative")
         return value
 
-    @field_validator("start_method")
+    @field_validator("start_method", mode="after")
     @classmethod
     def validate_start_method(cls, value):
         if value is None:
@@ -984,7 +1158,7 @@ class Settings(BaseModel, validate_assignment=True):
 
     @computed_field  # type: ignore[prop-decorator]
     @property
-    def _args(self) -> list[str]:
+    def _args(self) -> List[str]:
         if not self._jupyter:
             return sys.argv[1:]
         return []
@@ -1006,7 +1180,7 @@ class Settings(BaseModel, validate_assignment=True):
 
     @computed_field  # type: ignore[prop-decorator]
     @property
-    def _code_path_local(self) -> str | None:
+    def _code_path_local(self) -> Optional[str]:
         """The relative path from the current working directory to the code path.
 
         For example, if the code path is /home/user/project/example.py, and the
@@ -1089,12 +1263,7 @@ class Settings(BaseModel, validate_assignment=True):
     @computed_field  # type: ignore[prop-decorator]
     @property
     def _tmp_code_dir(self) -> str:
-        return _path_convert(
-            self.wandb_dir,
-            f"{self.run_mode}-{self.timespec}-{self.run_id}",
-            "tmp",
-            "code",
-        )
+        return _path_convert(self.sync_dir, "tmp", "code")
 
     @computed_field  # type: ignore[prop-decorator]
     @property
@@ -1103,7 +1272,7 @@ class Settings(BaseModel, validate_assignment=True):
 
     @computed_field  # type: ignore[prop-decorator]
     @property
-    def colab_url(self) -> str | None:
+    def colab_url(self) -> Optional[str]:
         """The URL to the Colab notebook, if running in Colab."""
         if not self._colab:
             return None
@@ -1121,11 +1290,7 @@ class Settings(BaseModel, validate_assignment=True):
     @property
     def files_dir(self) -> str:
         """Absolute path to the local directory where the run's files are stored."""
-        return self.x_files_dir or _path_convert(
-            self.wandb_dir,
-            f"{self.run_mode}-{self.timespec}-{self.run_id}",
-            "files",
-        )
+        return self.x_files_dir or _path_convert(self.sync_dir, "files")
 
     @computed_field  # type: ignore[prop-decorator]
     @property
@@ -1136,9 +1301,7 @@ class Settings(BaseModel, validate_assignment=True):
     @property
     def log_dir(self) -> str:
         """The directory for storing log files."""
-        return _path_convert(
-            self.wandb_dir, f"{self.run_mode}-{self.timespec}-{self.run_id}", "logs"
-        )
+        return _path_convert(self.sync_dir, "logs")
 
     @computed_field  # type: ignore[prop-decorator]
     @property
@@ -1219,7 +1382,8 @@ class Settings(BaseModel, validate_assignment=True):
     @property
     def sync_dir(self) -> str:
         return _path_convert(
-            self.wandb_dir, f"{self.run_mode}-{self.timespec}-{self.run_id}"
+            self.wandb_dir,
+            f"{self.run_mode}-{self.timespec}-{self.run_id}",
         )
 
     @computed_field  # type: ignore[prop-decorator]
@@ -1241,29 +1405,13 @@ class Settings(BaseModel, validate_assignment=True):
     @computed_field  # type: ignore[prop-decorator]
     @property
     def wandb_dir(self) -> str:
-        """Full path to the wandb directory.
-
-        The setting exposed to users as `dir=` or `WANDB_DIR` is the `root_dir`.
-        We add the `__stage_dir__` to it to get the full `wandb_dir`
-        """
-        root_dir = self.root_dir or ""
-
-        # We use the hidden version if it already exists, otherwise non-hidden.
-        if os.path.exists(os.path.join(root_dir, ".wandb")):
-            __stage_dir__ = ".wandb" + os.sep
-        else:
-            __stage_dir__ = "wandb" + os.sep
-
-        path = os.path.join(root_dir, __stage_dir__)
-        if not os.access(root_dir or ".", os.W_OK):
-            termwarn(
-                f"Path {path} wasn't writable, using system temp directory.",
-                repeat=False,
-            )
-            path = os.path.join(
-                tempfile.gettempdir(), __stage_dir__ or ("wandb" + os.sep)
-            )
-
+        """Full path to the wandb directory."""
+        stage_dir = (
+            ".wandb" + os.sep
+            if os.path.exists(os.path.join(self.root_dir, ".wandb"))
+            else "wandb" + os.sep
+        )
+        path = os.path.join(self.root_dir, stage_dir)
         return os.path.expanduser(path)
 
     # Methods to collect and update settings from different sources.
@@ -1289,7 +1437,7 @@ class Settings(BaseModel, validate_assignment=True):
             if value is not None:
                 setattr(self, key, value)
 
-    def update_from_env_vars(self, environ: dict[str, Any]):
+    def update_from_env_vars(self, environ: Dict[str, Any]):
         """Update settings from environment variables."""
         env_prefix: str = "WANDB_"
         private_env_prefix: str = env_prefix + "_"
@@ -1407,7 +1555,7 @@ class Settings(BaseModel, validate_assignment=True):
 
         self.program = program
 
-    def update_from_dict(self, settings: dict[str, Any]) -> None:
+    def update_from_dict(self, settings: Dict[str, Any]) -> None:
         """Update settings from a dictionary."""
         for key, value in dict(settings).items():
             if value is not None:
@@ -1425,7 +1573,11 @@ class Settings(BaseModel, validate_assignment=True):
         """Generate a protobuf representation of the settings."""
         settings_proto = wandb_settings_pb2.Settings()
         for k, v in self.model_dump(exclude_none=True).items():
-            # special case for x_stats_open_metrics_filters
+            # Client-only settings that don't exist on the protobuf.
+            if k in ("reinit",):
+                continue
+
+            # Special case for x_stats_open_metrics_filters.
             if k == "x_stats_open_metrics_filters":
                 if isinstance(v, (list, set, tuple)):
                     setting = getattr(settings_proto, k)
@@ -1439,12 +1591,16 @@ class Settings(BaseModel, validate_assignment=True):
                     raise TypeError(f"Unsupported type {type(v)} for setting {k}")
                 continue
 
-            # special case for RunMoment fields
+            # Special case for RunMoment fields.
             if k in ("fork_from", "resume_from"):
-                run_moment = RunMoment(
-                    run=v.get("run"),
-                    value=v.get("value"),
-                    metric=v.get("metric"),
+                run_moment = (
+                    v
+                    if isinstance(v, RunMoment)
+                    else RunMoment(
+                        run=v.get("run"),
+                        value=v.get("value"),
+                        metric=v.get("metric"),
+                    )
                 )
                 getattr(settings_proto, k).CopyFrom(
                     wandb_settings_pb2.RunMoment(
@@ -1480,18 +1636,7 @@ class Settings(BaseModel, validate_assignment=True):
 
         return settings_proto
 
-    @staticmethod
-    def validate_url(url: str) -> None:
-        """Validate a URL string."""
-        url_validator = SchemaValidator(
-            core_schema.url_schema(
-                allowed_schemes=["http", "https"],
-                strict=True,
-            )
-        )
-        url_validator.validate_python(url)
-
-    def _get_program(self) -> str | None:
+    def _get_program(self) -> Optional[str]:
         """Get the program that started the current process."""
         if not self._jupyter:
             # If not in a notebook, try to get the program from the environment
@@ -1521,7 +1666,7 @@ class Settings(BaseModel, validate_assignment=True):
                 return self.x_jupyter_path
 
     @staticmethod
-    def _get_program_relpath(program: str, root: str | None = None) -> str | None:
+    def _get_program_relpath(program: str, root: Optional[str] = None) -> Optional[str]:
         """Get the relative path to the program from the root directory."""
         if not program:
             return None
@@ -1530,6 +1675,11 @@ class Settings(BaseModel, validate_assignment=True):
         if not root:
             return None
 
+        # For windows if the root and program are on different drives,
+        # os.path.relpath will raise a ValueError.
+        if not util.are_paths_on_same_drive(root, program):
+            return None
+
         full_path_to_program = os.path.join(
             root, os.path.relpath(os.getcwd(), root), program
         )
@@ -1547,7 +1697,7 @@ class Settings(BaseModel, validate_assignment=True):
         parser = configparser.ConfigParser()
         parser.add_section(section)
         parser.read(file_name)
-        config: dict[str, Any] = dict()
+        config: Dict[str, Any] = dict()
         for k in parser[section]:
             config[k] = parser[section][k]
             if k == "ignore_globs":
@@ -1573,9 +1723,79 @@ class Settings(BaseModel, validate_assignment=True):
         return f"?{urlencode({'apiKey': api_key})}"
 
     @staticmethod
-    def _runmoment_preprocessor(val: RunMoment | str | None) -> RunMoment | None:
+    def _runmoment_preprocessor(
+        val: Union[RunMoment, str, None],
+    ) -> Optional[RunMoment]:
         """Preprocess the setting for forking or resuming a run."""
         if isinstance(val, RunMoment) or val is None:
             return val
         elif isinstance(val, str):
             return RunMoment.from_uri(val)
+
+    if not IS_PYDANTIC_V2:
+
+        def model_copy(self, *args, **kwargs):
+            return self.copy(*args, **kwargs)
+
+        def model_dump(self, **kwargs):
+            """Compatibility method for Pydantic v1 to mimic v2's model_dump.
+
+            In v1, this is equivalent to dict() but also includes computed properties.
+
+            Args:
+                **kwargs: Options passed to the dict method
+                    - exclude_none: Whether to exclude fields with None values
+
+            Returns:
+                A dictionary of the model's fields and computed properties
+            """
+            # Handle exclude_none separately since it's named differently in v1
+            exclude_none = kwargs.pop("exclude_none", False)
+
+            # Start with regular fields from dict()
+            result = self.dict(**kwargs)
+
+            # Get all computed properties
+            for name in dir(self.__class__):
+                attr = getattr(self.__class__, name, None)
+                if isinstance(attr, property):
+                    try:
+                        # Only include properties that don't raise errors
+                        value = getattr(self, name)
+                        result[name] = value
+                    except (AttributeError, NotImplementedError, TypeError, ValueError):
+                        # Skip properties that can't be accessed or raise errors
+                        pass
+                elif isinstance(attr, RunMoment):
+                    value = getattr(self, name)
+                    result[name] = value
+
+            # Special Pydantic attributes that should always be excluded
+            exclude_fields = {
+                "model_config",
+                "model_fields",
+                "model_fields_set",
+                "__fields__",
+                "__model_fields_set",
+                "__pydantic_self__",
+                "__pydantic_initialised__",
+            }
+
+            # Remove special Pydantic attributes
+            for field in exclude_fields:
+                if field in result:
+                    del result[field]
+
+            if exclude_none:
+                # Remove None values from the result
+                return {k: v for k, v in result.items() if v is not None}
+
+            return result
+
+        @property
+        def model_fields_set(self) -> set:
+            """Return a set of fields that have been explicitly set.
+
+            This is a compatibility property for Pydantic v1 to mimic v2's model_fields_set.
+            """
+            return getattr(self, "__fields_set__", set())