wisent 0.5.4__tar.gz → 0.5.6__tar.gz

{wisent-0.5.4 → wisent-0.5.6}/PKG-INFO

@@ -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.6/wisent/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.5.6"

{wisent-0.5.4 → wisent-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-0.5.4 → wisent-0.5.6}/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-0.5.4 → wisent-0.5.6}/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 → wisent-0.5.6}/wisent.egg-info/PKG-INFO

@@ -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/wisent/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.5.4"