wisent 0.5.1__tar.gz → 0.5.2__tar.gz

{wisent-0.5.1 → wisent-0.5.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wisent
-Version: 0.5.1
+Version: 0.5.2
 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/wisent/__init__.py

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

{wisent-0.5.1 → wisent-0.5.2}/wisent/benchmarks/coding/metrics/evaluator.py

@@ -2,19 +2,19 @@ from __future__ import annotations
 from dataclasses import dataclass
 from typing import Callable, Iterable, Optional, TYPE_CHECKING
 
-from wisent_guard.benchmarks.coding.safe_docker.core.runtime import DockerSandboxExecutor
-from wisent_guard.benchmarks.coding.safe_docker.recipes import RECIPE_REGISTRY
-from wisent_guard.benchmarks.coding.metrics.core.atoms import SampleOutcome, Evaluator
+from wisent.benchmarks.coding.safe_docker.core.runtime import DockerSandboxExecutor
+from wisent.benchmarks.coding.safe_docker.recipes import RECIPE_REGISTRY
+from wisent.benchmarks.coding.metrics.core.atoms import SampleOutcome, Evaluator
 
-from wisent_guard.benchmarks.coding.output_sanitizer.core.atoms import TaskSchema
-from wisent_guard.benchmarks.coding.output_sanitizer.python_sanitizer import PythonStandardizer
-from wisent_guard.benchmarks.coding.output_sanitizer.cpp_sanitizer import CppStandardizer
-from wisent_guard.benchmarks.coding.output_sanitizer.java_sanitizer import JavaStandardizer
+from wisent.benchmarks.coding.output_sanitizer.core.atoms import TaskSchema
+from wisent.benchmarks.coding.output_sanitizer.python_sanitizer import PythonStandardizer
+from wisent.benchmarks.coding.output_sanitizer.cpp_sanitizer import CppStandardizer
+from wisent.benchmarks.coding.output_sanitizer.java_sanitizer import JavaStandardizer
 
 if TYPE_CHECKING:
-    from wisent_guard.benchmarks.coding.safe_docker.core.atoms import Result
-    from wisent_guard.benchmarks.coding.providers.core.atoms import Provider, CodingTask
-    from wisent_guard.benchmarks.coding.output_sanitizer.core.atoms import CodeStandardizer
+    from wisent.benchmarks.coding.safe_docker.core.atoms import Result
+    from wisent.benchmarks.coding.providers.core.atoms import Provider, CodingTask
+    from wisent.benchmarks.coding.output_sanitizer.core.atoms import CodeStandardizer
 
 RepairFn = Callable[[str, dict[str,str], str], dict[str,str]]  
 
@@ -82,7 +82,7 @@ def _make_schema(task: CodingTask) -> TaskSchema:
         and allow_wrapper set appropriately.
         
     example:
-        >>> from wisent_guard.benchmarks.coding.providers.core.atoms import CodingTask
+        >>> from wisent.benchmarks.coding.providers.core.atoms import CodingTask
         >>> task = CodingTask(language="python", files={}, options={"entry_point":"add","file_name":"my_solution.py"})
         >>> schema = _make_schema(task)
         >>> schema.language
@@ -128,7 +128,7 @@ class CodingEvaluator(Evaluator):
             Feedback string summarizing the result, truncated to cfg.feedback_max_chars.
             
         examples:
-            >>> from wisent_guard.benchmarks.coding.safe_docker.core.atoms import Result
+            >>> from wisent.benchmarks.coding.safe_docker.core.atoms import Result
             >>> res = Result(status="timeout", stdout="", stderr="", elapsed=10.0)
             >>> evaluator = CodingEvaluator(provider=None, model_fn=lambda x: {}, cfg=EvaluatorConfig())
             >>> evaluator._feedback(res)
@@ -163,8 +163,8 @@ class CodingEvaluator(Evaluator):
             Result object containing the status, stdout, stderr, and elapsed time.
 
         examples:
-            >>> from wisent_guard.benchmarks.coding.providers.core.atoms import CodingTask
-            >>> from wisent_guard.benchmarks.coding.safe_docker.core.atoms import Result
+            >>> from wisent.benchmarks.coding.providers.core.atoms import CodingTask
+            >>> from wisent.benchmarks.coding.safe_docker.core.atoms import Result
             >>> task = CodingTask(language="python", files={}, options={})
             >>> files = {"solution.py": "def add(a,b): return a + b", "tests.py": "from solution import add\ndef test_ok(): assert add(1,2)==3"}
             >>> evaluator = CodingEvaluator(provider=None, model_fn=lambda x: {})
@@ -201,7 +201,7 @@ class CodingEvaluator(Evaluator):
             The sanitized files if pre_sanitize is True and a sanitizer exists for the language; otherwise, the original files.
         
         examples:
-            >>> from wisent_guard.benchmarks.coding.providers.core.atoms import CodingTask
+            >>> from wisent.benchmarks.coding.providers.core.atoms import CodingTask
             >>> task = CodingTask(language="python", files={}, options={"entry_point":"add","file_name":"my_solution.py"})
             >>> files = {"my_solution.py": "def add(a,b): return a - b  # BUG"}
             >>> evaluator = CodingEvaluator(provider=None, model_fn=lambda x: {}, cfg=EvaluatorConfig(pre_sanitize=True))
@@ -234,7 +234,7 @@ class CodingEvaluator(Evaluator):
             SampleOutcome for each task, indicating pass/fail status and elapsed time.
 
         examples:
-            >>> from wisent_guard.benchmarks.coding.providers.core.atoms import CodingTask, Provider
+            >>> from wisent.benchmarks.coding.providers.core.atoms import CodingTask, Provider
             >>> class DummyProvider:
             ...     name = "dummy"
             ...     def iter_tasks(self):

{wisent-0.5.1 → wisent-0.5.2}/wisent/benchmarks/coding/output_sanitizer/cpp_sanitizer.py

@@ -1,8 +1,8 @@
 from __future__ import annotations
 import re
 from typing import List
-from wisent_guard.benchmarks.coding.output_sanitizer.core.atoms import TaskSchema, NormalizeResult, CodeStandardizer
-from wisent_guard.benchmarks.coding.output_sanitizer.utils import extract_code_block, normalize_whitespace
+from wisent.benchmarks.coding.output_sanitizer.core.atoms import TaskSchema, NormalizeResult, CodeStandardizer
+from wisent.benchmarks.coding.output_sanitizer.utils import extract_code_block, normalize_whitespace
 
 FUNC_RE = re.compile(r"^\s*(?:template<[^>]+>\s*)?(?:[\w:\s*&<>,]+)\s+(\w+)\s*\(", re.MULTILINE)
 CLASS_RE = re.compile(r"^\s*class\s+(\w+)\s*[{:]", re.MULTILINE)

{wisent-0.5.1 → wisent-0.5.2}/wisent/benchmarks/coding/output_sanitizer/java_sanitizer.py

@@ -2,8 +2,8 @@
 from __future__ import annotations
 import re
 from typing import List
-from wisent_guard.benchmarks.coding.output_sanitizer.core.atoms import TaskSchema, NormalizeResult, CodeStandardizer
-from wisent_guard.benchmarks.coding.output_sanitizer.utils import extract_code_block, normalize_whitespace
+from wisent.benchmarks.coding.output_sanitizer.core.atoms import TaskSchema, NormalizeResult, CodeStandardizer
+from wisent.benchmarks.coding.output_sanitizer.utils import extract_code_block, normalize_whitespace
 
 CLASS_RE = re.compile(r"\bclass\s+([A-Za-z_]\w*)")
 METHOD_RE = re.compile(r"(public\s+static\s+[\w\<\>\[\]]+\s+)(\w+)\s*\(")

{wisent-0.5.1 → wisent-0.5.2}/wisent/benchmarks/coding/output_sanitizer/python_sanitizer.py

@@ -2,8 +2,8 @@
 from __future__ import annotations
 import ast, re
 from typing import List
-from wisent_guard.benchmarks.coding.output_sanitizer.core.atoms import TaskSchema, NormalizeResult, CodeStandardizer
-from wisent_guard.benchmarks.coding.output_sanitizer.utils import extract_code_block, normalize_whitespace, maybe_black
+from wisent.benchmarks.coding.output_sanitizer.core.atoms import TaskSchema, NormalizeResult, CodeStandardizer
+from wisent.benchmarks.coding.output_sanitizer.utils import extract_code_block, normalize_whitespace, maybe_black
 
 class PythonStandardizer(CodeStandardizer):
     language = "python"

{wisent-0.5.1 → wisent-0.5.2}/wisent/benchmarks/coding/safe_docker/core/runtime.py

@@ -1,10 +1,10 @@
 from __future__ import annotations
 import json, os, subprocess, tempfile
 from typing import TYPE_CHECKING
-from wisent_guard.benchmarks.coding.safe_docker.core.atoms import Result, SandboxExecutor
+from wisent.benchmarks.coding.safe_docker.core.atoms import Result, SandboxExecutor
 
 if TYPE_CHECKING:
-    from wisent_guard.benchmarks.coding.safe_docker.core.atoms import Job
+    from wisent.benchmarks.coding.safe_docker.core.atoms import Job
 
 __all__ = ["DockerSandboxExecutor"]
 
@@ -49,8 +49,8 @@ class DockerSandboxExecutor(SandboxExecutor):
             A Result object with the outcome of the execution.
 
         example (pythonm add function)
-        >>> from wisent_guard.benchmarks.coding.safe_docker.core.atoms import Job, Result
-        >>> from wisent_guard.benchmarks.coding.safe_docker.core.runtime import DockerSandboxExecutor
+        >>> from wisent.benchmarks.coding.safe_docker.core.atoms import Job, Result
+        >>> from wisent.benchmarks.coding.safe_docker.core.runtime import DockerSandboxExecutor
         >>> job = Job(
         ...     language="python",
         ...     compile_argv=None,

{wisent-0.5.1 → wisent-0.5.2}/wisent/benchmarks/coding/safe_docker/entrypoint.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 import json, os, shutil, subprocess, sys, time, signal, resource
 
-from wisent_guard.benchmarks.coding.safe_docker.core.atoms import Job
+from wisent.benchmarks.coding.safe_docker.core.atoms import Job
 
 JOB_FILE = "/job/job.json"
 WORKDIR = "/work"

{wisent-0.5.1 → wisent-0.5.2}/wisent/benchmarks/coding/safe_docker/recipes.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 from typing import Dict
-from wisent_guard.benchmarks.coding.safe_docker.core.atoms import Job, LanguageRecipe
+from wisent.benchmarks.coding.safe_docker.core.atoms import Job, LanguageRecipe
 
 class PythonRecipe(LanguageRecipe):
     """

{wisent-0.5.1 → wisent-0.5.2}/wisent/classifiers/models/logistic.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import torch
 from torch import nn
 
-from wisent_guard.classifiers.core.atoms import BaseClassifier
+from wisent.classifiers.core.atoms import BaseClassifier
 
 __all__ = ["LogisticClassifier"]
 

{wisent-0.5.1 → wisent-0.5.2}/wisent/classifiers/models/mlp.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import torch
 from torch import nn
 
-from wisent_guard.classifiers.core.atoms import BaseClassifier
+from wisent.classifiers.core.atoms import BaseClassifier
 
 __all__ = ["MLPClassifier"]
 

{wisent-0.5.1 → wisent-0.5.2}/wisent/cli/classifiers/classifier_rotator.py

@@ -7,7 +7,7 @@ import pkgutil
 from pathlib import Path
 from typing import Any
 
-from wisent_guard.core.classifiers.core.atoms import BaseClassifier, ClassifierError, ClassifierTrainReport
+from wisent.core.classifiers.core.atoms import BaseClassifier, ClassifierError, ClassifierTrainReport
 
 __all__ = ["ClassifierRotator"]
 

{wisent-0.5.1 → wisent-0.5.2}/wisent/cli/data_loaders/data_loader_rotator.py

@@ -7,7 +7,7 @@ import pkgutil
 from pathlib import Path
 from typing import Any, Dict, List, Optional, Type, Union
 
-from wisent_guard.core.data_loaders.core.atoms import BaseDataLoader, DataLoaderError, LoadDataResult
+from wisent.core.data_loaders.core.atoms import BaseDataLoader, DataLoaderError, LoadDataResult
 
 class DataLoaderRotator:
     """Discover/select a data loader and use it to load data."""

{wisent-0.5.1 → wisent-0.5.2}/wisent/cli/evaluators/evaluator_rotator.py

@@ -8,7 +8,7 @@ from typing import Any, Dict, List, Optional, Sequence, Union, Type
 import inspect
 import logging
 
-from wisent_guard.core.evaluators.core.atoms import BaseEvaluator, EvalResult, EvaluatorError
+from wisent.core.evaluators.core.atoms import BaseEvaluator, EvalResult, EvaluatorError
 
 logger = logging.getLogger(__name__)
 

{wisent-0.5.1 → wisent-0.5.2}/wisent/cli/steering_methods/steering_rotator.py

@@ -8,10 +8,10 @@ import pkgutil
 from pathlib import Path
 from typing import Any, Type
 
-from wisent_guard.core.steering_methods.core.atoms import BaseSteeringError, BaseSteeringMethod
-from wisent_guard.core.contrastive_pairs.core.set import ContrastivePairSet
+from wisent.core.steering_methods.core.atoms import BaseSteeringError, BaseSteeringMethod
+from wisent.core.contrastive_pairs.core.set import ContrastivePairSet
 
-from wisent_guard.core.activations.core.atoms import LayerActivations
+from wisent.core.activations.core.atoms import LayerActivations
 
 __all__ = [
     "SteeringMethodRotator",

{wisent-0.5.1 → wisent-0.5.2}/wisent/cli/wisent_cli/commands/help_cmd.py

@@ -9,7 +9,7 @@ app = typer.Typer(help="Human-friendly help router")
 
 def _run_cli_line(line: str) -> None:
     from typer.main import get_command
-    from wisent_guard.cli.wisent_cli.main import app as root_app
+    from wisent.cli.wisent_cli.main import app as root_app
     click_cmd = get_command(root_app)
     args = shlex.split(line)
     try:

{wisent-0.5.1 → wisent-0.5.2}/wisent/cli/wisent_cli/commands/listing.py

@@ -4,8 +4,8 @@ from typing import Optional
 
 import typer
 
-from wisent_guard.cli.wisent_cli.ui import echo
-from wisent_guard.cli.wisent_cli.util import aggregations as aggs
+from wisent.cli.wisent_cli.ui import echo
+from wisent.cli.wisent_cli.util import aggregations as aggs
 
 try:
     from rich.table import Table
@@ -33,7 +33,7 @@ def list_aggregations():
 
 @app.command("list-methods")
 def list_methods():
-    from wisent_guard.cli.steering_methods.steering_rotator import SteeringMethodRotator  # type: ignore
+    from wisent.cli.steering_methods.steering_rotator import SteeringMethodRotator  # type: ignore
     rot = SteeringMethodRotator()
     methods = rot.list_methods()
     if not methods:
@@ -57,7 +57,7 @@ def list_loaders(
     loaders_location: Optional[str] = typer.Option(None, help="Package path or directory containing data loader modules"),
     scope_prefix: Optional[str] = typer.Option(None, help="Limit list to module path prefix"),
 ):
-    from wisent_guard.cli.data_loaders.data_loader_rotator import DataLoaderRotator  # type: ignore
+    from wisent.cli.data_loaders.data_loader_rotator import DataLoaderRotator  # type: ignore
     if loaders_location:
         DataLoaderRotator.discover_loaders(loaders_location)
     loaders = DataLoaderRotator.list_loaders(scope_prefix=scope_prefix)
@@ -84,8 +84,8 @@ def explain(
     loaders_location: Optional[str] = typer.Option(None, help="Where to discover data loaders"),
     aggregation: Optional[str] = typer.Option(None, help="Aggregation to describe"),
 ):
-    from wisent_guard.cli.data_loaders.data_loader_rotator import DataLoaderRotator  # type: ignore
-    from wisent_guard.cli.steering_methods.steering_rotator import SteeringMethodRotator  # type: ignore
+    from wisent.cli.data_loaders.data_loader_rotator import DataLoaderRotator  # type: ignore
+    from wisent.cli.steering_methods.steering_rotator import SteeringMethodRotator  # type: ignore
 
     if loaders_location:
         DataLoaderRotator.discover_loaders(loaders_location)
@@ -130,7 +130,7 @@ def loader_args(
     Useful so users know precisely what to pass (e.g., for `custom`:
     `path, split_ratio, seed, training_limit, testing_limit`).
     """
-    from wisent_guard.cli.data_loaders.data_loader_rotator import DataLoaderRotator  # type: ignore
+    from wisent.cli.data_loaders.data_loader_rotator import DataLoaderRotator  # type: ignore
     if loaders_location:
         DataLoaderRotator.discover_loaders(loaders_location)
     rot = DataLoaderRotator(loader=name, loaders_location=loaders_location or "wisent_guard.core.data_loaders.loaders")

{wisent-0.5.1 → wisent-0.5.2}/wisent/cli/wisent_cli/commands/train_cmd.py

@@ -6,9 +6,9 @@ from typing import Dict, List, Optional
 
 import typer
 
-from wisent_guard.cli.wisent_cli.ui import echo
-from wisent_guard.cli.wisent_cli.util import aggregations as aggs
-from wisent_guard.cli.wisent_cli.util.parsing import (
+from wisent.cli.wisent_cli.ui import echo
+from wisent.cli.wisent_cli.util import aggregations as aggs
+from wisent.cli.wisent_cli.util.parsing import (
     parse_natural_tokens, parse_kv, parse_layers, to_bool, DTYPE_MAP,
 )
 
@@ -25,7 +25,7 @@ __all__ = ["app", "train"]
 app = typer.Typer(help="Training workflow")
 
 def _resolve_method(method_name: Optional[str], methods_location: Optional[str]):
-    from wisent_guard.cli.steering_methods.steering_rotator import SteeringMethodRotator  # type: ignore
+    from wisent.cli.steering_methods.steering_rotator import SteeringMethodRotator  # type: ignore
     # Best effort discovery if available
     try:
         if methods_location and hasattr(SteeringMethodRotator, "discover_methods"):
@@ -96,11 +96,11 @@ def _show_plan(
 
     code = f"""
 # Example: Training steering vectors (auto-generated plan)
-from wisent_guard.core.trainers.steering_trainer import WisentSteeringTrainer
-from wisent_guard.core.models.wisent_model import WisentModel
-from wisent_guard.cli.data_loaders.data_loader_rotator import DataLoaderRotator
-from wisent_guard.cli.steering_methods.steering_rotator import SteeringMethodRotator
-from wisent_guard.core.activations.core.atoms import ActivationAggregationStrategy
+from wisent.core.trainers.steering_trainer import WisentSteeringTrainer
+from wisent.core.models.wisent_model import WisentModel
+from wisent.cli.data_loaders.data_loader_rotator import DataLoaderRotator
+from wisent.cli.steering_methods.steering_rotator import SteeringMethodRotator
+from wisent.core.activations.core.atoms import ActivationAggregationStrategy
 
 # 1) Model
 model = WisentModel(model_name={model!r}, layers={{}}, device={store_device!r})
@@ -152,9 +152,9 @@ def train(ctx: typer.Context, params: List[str] = typer.Argument(None)):
     See `wisent loader-args custom` to view the exact loader arguments.
     """
     # Lazy imports
-    from wisent_guard.cli.data_loaders.data_loader_rotator import DataLoaderRotator  # type: ignore
-    from wisent_guard.core.models.wisent_model import WisentModel  # type: ignore
-    from wisent_guard.core.trainers.steering_trainer import WisentSteeringTrainer  # type: ignore
+    from wisent.cli.data_loaders.data_loader_rotator import DataLoaderRotator  # type: ignore
+    from wisent.core.models.wisent_model import WisentModel  # type: ignore
+    from wisent.core.trainers.steering_trainer import WisentSteeringTrainer  # type: ignore
 
     tokens = list(params or []) + list(ctx.args or [])
     top, loader_kv_raw, method_kv_raw = parse_natural_tokens(tokens)
@@ -278,11 +278,11 @@ def train(ctx: typer.Context, params: List[str] = typer.Argument(None)):
 
     # -- Model -----------------------------------------------------------------
     typer.echo(f"[+] Loading model: {model}")
-    from wisent_guard.core.models.wisent_model import WisentModel  # type: ignore
+    from wisent.core.models.wisent_model import WisentModel  # type: ignore
     wmodel = WisentModel(model_name=model, layers={}, device=store_device)
 
     # -- Data loader -----------------------------------------------------------
-    from wisent_guard.cli.data_loaders.data_loader_rotator import DataLoaderRotator  # type: ignore
+    from wisent.cli.data_loaders.data_loader_rotator import DataLoaderRotator  # type: ignore
     if loaders_location:
         DataLoaderRotator.discover_loaders(loaders_location)
     dl_rot = DataLoaderRotator(loader=loader, loaders_location=loaders_location or "wisent_guard.core.data_loaders.loaders")
@@ -297,7 +297,7 @@ def train(ctx: typer.Context, params: List[str] = typer.Argument(None)):
     typer.echo(f"[+] Steering method: {name_shown}")
 
     # -- Trainer ---------------------------------------------------------------
-    from wisent_guard.core.trainers.steering_trainer import WisentSteeringTrainer  # type: ignore
+    from wisent.core.trainers.steering_trainer import WisentSteeringTrainer  # type: ignore
     torch_dtype = None if dtype is None else __import__("torch").__dict__[DTYPE_MAP[dtype]]
     trainer = WisentSteeringTrainer(
         model=wmodel,

{wisent-0.5.1 → wisent-0.5.2}/wisent/cli/wisent_cli/main.py

@@ -3,12 +3,12 @@ from typing import Optional
 
 import typer
 
-from wisent_guard.cli.wisent_cli.version import APP_NAME, APP_VERSION
-from wisent_guard.cli.wisent_cli.ui import print_banner
-from wisent_guard.cli.wisent_cli.commands.listing import app as listing_app
-from wisent_guard.cli.wisent_cli.commands.train_cmd import app as train_app
-from wisent_guard.cli.wisent_cli.commands.help_cmd import app as help_router_app
-from wisent_guard.cli.wisent_cli.shell import app as shell_app
+from wisent.cli.wisent_cli.version import APP_NAME, APP_VERSION
+from wisent.cli.wisent_cli.ui import print_banner
+from wisent.cli.wisent_cli.commands.listing import app as listing_app
+from wisent.cli.wisent_cli.commands.train_cmd import app as train_app
+from wisent.cli.wisent_cli.commands.help_cmd import app as help_router_app
+from wisent.cli.wisent_cli.shell import app as shell_app
 
 app = typer.Typer(
     no_args_is_help=True,
@@ -81,7 +81,7 @@ Tip: add [b]interactive true[/] for a guided wizard.
 """
     try:
         from rich.panel import Panel
-        from wisent_guard.cli.wisent_cli.ui import echo
+        from wisent.cli.wisent_cli.ui import echo
         echo(Panel.fit(msg, title="Instructions", border_style="green"))
     except Exception:
         print(msg)

{wisent-0.5.1 → wisent-0.5.2}/wisent/cli/wisent_cli/shell.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 import shlex
 import typer
 
-from wisent_guard.cli.wisent_cli.ui import print_banner, echo
+from wisent.cli.wisent_cli.ui import print_banner, echo
 
 try:
     from rich.panel import Panel
@@ -16,7 +16,7 @@ app = typer.Typer(help="Interactive shell")
 
 def _run_cli_line(line: str) -> None:
     from typer.main import get_command
-    from wisent_guard.cli.wisent_cli.main import app as root_app
+    from wisent.cli.wisent_cli.main import app as root_app
     click_cmd = get_command(root_app)
     args = shlex.split(line)
     try:

{wisent-0.5.1 → wisent-0.5.2}/wisent/cli/wisent_cli/util/aggregations.py

@@ -4,7 +4,7 @@ from typing import Dict, Optional
 __all__ = ["agg_cls", "descriptions", "name_map", "pick"]
 
 def agg_cls():
-    from wisent_guard.core.activations.core.atoms import ActivationAggregationStrategy
+    from wisent.core.activations.core.atoms import ActivationAggregationStrategy
     return ActivationAggregationStrategy
 
 def descriptions() -> Dict[object, str]:

{wisent-0.5.1 → wisent-0.5.2}/wisent/core/__init__.py

@@ -1,5 +1,5 @@
-from wisent_guard.core.activations import Activations
-from wisent_guard.core.classifier.classifier import ActivationClassifier, Classifier
+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

{wisent-0.5.1 → wisent-0.5.2}/wisent/core/activations/activations_collector.py

@@ -4,9 +4,9 @@ from typing import Sequence
 import torch
 
 
-from wisent_guard.core.contrastive_pairs.core.pair import ContrastivePair
-from wisent_guard.core.activations.core.atoms import LayerActivations, ActivationAggregationStrategy, LayerName, RawActivationMap
-from wisent_guard.core.models.wisent_model import WisentModel 
+from wisent.core.contrastive_pairs.core.pair import ContrastivePair
+from wisent.core.activations.core.atoms import LayerActivations, ActivationAggregationStrategy, LayerName, RawActivationMap
+from wisent.core.models.wisent_model import WisentModel 
 __all__ = ["ActivationCollector"]
 
 @dataclass(slots=True)
@@ -315,8 +315,8 @@ def _resp_text(resp_obj: object) -> str:
     return str(resp_obj)
 
 if __name__ == "__main__":
-    from wisent_guard.core.contrastive_pairs.core.pair import ContrastivePair
-    from wisent_guard.core.contrastive_pairs.core.response import PositiveResponse, NegativeResponse
+    from wisent.core.contrastive_pairs.core.pair import ContrastivePair
+    from wisent.core.contrastive_pairs.core.response import PositiveResponse, NegativeResponse
 
     model = WisentModel(model_name="/home/gg/.cache/huggingface/hub/models--meta-llama--Llama-3.2-1B-Instruct/snapshots/9213176726f574b556790deb65791e0c5aa438b6")
     collector = ActivationCollector(model=model, store_device="cpu")

{wisent-0.5.1 → wisent-0.5.2}/wisent/core/agent/device_benchmarks.py

@@ -19,7 +19,7 @@ import hashlib
 
 import torch
 
-from wisent_guard.core.utils.device import resolve_default_device
+from wisent.core.utils.device import resolve_default_device
 
 
 @dataclass
@@ -145,7 +145,7 @@ sys.path.append('.')
 
 start_time = time.time()
 try:
-    from wisent_guard.core.model import Model
+    from wisent.core.model import Model
     # Use the actual model that will be used in production
     model = Model("meta-llama/Llama-3.1-8B-Instruct")
     end_time = time.time()
@@ -194,7 +194,7 @@ print("BENCHMARK_DEBUG: Starting evaluation benchmark")
 start_time = time.time()
 try:
     print("BENCHMARK_DEBUG: Importing CLI...")
-    from wisent_guard.cli import run_task_pipeline
+    from wisent.cli import run_task_pipeline
     print("BENCHMARK_DEBUG: CLI imported successfully")
     
     print("BENCHMARK_DEBUG: Running task pipeline...")
@@ -286,9 +286,9 @@ from pathlib import Path
 from typing import Dict, Optional
 try:
     print("BENCHMARK_DEBUG: Importing required modules...")
-    from wisent_guard.core.model import Model
-    from wisent_guard.core.agent.diagnose.synthetic_classifier_option import create_classifier_from_trait_description
-    from wisent_guard.core.agent.budget import set_time_budget
+    from wisent.core.model import Model
+    from wisent.core.agent.diagnose.synthetic_classifier_option import create_classifier_from_trait_description
+    from wisent.core.agent.budget import set_time_budget
     import time
     print("BENCHMARK_DEBUG: All modules imported successfully")
     
@@ -389,7 +389,7 @@ sys.path.append('.')
 
 start_time = time.time()
 try:
-    from wisent_guard.cli import run_task_pipeline
+    from wisent.cli import run_task_pipeline
     
     # Run actual steering with real model and minimal examples
     run_task_pipeline(
@@ -454,8 +454,8 @@ sys.path.append('.')
 
 start_time = time.time()
 try:
-    from wisent_guard.core.model import Model
-    from wisent_guard.core.contrastive_pairs.generate_synthetically import SyntheticContrastivePairGenerator
+    from wisent.core.model import Model
+    from wisent.core.contrastive_pairs.generate_synthetically import SyntheticContrastivePairGenerator
     
     # Load the actual model
     model = Model("meta-llama/Llama-3.1-8B-Instruct")

{wisent-0.5.1 → wisent-0.5.2}/wisent/core/agent/diagnose/classifier_marketplace.py

@@ -7,7 +7,7 @@ import time
 from datetime import datetime
 import numpy as np
 
-from wisent_guard.core.utils.device import resolve_default_device
+from wisent.core.utils.device import resolve_default_device
 
 @dataclass
 class ClassifierListing:

{wisent-0.5.1 → wisent-0.5.2}/wisent/core/agent/diagnose/create_classifier.py

@@ -12,7 +12,7 @@ import time
 from dataclasses import dataclass
 from typing import Any, Dict, List, Optional, Tuple
 
-from wisent_guard.core.classifier.classifier import ActivationClassifier, Classifier
+from wisent.core.classifier.classifier import ActivationClassifier, Classifier
 
 from ...activations import Activations
 from ...layer import Layer

{wisent-0.5.1 → wisent-0.5.2}/wisent/core/agent/diagnose/response_diagnostics.py

@@ -11,8 +11,8 @@ This module handles:
 from dataclasses import dataclass
 from typing import Any, Dict, List
 
-from wisent_guard.core.activations import ActivationAggregationStrategy, Activations
-from wisent_guard.core.classifier.classifier import Classifier
+from wisent.core.activations import ActivationAggregationStrategy, Activations
+from wisent.core.classifier.classifier import Classifier
 
 from ...layer import Layer
 from ...model import Model

{wisent-0.5.1 → wisent-0.5.2}/wisent/core/agent/diagnose/synthetic_classifier_option.py

@@ -11,7 +11,7 @@ import time
 from dataclasses import dataclass
 from typing import List, Tuple
 
-from wisent_guard.core.classifier.classifier import ActivationClassifier
+from wisent.core.classifier.classifier import ActivationClassifier
 
 from ....core.agent.budget import ResourceType, calculate_max_tasks_for_time_budget, get_budget_manager
 from ....core.contrastive_pairs.generate_synthetically import SyntheticContrastivePairGenerator
@@ -136,7 +136,7 @@ class SyntheticClassifierFactory:
             logging.info(f"Extracting activations from {len(pair_set.pairs)} pairs...")
 
             # Create Layer object for activation extraction
-            from wisent_guard.core.layer import Layer
+            from wisent.core.layer import Layer
 
             layer_obj = Layer(index=15, type="transformer")
             logging.info(f"Created Layer object: index={layer_obj.index}, type={layer_obj.type}")
@@ -193,13 +193,13 @@ class SyntheticClassifierFactory:
             logging.info("Starting classifier training...")
             try:
                 # Convert activations to the format expected by train_on_activations method
-                from wisent_guard.core.activations import Activations
+                from wisent.core.activations import Activations
 
                 # Convert torch tensors to Activations objects if needed
                 harmful_activations = []
                 harmless_activations = []
 
-                from wisent_guard.core.layer import Layer
+                from wisent.core.layer import Layer
 
                 layer_obj = Layer(index=15, type="transformer")
 
@@ -598,7 +598,7 @@ def create_classifier_from_trait_description(
     log_and_print(f"🧠 Extracting activations from {len(pair_set.pairs)} pairs...")
 
     # Create Layer object for activation extraction
-    from wisent_guard.core.layer import Layer
+    from wisent.core.layer import Layer
 
     layer_obj = Layer(index=15, type="transformer")
     log_and_print(f"🔧 Created Layer object: index={layer_obj.index}, type={layer_obj.type}")
@@ -655,7 +655,7 @@ def create_classifier_from_trait_description(
     log_and_print("🎯 Starting classifier training...")
     try:
         # Convert activations to the format expected by train_on_activations method
-        from wisent_guard.core.activations import Activations
+        from wisent.core.activations import Activations
 
         # Convert torch tensors to Activations objects if needed
         harmful_activations = []

{wisent-0.5.1 → wisent-0.5.2}/wisent/core/agent/diagnose/test_synthetic_classifier.py

@@ -7,12 +7,12 @@ from pathlib import Path
 project_root = Path(__file__).parent.parent.parent.parent
 sys.path.insert(0, str(project_root))
 
-from wisent_guard.core.model import Model
-from wisent_guard.core.agent.diagnose.synthetic_classifier_option import (
+from wisent.core.model import Model
+from wisent.core.agent.diagnose.synthetic_classifier_option import (
     create_classifiers_for_prompt,
     apply_classifiers_to_response
 )
-from wisent_guard.core.agent.budget import set_time_budget
+from wisent.core.agent.budget import set_time_budget
 
 class TimeoutError(Exception):
     """Raised when test exceeds time budget."""

{wisent-0.5.1 → wisent-0.5.2}/wisent/core/agent/diagnose.py

@@ -11,8 +11,8 @@ This module handles:
 from dataclasses import dataclass
 from typing import Any, Dict, List
 
-from wisent_guard.core.activations import ActivationAggregationStrategy, Activations
-from wisent_guard.core.classifier.classifier import Classifier
+from wisent.core.activations import ActivationAggregationStrategy, Activations
+from wisent.core.classifier.classifier import Classifier
 
 from ..layer import Layer
 from ..model import Model