wisent 0.5.2__py3-none-any.whl → 0.5.4__py3-none-any.whl

wisent/__init__.py

@@ -1 +1 @@
-__version__ = "0.5.2"
+__version__ = "0.5.4"

wisent/core/__init__.py

@@ -1,24 +1,9 @@
-from wisent.core.activations import Activations
-from wisent.core.classifier.classifier import ActivationClassifier, Classifier
-
 from .utils.device import empty_device_cache, preferred_dtype, resolve_default_device, resolve_device, resolve_torch_device
 from .steering import SteeringMethod, SteeringType
 
 __all__ = [
-    "ActivationClassifier",
-    "ActivationHooks",
-    "Activations",
-    "Classifier",
-    "ContrastivePairSet",
-    "Layer",
-    "Model",
-    "ModelParameterOptimizer",
-    "PromptFormat",
-    "SecureCodeEvaluator",
     "SteeringMethod",
     "SteeringType",
-    "TokenScore",
-    "enforce_secure_execution",
     "empty_device_cache",
     "preferred_dtype",
     "resolve_default_device",

wisent/core/utils/__init__.py

@@ -0,0 +1,21 @@
+from .device import (
+    DeviceKind,
+    empty_device_cache,
+    ensure_tensor_on_device,
+    move_module_to_preferred_device,
+    preferred_dtype,
+    resolve_default_device,
+    resolve_device,
+    resolve_torch_device,
+)
+
+__all__ = [
+    "DeviceKind",
+    "empty_device_cache",
+    "ensure_tensor_on_device",
+    "move_module_to_preferred_device",
+    "preferred_dtype",
+    "resolve_default_device",
+    "resolve_device",
+    "resolve_torch_device",
+]

wisent/core/utils/device.py

@@ -0,0 +1,56 @@
+"""Centralized torch device selection helpers."""
+
+from __future__ import annotations
+
+from functools import lru_cache
+from typing import Literal
+
+import torch
+
+DeviceKind = Literal["cuda", "mps", "cpu"]
+
+
+def _mps_available() -> bool:
+    return getattr(torch.backends, "mps", None) is not None and torch.backends.mps.is_available()
+
+
+@lru_cache(maxsize=1)
+def resolve_default_device() -> DeviceKind:
+    if torch.cuda.is_available():
+        return "cuda"
+    if _mps_available():
+        return "mps"
+    return "cpu"
+
+
+def resolve_torch_device() -> torch.device:
+    return torch.device(resolve_default_device())
+
+
+def resolve_device(kind: DeviceKind | None = None) -> torch.device:
+    return torch.device(kind or resolve_default_device())
+
+
+def preferred_dtype(kind: DeviceKind | None = None) -> torch.dtype:
+    chosen = kind or resolve_default_device()
+    return torch.float16 if chosen in {"cuda", "mps"} else torch.float32
+
+
+def empty_device_cache(kind: DeviceKind | None = None) -> None:
+    chosen = kind or resolve_default_device()
+    if chosen == "cuda" and torch.cuda.is_available():
+        torch.cuda.empty_cache()
+    elif chosen == "mps" and _mps_available():
+        try:
+            torch.mps.empty_cache()  # type: ignore[attr-defined]
+        except AttributeError:
+            pass
+
+
+def move_module_to_preferred_device(module: torch.nn.Module) -> torch.nn.Module:
+    return module.to(resolve_torch_device())
+
+
+def ensure_tensor_on_device(tensor: torch.Tensor) -> torch.Tensor:
+    target = resolve_torch_device()
+    return tensor.to(target) if tensor.device != target else tensor

{wisent-0.5.2.dist-info → wisent-0.5.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wisent
-Version: 0.5.2
+Version: 0.5.4
 Summary: Monitor and guard against harmful content in language models
 Home-page: https://github.com/yourusername/wisent-activation-guardrails
 Author: Wisent Team

{wisent-0.5.2.dist-info → wisent-0.5.4.dist-info}/RECORD

@@ -1,4 +1,4 @@
-wisent/__init__.py,sha256=isJrmDBLRag7Zc2UK9ZovWGOv7ji1Oh-zJtJMNJFkXw,22
+wisent/__init__.py,sha256=DITpct-LrdIsTgwx2NgH5Ghx5y8Xgz1YMimy1ZV5RTY,22
 wisent/benchmarks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wisent/benchmarks/coding/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wisent/benchmarks/coding/metrics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -52,7 +52,7 @@ wisent/cli/wisent_cli/commands/train_cmd.py,sha256=Pp_DY6P-zk8PvCuz6ds9JxH7BWpqQ
 wisent/cli/wisent_cli/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wisent/cli/wisent_cli/util/aggregations.py,sha256=RW2f-B18k4R0gFlIFACQmrhFKSwfxREUqqu5PaI951Y,1415
 wisent/cli/wisent_cli/util/parsing.py,sha256=DvBTcoBItDGJQI-AE4bs0otBJ7ElLynNhqRDzgeQjzs,4067
-wisent/core/__init__.py,sha256=V48vT6tZj0iwp9rPWuElGJ5pyUyL1foWbQLS3tf4RiE,757
+wisent/core/__init__.py,sha256=n2ytkljR_PHZDqXl6Q4O2kwoNbtCiP5l0JBHfkXkE2s,361
 wisent/core/autonomous_agent.py,sha256=2k1PLWm2DZ6C2fmsjMDKmQ1_wfN7KhpilHrjkTw8nMw,52489
 wisent/core/bigcode_integration.py,sha256=TIaPQDbPRDPdnCq8U-Gwl4lgayPfhOabOVQddqxotY4,19927
 wisent/core/detection_handling.py,sha256=iiuKpzAbJfx_KFn2SFABQHOeeWblDJMXjzGwGDeKqcs,11127
@@ -180,6 +180,8 @@ wisent/core/trainers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 wisent/core/trainers/steering_trainer.py,sha256=JAdYwPhiakqN70WLqpZFZBsGHu2yed5v-TSM7ppjKJc,10557
 wisent/core/trainers/core/__init__.py,sha256=D0JX0-XCHdtLrCXhVDHNQafvyWCvJ4-o4UKtkH1lI1k,1257
 wisent/core/trainers/core/atoms.py,sha256=ycWk0G-7EIAEOnQL-o5_V5B8KTQ7CQUilGF4ibjighM,1536
+wisent/core/utils/__init__.py,sha256=NavuBkpDSz6q1dN6m34-I-l8Aps1Sgcnx82FHxFufzY,457
+wisent/core/utils/device.py,sha256=5brw9tclTU77NNAokzLXYrKUnjjYBzFfA7wNJPM9ytM,1609
 wisent/opti/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wisent/opti/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wisent/opti/core/atoms.py,sha256=9UZeb_SOdDxQ6FBhdAf9qXaEXElImKUsoAMsV0c4yZg,5266
@@ -211,8 +213,8 @@ wisent/synthetic/generators/diversities/core/__init__.py,sha256=47DEQpj8HBSa-_TI
 wisent/synthetic/generators/diversities/core/core.py,sha256=TjSj5T7NE5kRH-ABcFqb1Hz_j3Z6F_TcV-95uHD5Xw8,2201
 wisent/synthetic/generators/diversities/methods/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wisent/synthetic/generators/diversities/methods/fast_diversity.py,sha256=Z2UzTbzyJFM_ToxCoXM_LQQQ1Jc6BZknrbpikTG1MRw,8522
-wisent-0.5.2.dist-info/licenses/LICENSE,sha256=wy0iaw8b2tyqZAfKHib3lP3PJ9o88FDCg92oUHh3sDQ,1073
-wisent-0.5.2.dist-info/METADATA,sha256=IRtye3gZz9T6U3O0fFUap_xvO1rcnnE52hWlIKvPq4g,2424
-wisent-0.5.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-wisent-0.5.2.dist-info/top_level.txt,sha256=2Ts9Iyldnb3auIN2HBBaHPknRy7nSRDm2f6RGzYgr8A,7
-wisent-0.5.2.dist-info/RECORD,,
+wisent-0.5.4.dist-info/licenses/LICENSE,sha256=wy0iaw8b2tyqZAfKHib3lP3PJ9o88FDCg92oUHh3sDQ,1073
+wisent-0.5.4.dist-info/METADATA,sha256=WO1PZ7JzoAyWKxfGNY66bIqyV6lpNk25teYSOhm49hE,2424
+wisent-0.5.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+wisent-0.5.4.dist-info/top_level.txt,sha256=2Ts9Iyldnb3auIN2HBBaHPknRy7nSRDm2f6RGzYgr8A,7
+wisent-0.5.4.dist-info/RECORD,,