mblt-npu-python 0.0.1__tar.gz → 0.0.2__tar.gz

{mblt_npu_python-0.0.1 → mblt_npu_python-0.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mblt-npu-python
-Version: 0.0.1
+Version: 0.0.2
 Summary: Shared NPU access for the Mobilint Python packages
 Author: Mobilint
 License: BSD-3-Clause

{mblt_npu_python-0.0.1 → mblt_npu_python-0.0.2}/mblt_npu/__init__.py

@@ -22,7 +22,7 @@ from .npu_backend import (
 from .logging import log_model_details
 from .onnx_backend import ONNXBackend
 
-__version__ = "0.0.1"
+__version__ = "0.0.2"
 
 __all__ = [
     "BACKEND_CLASSES",

{mblt_npu_python-0.0.1 → mblt_npu_python-0.0.2}/mblt_npu/npu_backend.py

@@ -31,7 +31,7 @@ import math
 import os
 import re
 import sys
-from typing import Any, Dict, List, Literal, Optional, Union
+from typing import Any, Dict, List, Literal, Optional, Sequence, Union
 
 from huggingface_hub import HfApi, hf_hub_download
 from huggingface_hub.errors import EntryNotFoundError
@@ -211,7 +211,7 @@ class MobilintNPUBackend:
         dev_no: Optional[Union[int, List[int]]] = None,
         core_mode: CoreMode = "single",
         target_cores: Optional[List[Union[str, "CoreId"]]] = None,
-        target_clusters: Optional[List[Union[int, str, "Cluster"]]] = None,
+        target_clusters: Optional[Sequence[Union[int, str, "Cluster"]]] = None,
         revision: Optional[str] = None,
         commit_hash: Optional[str] = None,
         target_device: Optional[str] = None,
@@ -1350,7 +1350,7 @@ class MobilintNPUBackend:
         return result
 
     @target_clusters.setter
-    def target_clusters(self, values: List[Union[int, str, "Cluster"]]) -> None:
+    def target_clusters(self, values: Sequence[Union[int, str, "Cluster"]]) -> None:
         """Record a raw ``target_clusters`` override on the pending accumulator.
 
         Normalization (legacy migration, grain fold/unfold, device-set

{mblt_npu_python-0.0.1 → mblt_npu_python-0.0.2}/mblt_npu/npu_target.py

@@ -52,11 +52,32 @@ from .core_mode import CoreMode, normalize_core_mode
 # Default device index for ``dev_no`` when the caller does not pin one.
 _DEFAULT_DEV_NO: int = 0
 
+
 # Sentinel used to distinguish "field not overridden this session" from
 # "field explicitly set to None/empty". Kept module-private and reused by
 # both :class:`NPUTargetSpec` (single-call ``_with`` shim) and
 # :class:`NPUTargetSpecPending` (accumulator).
-_UNSET: Any = object()
+#
+# Identity, not equality, is what every ``raw_* is not _UNSET`` check relies
+# on. A bare ``object()`` does not survive ``copy.deepcopy`` as the same
+# instance, and upstream ``transformers.PretrainedConfig.to_dict()`` runs
+# ``copy.deepcopy(self.__dict__)`` on every config it serializes -- including
+# nested sub-configs whose ``npu_backend`` still holds a pending spec with
+# unresolved ``_UNSET`` slots. That deep-copied sentinel then fails every
+# ``is not _UNSET`` check, so callers see a bogus "overridden" field. Make the
+# sentinel copy/deepcopy to itself so its identity survives that path.
+class _UnsetType:
+    def __repr__(self) -> str:
+        return "_UNSET"
+
+    def __copy__(self) -> "_UnsetType":
+        return self
+
+    def __deepcopy__(self, memo: Dict[int, Any]) -> "_UnsetType":
+        return self
+
+
+_UNSET: Any = _UnsetType()
 
 
 cluster_map: Dict[int, "Cluster"] = {

{mblt_npu_python-0.0.1 → mblt_npu_python-0.0.2}/mblt_npu_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mblt-npu-python
-Version: 0.0.1
+Version: 0.0.2
 Summary: Shared NPU access for the Mobilint Python packages
 Author: Mobilint
 License: BSD-3-Clause

{mblt_npu_python-0.0.1 → mblt_npu_python-0.0.2}/tests/test_target_device.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import copy
 from pathlib import Path
 from typing import Any, cast
 
@@ -9,6 +10,7 @@ import pytest
 
 from mblt_npu import MobilintAriesBackend, MobilintNPUBackend, MobilintRegulusBackend
 import mblt_npu.npu_backend as npu_backend
+from mblt_npu.npu_target import _UNSET
 
 
 @pytest.mark.parametrize(
@@ -31,6 +33,40 @@ def test_target_device_selects_the_product_backend(
     assert backend.to_dict()["target_device"] == target_device
 
 
+def test_unset_sentinel_survives_deepcopy() -> None:
+    """Keep the ``_UNSET`` sentinel's identity across ``copy.deepcopy``.
+
+    ``transformers.PretrainedConfig.to_dict()`` runs ``copy.deepcopy(self.__dict__)``
+    on every config it serializes, including nested sub-configs whose
+    ``npu_backend`` still holds unresolved ``_UNSET`` slots. A bare ``object()``
+    sentinel would be reconstructed as a distinct instance there, so every
+    ``is not _UNSET`` override check on the copy would wrongly report the
+    field as overridden.
+    """
+
+    assert copy.deepcopy(_UNSET) is _UNSET
+    assert copy.copy(_UNSET) is _UNSET
+
+
+def test_backend_survives_deepcopy_before_finalization() -> None:
+    """Finalize identically after a deepcopy taken before the next ``_spec`` read.
+
+    Reproduces the HF ``PretrainedConfig.to_dict()`` path: overriding one field
+    (``target_clusters``) invalidates ``_finalized`` without touching the other
+    fields' still-``_UNSET`` raw slots. Deep-copying the backend in that state
+    (as upstream does for nested sub-configs, before their own ``to_dict()``
+    ever ran) must not corrupt those slots when the copy is finalized.
+    """
+
+    backend = MobilintAriesBackend()
+    backend.target_clusters = [0, 1]
+    assert backend._finalized is None
+
+    copied = copy.deepcopy(backend)
+
+    assert copied.to_dict() == backend.to_dict()
+
+
 def test_default_target_device_is_aries_rb() -> None:
     """Use Aries RB when callers do not declare a board."""