wisent 0.5.3__py3-none-any.whl → 0.5.5__py3-none-any.whl

wisent/__init__.py

@@ -1 +1 @@
-__version__ = "0.5.3"
+__version__ = "0.5.5"

wisent/core/__init__.py

@@ -1,9 +1,10 @@
 from .utils.device import empty_device_cache, preferred_dtype, resolve_default_device, resolve_device, resolve_torch_device
-from .steering import SteeringMethod, SteeringType
+# Note: SteeringMethod and SteeringType import temporarily disabled due to missing dependencies
+# from .steering import SteeringMethod, SteeringType
 
 __all__ = [
-    "SteeringMethod",
-    "SteeringType",
+    # "SteeringMethod",
+    # "SteeringType",
     "empty_device_cache",
     "preferred_dtype",
     "resolve_default_device",

wisent/core/steering.py

@@ -7,8 +7,9 @@ from typing import Any, Dict, List, Optional, Union
 import torch
 import torch.nn.functional as F
 
-from wisent.core.activations import Activations
-from wisent.core.classifier.classifier import Classifier
+# Note: Activations and Classifier imports removed - these classes don't exist in the current package structure
+# from wisent.core.activations import Activations
+# from wisent.core.classifier.classifier import Classifier
 
 from .contrastive_pairs import ContrastivePairSet
 from .steering_method import CAA

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.3.dist-info → wisent-0.5.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wisent
-Version: 0.5.3
+Version: 0.5.5
 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.3.dist-info → wisent-0.5.5.dist-info}/RECORD

@@ -1,4 +1,4 @@
-wisent/__init__.py,sha256=tgzuqHKcEdKBaP57F5oXxq4XlW2n9J4Fj8ZGu7nGOZg,22
+wisent/__init__.py,sha256=78mfpLewKVki6c9UONSUdlVme_JsN9ZwIfp4Hf4jmG0,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=n2ytkljR_PHZDqXl6Q4O2kwoNbtCiP5l0JBHfkXkE2s,361
+wisent/core/__init__.py,sha256=yNBOdCXpkwMuo1H1sNAYgPf5_-Hhf7y4H-RZJPCiOpI,463
 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
@@ -70,7 +70,7 @@ wisent/core/representation.py,sha256=hBl_N9qbr5Gsa7GCQ0nMWRm82RqYEfhd9cyf0PPH5LY
 wisent/core/sample_size_optimizer.py,sha256=6wegGXZpdGpiR4R0YJ1D2JqLr6yinMndEx2gB5FL80s,23666
 wisent/core/sample_size_optimizer_v2.py,sha256=bVYJRZC4_Mrq-HFlYLyv-9tWvqEHJ3kCeIwlmYOwI6I,13286
 wisent/core/save_results.py,sha256=PRwaA5qO6EOsvURvLBl3YhvanlC0D0G4iYqxYAQ7sw8,13737
-wisent/core/steering.py,sha256=GJsARTiaB72fyGQ8UGmTPBmIRb_0OL_-2BXQANSnKqg,22535
+wisent/core/steering.py,sha256=I1mSLOQ6MMdUcJ_MmfmaCUH0IKxeCYlcsMCO6x6wLg4,22651
 wisent/core/steering_method.py,sha256=-hZqtvwRS7sGqQJUd36MoPm0rjbO1LrtPAYmcIk8BqQ,462
 wisent/core/steering_optimizer.py,sha256=wxa4p4aMjJWOknt2Jph28xPgYoEMxZVmp1GFA1pM3Wk,54759
 wisent/core/task_interface.py,sha256=OlWdcxkprmZcOto-bXmg75kzUcWzH_kyW_e7w2FdPLM,4471
@@ -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.3.dist-info/licenses/LICENSE,sha256=wy0iaw8b2tyqZAfKHib3lP3PJ9o88FDCg92oUHh3sDQ,1073
-wisent-0.5.3.dist-info/METADATA,sha256=Qomo-TEm-i_-gjiAzbGz_Vr9WpeAKjU9tO8rrbeiWbQ,2424
-wisent-0.5.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-wisent-0.5.3.dist-info/top_level.txt,sha256=2Ts9Iyldnb3auIN2HBBaHPknRy7nSRDm2f6RGzYgr8A,7
-wisent-0.5.3.dist-info/RECORD,,
+wisent-0.5.5.dist-info/licenses/LICENSE,sha256=wy0iaw8b2tyqZAfKHib3lP3PJ9o88FDCg92oUHh3sDQ,1073
+wisent-0.5.5.dist-info/METADATA,sha256=EkqfXuE8L_nXAiDFvzCwR7dP21ebVgfHW0TQOMcirwc,2424
+wisent-0.5.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+wisent-0.5.5.dist-info/top_level.txt,sha256=2Ts9Iyldnb3auIN2HBBaHPknRy7nSRDm2f6RGzYgr8A,7
+wisent-0.5.5.dist-info/RECORD,,