wandb 0.16.4__py3-none-any.whl → 0.16.5__py3-none-any.whl

wandb/__init__.py

@@ -11,8 +11,8 @@ For scripts and interactive notebooks, see https://github.com/wandb/examples.
 
 For reference documentation, see https://docs.wandb.com/ref/python.
 """
-__version__ = "0.16.4"
-_minimum_core_version = "0.17.0b9"
+__version__ = "0.16.5"
+_minimum_core_version = "0.17.0b10"
 
 # Used with pypi checks and other messages related to pip
 _wandb_module = "wandb"

wandb/agents/pyagent.py

@@ -347,7 +347,7 @@ def pyagent(sweep_id, function, entity=None, project=None, count=None):
         count (int, optional): the number of trials to run.
     """
     if not callable(function):
-        raise Exception("function paramter must be callable!")
+        raise Exception("function parameter must be callable!")
     agent = Agent(
         sweep_id,
         function=function,

wandb/apis/public/api.py

@@ -313,15 +313,15 @@ class Api:
             entity: (str) Optional name of the entity to create the queue. If None, will use the configured or default entity.
             prioritization_mode: (str) Optional version of prioritization to use. Either "V0" or None
             config: (dict) Optional default resource configuration to be used for the queue. Use handlebars (eg. "{{var}}") to specify template variables.
-            template_variables (dict): A dictionary of template variable schemas to be used with the config. Expected format of:
+            template_variables: (dict) A dictionary of template variable schemas to be used with the config. Expected format of:
                 {
                     "var-name": {
                         "schema": {
-                            "type": "<string | number | integer>",
-                            "default": <optional value>,
-                            "minimum": <optional minimum>,
-                            "maximum": <optional maximum>,
-                            "enum": [..."<options>"]
+                            "type": ("string", "number", or "integer"),
+                            "default": (optional value),
+                            "minimum": (optional minimum),
+                            "maximum": (optional maximum),
+                            "enum": [..."(options)"]
                         }
                     }
                 }

wandb/apis/reports/v2/interface.py

@@ -4,6 +4,8 @@ from datetime import datetime
 from typing import Dict, Iterable, Optional, Tuple, Union
 from typing import List as LList
 
+from annotated_types import Annotated, Ge, Le
+
 try:
     from typing import Literal
 except ImportError:
@@ -758,14 +760,8 @@ block_mapping = {
 
 @dataclass(config=dataclass_config)
 class GradientPoint(Base):
-    color: str
-    offset: float = Field(0, ge=0, le=100)
-
-    @validator("color")
-    def validate_color(cls, v):  # noqa: N805
-        if not internal.is_valid_color(v):
-            raise ValueError("invalid color, value should be hex, rgb, or rgba")
-        return v
+    color: Annotated[str, internal.ColorStrConstraints]
+    offset: Annotated[float, Ge(0), Le(100)] = 0
 
     def to_model(self):
         return internal.GradientPoint(color=self.color, offset=self.offset)

wandb/apis/reports/v2/internal.py

@@ -1,18 +1,19 @@
 """JSONSchema for internal types.  Hopefully this is auto-generated one day!"""
 import json
 import random
-import re
 from copy import deepcopy
 from datetime import datetime
 from typing import Any, Dict, Optional, Tuple, Union
 from typing import List as LList
 
+from annotated_types import Annotated, Ge, Le
+
 try:
     from typing import Literal
 except ImportError:
     from typing_extensions import Literal
 
-from pydantic import BaseModel, ConfigDict, Field, validator
+from pydantic import BaseModel, ConfigDict, Field, StringConstraints, validator
 from pydantic.alias_generators import to_camel
 
 
@@ -48,6 +49,13 @@ def _generate_name(length: int = 12) -> str:
     return rand36.lower()[:length]
 
 
+hex_pattern = r"^#(?:[0-9a-fA-F]{3}){1,2}$"
+rgb_pattern = r"^rgb\(\s*(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\s*,\s*(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\s*,\s*(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\s*\)$"
+rgba_pattern = r"^rgba\(\s*(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\s*,\s*(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\s*,\s*(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\s*,\s*(1|0|0?\.\d+)\s*\)$"
+ColorStrConstraints = StringConstraints(
+    pattern=f"{hex_pattern}|{rgb_pattern}|{rgba_pattern}"
+)
+
 LinePlotStyle = Literal["line", "stacked-area", "pct-area"]
 BarPlotStyle = Literal["bar", "boxplot", "violin"]
 FontSize = Literal["small", "medium", "large", "auto"]
@@ -609,14 +617,8 @@ class LinePlot(Panel):
 
 
 class GradientPoint(ReportAPIBaseModel):
-    color: str
-    offset: float = Field(0, ge=0, le=100)
-
-    @validator("color")
-    def validate_color(cls, v):  # noqa: N805
-        if not is_valid_color(v):
-            raise ValueError("invalid color, value should be hex, rgb, or rgba")
-        return v
+    color: Annotated[str, ColorStrConstraints]
+    offset: Annotated[float, Ge(0), Le(100)] = 0
 
 
 class ScatterPlotConfig(ReportAPIBaseModel):
@@ -866,38 +868,3 @@ block_type_mapping = {
     "table-of-contents": TableOfContents,
     "block-quote": BlockQuote,
 }
-
-
-def is_valid_color(color_str: str) -> bool:
-    # Regular expression for hex color validation
-    hex_color_pattern = r"^#(?:[0-9a-fA-F]{3}){1,2}$"
-
-    # Check if it's a valid hex color
-    if re.match(hex_color_pattern, color_str):
-        return True
-
-    # Try parsing it as an RGB or RGBA tuple
-    try:
-        # Strip 'rgb(' or 'rgba(' and the closing ')'
-        if color_str.startswith("rgb(") and color_str.endswith(")"):
-            parts = color_str[4:-1].split(",")
-        elif color_str.startswith("rgba(") and color_str.endswith(")"):
-            parts = color_str[5:-1].split(",")
-        else:
-            return False
-
-        # Convert parts to integers and validate ranges
-        parts = [int(p.strip()) for p in parts]
-        if len(parts) == 3 and all(0 <= p <= 255 for p in parts):
-            return True  # Valid RGB
-        if (
-            len(parts) == 4
-            and all(0 <= p <= 255 for p in parts[:-1])
-            and 0 <= parts[-1] <= 1
-        ):
-            return True  # Valid RGBA
-
-    except ValueError:
-        pass
-
-    return False

wandb/cli/cli.py

@@ -2345,8 +2345,29 @@ def artifact():
     default=None,
     help="Resume the last run from your current directory.",
 )
+@click.option(
+    "--skip_cache",
+    is_flag=True,
+    default=False,
+    help="Skip caching while uploading artifact files.",
+)
+@click.option(
+    "--policy",
+    default="mutable",
+    help="Set the storage policy while uploading artifact files.",
+)
 @display_error
-def put(path, name, description, type, alias, run_id, resume):
+def put(
+    path,
+    name,
+    description,
+    type,
+    alias,
+    run_id,
+    resume,
+    skip_cache,
+    policy,
+):
     if name is None:
         name = os.path.basename(path)
     public_api = PublicApi()
@@ -2361,10 +2382,10 @@ def put(path, name, description, type, alias, run_id, resume):
     artifact_path = f"{entity}/{project}/{artifact_name}:{alias[0]}"
     if os.path.isdir(path):
         wandb.termlog(f'Uploading directory {path} to: "{artifact_path}" ({type})')
-        artifact.add_dir(path)
+        artifact.add_dir(path, skip_cache=skip_cache, policy=policy)
     elif os.path.isfile(path):
         wandb.termlog(f'Uploading file {path} to: "{artifact_path}" ({type})')
-        artifact.add_file(path)
+        artifact.add_file(path, skip_cache=skip_cache, policy=policy)
     elif "://" in path:
         wandb.termlog(
             f'Logging reference artifact from {path} to: "{artifact_path}" ({type})'

wandb/integration/ultralytics/callback.py

@@ -321,7 +321,6 @@ class WandBUltralyticsCallback:
                     )
             if self.enable_model_checkpointing:
                 self._save_model(trainer)
-            self.model.to("cpu")
             trainer.model.to(self.device)
 
     def on_train_end(self, trainer: TRAINER_TYPE):