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

wisent/__init__.py

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

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/multi_steering.py

@@ -5,13 +5,49 @@ import torch
 from typing import List, Tuple, Optional, Dict, Any
 from pathlib import Path
 
-from .layer import Layer
-from .model import Model
-from .steering_methods.caa import CAA
-from .steering_methods.dac import DAC
+# Note: layer and model modules don't exist, using inline stubs
+# from .layer import Layer
+# from .model import Model
+from .steering_methods.methods.caa import CAAMethod as CAA
+# DAC doesn't exist yet
+# from .steering_methods.dac import DAC
 from .utils.device import resolve_default_device
 
 
+# Stub classes for missing modules
+class Layer:
+    """Stub Layer class - module doesn't exist in wisent package."""
+    def __init__(self, index: int):
+        self.index = index
+
+
+class Model:
+    """Stub Model class - module doesn't exist in wisent package."""
+    def __init__(self, model_name: str, device: str = None):
+        self.model_name = model_name
+        self.device = device
+        # This is a stub - actual implementation would load the model
+        raise NotImplementedError("Model class not implemented in wisent package")
+
+
+class DAC:
+    """Stub DAC class - module doesn't exist in wisent package."""
+    def __init__(self, device: str = None):
+        self.device = device
+        self.steering_vector = None
+        self.layer_index = None
+        self.is_trained = False
+
+    @staticmethod
+    def combine_steering_vectors(vectors, weights, normalize_weights=True):
+        """Stub method for combining steering vectors."""
+        raise NotImplementedError("DAC steering method not implemented in wisent package")
+
+    def apply_steering(self, hidden_states, strength=1.0):
+        """Stub method for applying steering."""
+        raise NotImplementedError("DAC steering method not implemented in wisent package")
+
+
 class MultiSteeringError(Exception):
     """Exception raised for multi-steering errors."""
     pass

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

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

@@ -1,4 +1,4 @@
-wisent/__init__.py,sha256=DITpct-LrdIsTgwx2NgH5Ghx5y8Xgz1YMimy1ZV5RTY,22
+wisent/__init__.py,sha256=CMH34Gt1AqO7z_TqRj94XwohGoVCf8aes0djkqm45mk,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
@@ -64,13 +64,13 @@ wisent/core/managed_cached_benchmarks.py,sha256=JbvpZ1fgSuQQhyQVKEvqrQZRHGqfnjo9
 wisent/core/mixed_benchmark_sampler.py,sha256=tKQCHUXVuYeCyx4VZt8O1hGyB-TOY_SQ_SYi8cyApII,13585
 wisent/core/model_config_manager.py,sha256=rQAdSmk3GFlZXyHp3fSV1bORxiZWhmzIz1uo3H4JtkA,12009
 wisent/core/model_persistence.py,sha256=6_vc1Ndujd4v0O68giINSTvYhmb7-AiacWwAbqLOrls,10636
-wisent/core/multi_steering.py,sha256=IpaVrs28tZQM8jXbgUbrhq3bWbH4KaNhUZQcWit7_p0,11711
+wisent/core/multi_steering.py,sha256=PAn8eKFObR6QT-Mt1RbBB05AMr6RebGIBe3j5snSRHA,13069
 wisent/core/parser.py,sha256=_YDeSuQMx0zNknz9rX3Ls1YPT1x5eohoY8rfjeoqxV8,69091
 wisent/core/representation.py,sha256=hBl_N9qbr5Gsa7GCQ0nMWRm82RqYEfhd9cyf0PPH5LY,195
 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
@@ -213,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.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,,
+wisent-0.5.6.dist-info/licenses/LICENSE,sha256=wy0iaw8b2tyqZAfKHib3lP3PJ9o88FDCg92oUHh3sDQ,1073
+wisent-0.5.6.dist-info/METADATA,sha256=sR7d86PIFGErlOB8gKyjxB3iwlKX3ea__KEUsBCtKfg,2424
+wisent-0.5.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+wisent-0.5.6.dist-info/top_level.txt,sha256=2Ts9Iyldnb3auIN2HBBaHPknRy7nSRDm2f6RGzYgr8A,7
+wisent-0.5.6.dist-info/RECORD,,