wandb 0.19.10__py3-none-musllinux_1_2_aarch64.whl → 0.19.11__py3-none-musllinux_1_2_aarch64.whl

wandb/sdk/projects/_generated/fragments.py

@@ -0,0 +1,41 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/projects/
+
+from __future__ import annotations
+
+from typing import List, Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase, GQLId
+
+
+class RegistryFragment(GQLBase):
+    id: GQLId
+    allow_all_artifact_types_in_registry: bool = Field(
+        alias="allowAllArtifactTypesInRegistry"
+    )
+    artifact_types: RegistryFragmentArtifactTypes = Field(alias="artifactTypes")
+    name: str
+    description: Optional[str]
+    created_at: str = Field(alias="createdAt")
+    updated_at: Optional[str] = Field(alias="updatedAt")
+    access: Optional[str]
+
+
+class RegistryFragmentArtifactTypes(GQLBase):
+    edges: List[RegistryFragmentArtifactTypesEdges]
+
+
+class RegistryFragmentArtifactTypesEdges(GQLBase):
+    node: Optional[RegistryFragmentArtifactTypesEdgesNode]
+
+
+class RegistryFragmentArtifactTypesEdgesNode(GQLBase):
+    name: str
+
+
+RegistryFragment.model_rebuild()
+RegistryFragmentArtifactTypes.model_rebuild()
+RegistryFragmentArtifactTypesEdges.model_rebuild()
+RegistryFragmentArtifactTypesEdgesNode.model_rebuild()

wandb/sdk/projects/_generated/input_types.py

@@ -0,0 +1,13 @@
+# Generated by ariadne-codegen
+# Source: core/api/graphql/schemas/schema-latest.graphql
+
+from __future__ import annotations
+
+from typing import Optional
+
+from wandb._pydantic import GQLBase
+
+
+class ArtifactTypeInput(GQLBase):
+    name: str
+    description: Optional[str] = None

wandb/sdk/projects/_generated/operations.py

@@ -0,0 +1,88 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/projects/
+
+__all__ = [
+    "DELETE_PROJECT_GQL",
+    "FETCH_REGISTRY_GQL",
+    "RENAME_PROJECT_GQL",
+    "UPSERT_REGISTRY_PROJECT_GQL",
+]
+
+FETCH_REGISTRY_GQL = """
+query FetchRegistry($name: String, $entityName: String) {
+  entity(name: $entityName) {
+    project(name: $name) {
+      ...RegistryFragment
+    }
+  }
+}
+
+fragment RegistryFragment on Project {
+  id
+  allowAllArtifactTypesInRegistry
+  artifactTypes(includeAll: true) {
+    edges {
+      node {
+        name
+      }
+    }
+  }
+  name
+  description
+  createdAt
+  updatedAt
+  access
+}
+"""
+
+RENAME_PROJECT_GQL = """
+mutation renameProject($entityName: String!, $oldProjectName: String!, $newProjectName: String!) {
+  renameProject(
+    input: {entityName: $entityName, oldProjectName: $oldProjectName, newProjectName: $newProjectName}
+  ) {
+    project {
+      name
+    }
+    inserted
+  }
+}
+"""
+
+UPSERT_REGISTRY_PROJECT_GQL = """
+mutation UpsertRegistryProject($description: String, $entityName: String, $name: String, $access: String, $allowAllArtifactTypesInRegistry: Boolean, $artifactTypes: [ArtifactTypeInput!]) {
+  upsertModel(
+    input: {description: $description, entityName: $entityName, name: $name, access: $access, allowAllArtifactTypesInRegistry: $allowAllArtifactTypesInRegistry, artifactTypes: $artifactTypes}
+  ) {
+    project {
+      ...RegistryFragment
+    }
+    inserted
+  }
+}
+
+fragment RegistryFragment on Project {
+  id
+  allowAllArtifactTypesInRegistry
+  artifactTypes(includeAll: true) {
+    edges {
+      node {
+        name
+      }
+    }
+  }
+  name
+  description
+  createdAt
+  updatedAt
+  access
+}
+"""
+
+DELETE_PROJECT_GQL = """
+mutation deleteProject($id: String!) {
+  deleteModel(input: {id: $id}) {
+    success
+    __typename
+  }
+}
+"""

wandb/sdk/projects/_generated/rename_project.py

@@ -0,0 +1,27 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/projects/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+
+class RenameProject(GQLBase):
+    rename_project: Optional[RenameProjectRenameProject] = Field(alias="renameProject")
+
+
+class RenameProjectRenameProject(GQLBase):
+    project: Optional[RenameProjectRenameProjectProject]
+    inserted: Optional[bool]
+
+
+class RenameProjectRenameProjectProject(GQLBase):
+    name: str
+
+
+RenameProject.model_rebuild()
+RenameProjectRenameProject.model_rebuild()

wandb/sdk/projects/_generated/upsert_registry_project.py

@@ -0,0 +1,27 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/projects/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+from .fragments import RegistryFragment
+
+
+class UpsertRegistryProject(GQLBase):
+    upsert_model: Optional[UpsertRegistryProjectUpsertModel] = Field(
+        alias="upsertModel"
+    )
+
+
+class UpsertRegistryProjectUpsertModel(GQLBase):
+    project: Optional[RegistryFragment]
+    inserted: Optional[bool]
+
+
+UpsertRegistryProject.model_rebuild()
+UpsertRegistryProjectUpsertModel.model_rebuild()

wandb/sdk/service/service.py

@@ -15,7 +15,12 @@ import time
 from typing import TYPE_CHECKING, Any, Dict, Optional
 
 from wandb import _sentry
-from wandb.env import core_debug, error_reporting_enabled, is_require_legacy_service
+from wandb.env import (
+    core_debug,
+    dcgm_profiling_enabled,
+    error_reporting_enabled,
+    is_require_legacy_service,
+)
 from wandb.errors import Error, WandbCoreNotAvailableError
 from wandb.errors.term import termlog, termwarn
 from wandb.util import get_core_path, get_module
@@ -162,6 +167,9 @@ class _Service:
                 if core_debug(default="False"):
                     service_args.extend(["--log-level", "-4"])
 
+                if dcgm_profiling_enabled():
+                    service_args.append("--enable-dcgm-profiling")
+
                 exec_cmd_list = []
             else:
                 service_args.extend(["wandb", "service", "--debug"])

wandb/sdk/wandb_init.py

@@ -984,6 +984,9 @@ class _WandbInit:
             if settings.x_label:
                 tel.feature.user_provided_label = True
 
+            if wandb.env.dcgm_profiling_enabled():
+                tel.feature.dcgm_profiling_enabled = True
+
             tel.env.maybe_mp = _maybe_mp_process(backend)
 
         if not settings.label_disable:
@@ -1280,16 +1283,40 @@ def try_create_root_dir(settings: Settings) -> None:
             This function may update the root_dir to a temporary directory
             if the parent directory is not writable.
     """
+    fallback_to_temp_dir = False
+
     try:
-        if not os.path.exists(settings.root_dir):
-            os.makedirs(settings.root_dir, exist_ok=True)
+        os.makedirs(settings.root_dir, exist_ok=True)
     except OSError:
-        temp_dir = tempfile.gettempdir()
         wandb.termwarn(
-            f"Path {settings.root_dir} wasn't writable, using system temp directory {temp_dir}.",
+            f"Unable to create root directory {settings.root_dir}",
             repeat=False,
         )
-        settings.root_dir = temp_dir
+        fallback_to_temp_dir = True
+    else:
+        if not os.access(settings.root_dir, os.W_OK | os.R_OK):
+            wandb.termwarn(
+                f"Path {settings.root_dir} wasn't read/writable",
+                repeat=False,
+            )
+            fallback_to_temp_dir = True
+
+    if not fallback_to_temp_dir:
+        return
+
+    tmp_dir = tempfile.gettempdir()
+    if not os.access(tmp_dir, os.W_OK | os.R_OK):
+        raise ValueError(
+            f"System temp directory ({tmp_dir}) is not writable/readable, "
+            "please set the `dir` argument in `wandb.init()` to a writable/readable directory."
+        )
+
+    settings.root_dir = tmp_dir
+    wandb.termwarn(
+        f"Falling back to temporary directory {tmp_dir}.",
+        repeat=False,
+    )
+    os.makedirs(settings.root_dir, exist_ok=True)
 
 
 def init(  # noqa: C901

wandb/sdk/wandb_run.py

@@ -1222,7 +1222,14 @@ class Run:
             )
             return None
 
-        return self._log_artifact(art)
+        artifact = self._log_artifact(art)
+
+        self._config.update(
+            {"_wandb": {"code_path": artifact.name}},
+            allow_val_change=True,
+        )
+
+        return artifact
 
     @_log_to_run
     def get_sweep_url(self) -> str | None:
@@ -3086,7 +3093,7 @@ class Run:
         artifact: Artifact,
         target_path: str,
         aliases: list[str] | None = None,
-    ) -> None:
+    ) -> Artifact | None:
         """Link the given artifact to a portfolio (a promoted collection of artifacts).
 
         The linked artifact will be visible in the UI for the specified portfolio.
@@ -3100,7 +3107,7 @@ class Run:
             The alias "latest" will always be applied to the latest version of an artifact that is linked.
 
         Returns:
-            None
+            The linked artifact if linking was successful, otherwise None.
 
         """
         portfolio, project, entity = wandb.util._parse_entity_project_item(target_path)
@@ -3108,7 +3115,7 @@ class Run:
             aliases = []
 
         if not self._backend or not self._backend.interface:
-            return
+            return None
 
         if artifact.is_draft() and not artifact._is_draft_save_started():
             artifact = self._log_artifact(artifact)
@@ -3122,12 +3129,13 @@ class Run:
 
         organization = ""
         if is_artifact_registry_project(project):
-            organization = entity
+            organization = entity or self.settings.organization or ""
             # In a Registry linking, the entity is used to fetch the organization of the artifact
             # therefore the source artifact's entity is passed to the backend
             entity = artifact._source_entity
+        project = project or self.project
+        entity = entity or self.entity
         handle = self._backend.interface.deliver_link_artifact(
-            self,
             artifact,
             portfolio,
             aliases,
@@ -3143,6 +3151,24 @@ class Run:
         response = result.response.link_artifact_response
         if response.error_message:
             wandb.termerror(response.error_message)
+        if response.version_index is None:
+            wandb.termerror(
+                "Error fetching the linked artifact's version index after linking"
+            )
+            return None
+
+        try:
+            artifact_name = f"{entity}/{project}/{portfolio}:v{response.version_index}"
+            if is_artifact_registry_project(project):
+                if organization:
+                    artifact_name = f"{organization}/{project}/{portfolio}:v{response.version_index}"
+                else:
+                    artifact_name = f"{project}/{portfolio}:v{response.version_index}"
+            linked_artifact = self._public_api()._artifact(artifact_name)
+        except Exception as e:
+            wandb.termerror(f"Error fetching link artifact after linking: {e}")
+            return None
+        return linked_artifact
 
     @_log_to_run
     @_raise_if_finished
@@ -3687,7 +3713,7 @@ class Run:
         registered_model_name: str,
         name: str | None = None,
         aliases: list[str] | None = None,
-    ) -> None:
+    ) -> Artifact | None:
         """Log a model artifact version and link it to a registered model in the model registry.
 
         The linked model version will be visible in the UI for the specified registered model.
@@ -3749,7 +3775,7 @@ class Run:
             ValueError: if name has invalid special characters
 
         Returns:
-            None
+            The linked artifact if linking was successful, otherwise None.
         """
         name_parts = registered_model_name.split("/")
         if len(name_parts) != 1:
@@ -3778,7 +3804,9 @@ class Run:
             artifact = self._log_artifact(
                 artifact_or_path=path, name=name, type="model"
             )
-        self.link_artifact(artifact=artifact, target_path=target_path, aliases=aliases)
+        return self.link_artifact(
+            artifact=artifact, target_path=target_path, aliases=aliases
+        )
 
     @_log_to_run
     @_raise_if_finished

wandb/sdk/wandb_settings.py

@@ -465,11 +465,7 @@ class Settings(BaseModel, validate_assignment=True):
     save_code: Optional[bool] = None
     """Whether to save the code associated with the run."""
 
-    settings_system: str = Field(
-        default_factory=lambda: _path_convert(
-            os.path.join("~", ".config", "wandb", "settings")
-        )
-    )
+    settings_system: Optional[str] = None
     """Path to the system-wide settings file."""
 
     show_colors: Optional[bool] = None
@@ -1065,9 +1061,12 @@ class Settings(BaseModel, validate_assignment=True):
     @field_validator("settings_system", mode="after")
     @classmethod
     def validate_settings_system(cls, value):
-        if isinstance(value, pathlib.Path):
+        if value is None:
+            return None
+        elif isinstance(value, pathlib.Path):
             return str(_path_convert(value))
-        return _path_convert(value)
+        else:
+            return _path_convert(value)
 
     @field_validator("x_service_wait", mode="after")
     @classmethod

wandb/sdk/wandb_setup.py

@@ -15,12 +15,14 @@ from __future__ import annotations
 
 import logging
 import os
+import pathlib
 import sys
 import threading
 from typing import TYPE_CHECKING, Any, Union
 
 import wandb
 import wandb.integration.sagemaker as sagemaker
+from wandb.env import CONFIG_DIR
 from wandb.sdk.lib import import_hooks, wb_logging
 
 from . import wandb_settings
@@ -161,6 +163,16 @@ class _WandbSetup:
         self._logger.info(f"Configure stats pid to {pid}")
         s.x_stats_pid = pid
 
+        if settings and settings.settings_system:
+            s.settings_system = settings.settings_system
+        elif config_dir_str := os.getenv(CONFIG_DIR, None):
+            config_dir = pathlib.Path(config_dir_str).expanduser()
+            s.settings_system = str(config_dir / "settings")
+        else:
+            s.settings_system = str(
+                pathlib.Path("~", ".config", "wandb", "settings").expanduser()
+            )
+
         # load settings from the system config
         if s.settings_system:
             self._logger.info(f"Loading settings from {s.settings_system}")

wandb/util.py

@@ -1447,9 +1447,13 @@ def are_paths_on_same_drive(path1: str, path2: str) -> bool:
     if platform.system() != "Windows":
         return True
 
-    path1_drive = pathlib.Path(path1).resolve().drive
-    path2_drive = pathlib.Path(path2).resolve().drive
-    return path1_drive == path2_drive
+    try:
+        path1_drive = pathlib.Path(path1).resolve().drive
+        path2_drive = pathlib.Path(path2).resolve().drive
+        return path1_drive == path2_drive
+    # If either path is not a valid windows path, an OSError is raised.
+    except OSError:
+        return False
 
 
 # TODO(hugh): Deprecate version here and use wandb/sdk/lib/paths.py

{wandb-0.19.10.dist-info → wandb-0.19.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wandb
-Version: 0.19.10
+Version: 0.19.11
 Summary: A CLI and library for interacting with the Weights & Biases API.
 Project-URL: Source, https://github.com/wandb/wandb
 Project-URL: Bug Reports, https://github.com/wandb/wandb/issues