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

wisent/core/autonomous_agent.py

@@ -12,7 +12,7 @@ A model that can autonomously use wisent-guard capabilities on itself:
 import asyncio
 from typing import Any, Dict, List, Optional
 
-from wisent_guard.core.activations import ActivationAggregationStrategy, Activations
+from wisent.core.activations import ActivationAggregationStrategy, Activations
 
 from .agent.diagnose import AgentClassifierDecisionSystem, AnalysisResult, ClassifierMarketplace, ResponseDiagnostics
 from .agent.steer import ImprovementResult, ResponseSteering

wisent/core/contrastive_pairs/core/atoms.py

@@ -5,7 +5,7 @@ from abc import ABC, abstractmethod
 from typing import Iterable, TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from wisent_guard.core.activations.core.atoms import LayerActivations
+    from wisent.core.activations.core.atoms import LayerActivations
 
 __all__ = [
     "AtomResponse",

wisent/core/contrastive_pairs/core/buliders.py

@@ -4,9 +4,9 @@ import logging
 from typing import Iterable
 
 
-from wisent_guard.core.contrastive_pairs.core.response import NegativeResponse, PositiveResponse  
-from wisent_guard.core.contrastive_pairs.core.pair import ContrastivePair              
-from wisent_guard.core.contrastive_pairs.core.set import ContrastivePairSet
+from wisent.core.contrastive_pairs.core.response import NegativeResponse, PositiveResponse  
+from wisent.core.contrastive_pairs.core.pair import ContrastivePair              
+from wisent.core.contrastive_pairs.core.set import ContrastivePairSet
 
 __all__ = [
     "from_phrase_pairs",

wisent/core/contrastive_pairs/core/pair.py

@@ -2,13 +2,13 @@ from __future__ import annotations
 
 from dataclasses import dataclass, replace 
 
-from wisent_guard.core.contrastive_pairs.core.atoms import AtomContrastivePair
-from wisent_guard.core.contrastive_pairs.core.response import NegativeResponse, PositiveResponse
+from wisent.core.contrastive_pairs.core.atoms import AtomContrastivePair
+from wisent.core.contrastive_pairs.core.response import NegativeResponse, PositiveResponse
 
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from wisent_guard.core.activations.core.atoms import LayerActivations, RawActivationMap
+    from wisent.core.activations.core.atoms import LayerActivations, RawActivationMap
 
 __all__ = [
     "ContrastivePair",
@@ -167,7 +167,7 @@ class ContrastivePair(AtomContrastivePair):
          )          
         '''
 
-        from wisent_guard.core.contrastive_pairs.core.response import NegativeResponse, PositiveResponse
+        from wisent.core.contrastive_pairs.core.response import NegativeResponse, PositiveResponse
 
         return cls(
             prompt=str(data["prompt"]),

wisent/core/contrastive_pairs/core/response.py

@@ -2,8 +2,8 @@ from __future__ import annotations
 
 from dataclasses import dataclass, replace
 
-from wisent_guard.core.contrastive_pairs.core.atoms import AtomResponse
-from wisent_guard.core.activations.core.atoms import LayerActivations, RawActivationMap
+from wisent.core.contrastive_pairs.core.atoms import AtomResponse
+from wisent.core.activations.core.atoms import LayerActivations, RawActivationMap
 
 __all__ = [
     "Response",

wisent/core/contrastive_pairs/core/serialization.py

@@ -9,8 +9,8 @@ from pathlib import Path
 import numpy as np
 import torch
 
-from wisent_guard.core.contrastive_pairs.core.pair import ContrastivePair
-from wisent_guard.core.contrastive_pairs.core.set import ContrastivePairSet
+from wisent.core.contrastive_pairs.core.pair import ContrastivePair
+from wisent.core.contrastive_pairs.core.set import ContrastivePairSet
 
 __all__ = [
     "save_contrastive_pair_set",

wisent/core/contrastive_pairs/core/set.py

@@ -7,10 +7,10 @@ from dataclasses import dataclass, field
 from typing import Optional
 from typing import TYPE_CHECKING
 
-from wisent_guard.core.contrastive_pairs.core.atoms import AtomContrastivePairSet
-from wisent_guard.core.contrastive_pairs.diagnostics import DiagnosticsConfig, DiagnosticsReport, run_all_diagnostics
+from wisent.core.contrastive_pairs.core.atoms import AtomContrastivePairSet
+from wisent.core.contrastive_pairs.diagnostics import DiagnosticsConfig, DiagnosticsReport, run_all_diagnostics
 
-from wisent_guard.core.contrastive_pairs.core.pair import ContrastivePair 
+from wisent.core.contrastive_pairs.core.pair import ContrastivePair 
 
 __all__ = [
     "ContrastivePairSet",

wisent/core/contrastive_pairs/diagnostics/control_vectors.py

@@ -8,7 +8,7 @@ from typing import Mapping
 
 import torch
 
-from wisent_guard.core.activations.core.atoms import LayerActivations, RawActivationMap
+from wisent.core.activations.core.atoms import LayerActivations, RawActivationMap
 
 from .base import DiagnosticsIssue, DiagnosticsReport, MetricReport
 

wisent/core/contrastive_pairs/lm_eval_pairs/atoms.py

@@ -6,7 +6,7 @@ from abc import ABC, abstractmethod
 
 
 if TYPE_CHECKING:
-    from wisent_guard.core.contrastive_pairs.core.pair import ContrastivePair
+    from wisent.core.contrastive_pairs.core.pair import ContrastivePair
     from lm_eval.api.task import ConfigurableTask
 
 

wisent/core/contrastive_pairs/lm_eval_pairs/lm_extractor_registry.py

@@ -4,12 +4,12 @@ from typing import Type, Union
 import importlib
 import logging
 
-from wisent_guard.core.contrastive_pairs.lm_eval_pairs.atoms import (
+from wisent.core.contrastive_pairs.lm_eval_pairs.atoms import (
     LMEvalBenchmarkExtractor,
     UnsupportedLMEvalBenchmarkError,
 )
 
-from wisent_guard.core.contrastive_pairs.lm_eval_pairs.lm_extractor_manifest import EXTRACTORS as _MANIFEST
+from wisent.core.contrastive_pairs.lm_eval_pairs.lm_extractor_manifest import EXTRACTORS as _MANIFEST
 
 __all__ = [
     "register_extractor",
@@ -37,8 +37,8 @@ def register_extractor(name: str, ref: Union[str, Type[LMEvalBenchmarkExtractor]
             If the ref class does not subclass LMEvalBenchmarkExtractor.
 
     example:
-        >>> from wisent_guard.core.contrastive_pairs.lm_eval_pairs.lm_extractor_registry import register_extractor
-        >>> from wisent_guard.core.contrastive_pairs.lm_eval_pairs.atoms import LMEvalBenchmarkExtractor
+        >>> from wisent.core.contrastive_pairs.lm_eval_pairs.lm_extractor_registry import register_extractor
+        >>> from wisent.core.contrastive_pairs.lm_eval_pairs.atoms import LMEvalBenchmarkExtractor
         >>> class MyExtractor(LMEvalBenchmarkExtractor): ...
         >>> register_extractor("mytask", MyExtractor)
         >>> register_extractor("mytask2", "my_module:MyExtractor")

wisent/core/contrastive_pairs/lm_eval_pairs/lm_task_extractors/winogrande.py

@@ -2,10 +2,10 @@ from __future__ import annotations
 
 from typing import Any, TYPE_CHECKING
 
-from wisent_guard.core.contrastive_pairs.core.pair import ContrastivePair
-from wisent_guard.core.contrastive_pairs.core.response import NegativeResponse, PositiveResponse
-from wisent_guard.core.contrastive_pairs.lm_eval_pairs.atoms import LMEvalBenchmarkExtractor
-from wisent_guard.cli.cli_logger import setup_logger, bind
+from wisent.core.contrastive_pairs.core.pair import ContrastivePair
+from wisent.core.contrastive_pairs.core.response import NegativeResponse, PositiveResponse
+from wisent.core.contrastive_pairs.lm_eval_pairs.atoms import LMEvalBenchmarkExtractor
+from wisent.cli.cli_logger import setup_logger, bind
 
 if TYPE_CHECKING:
     from lm_eval.api.task import ConfigurableTask

wisent/core/contrastive_pairs/lm_eval_pairs/lm_task_pairs_generation.py

@@ -2,12 +2,12 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING
 
-from wisent_guard.core.contrastive_pairs.lm_eval_pairs.lm_extractor_registry import get_extractor
-from wisent_guard.cli.cli_logger import setup_logger, bind
+from wisent.core.contrastive_pairs.lm_eval_pairs.lm_extractor_registry import get_extractor
+from wisent.cli.cli_logger import setup_logger, bind
 
 if TYPE_CHECKING:
     from lm_eval.api.task import ConfigurableTask
-    from wisent_guard.core.contrastive_pairs.core.pair import ContrastivePair
+    from wisent.core.contrastive_pairs.core.pair import ContrastivePair
 
 __all__ = ["build_contrastive_pairs"]
 _LOG = setup_logger(__name__)

wisent/core/data_loaders/core/atoms.py

@@ -6,7 +6,7 @@ from typing import Any, Dict, Type
 
 from typing import TypedDict, Mapping
 from lm_eval.api.task import ConfigurableTask
-from wisent_guard.core.contrastive_pairs.core.set import ContrastivePairSet
+from wisent.core.contrastive_pairs.core.set import ContrastivePairSet
 
 __all__ = ["DataLoaderError", "BaseDataLoader"]
 

wisent/core/data_loaders/loaders/custom.py

@@ -2,9 +2,9 @@ from __future__ import annotations
 from typing import Any, Iterable
 import logging
 
-from wisent_guard.core.data_loaders.core.atoms import BaseDataLoader, DataLoaderError, LoadDataResult
-from wisent_guard.core.contrastive_pairs.core.set import ContrastivePairSet         
-from wisent_guard.core.contrastive_pairs.core.serialization import load_contrastive_pair_set
+from wisent.core.data_loaders.core.atoms import BaseDataLoader, DataLoaderError, LoadDataResult
+from wisent.core.contrastive_pairs.core.set import ContrastivePairSet         
+from wisent.core.contrastive_pairs.core.serialization import load_contrastive_pair_set
 
 __all__ = [
     "CustomUserDataLoader",

wisent/core/data_loaders/loaders/lm_loader.py

@@ -2,12 +2,12 @@ from __future__ import annotations
 from typing import Any, TYPE_CHECKING
 import logging
 
-from wisent_guard.core.data_loaders.core.atoms import BaseDataLoader, DataLoaderError, LoadDataResult
-from wisent_guard.core.contrastive_pairs.core.pair import ContrastivePair
-from wisent_guard.core.contrastive_pairs.core.set import ContrastivePairSet
+from wisent.core.data_loaders.core.atoms import BaseDataLoader, DataLoaderError, LoadDataResult
+from wisent.core.contrastive_pairs.core.pair import ContrastivePair
+from wisent.core.contrastive_pairs.core.set import ContrastivePairSet
 from lm_eval.tasks import get_task_dict
 from lm_eval.tasks import TaskManager as LMTaskManager
-from wisent_guard.core.contrastive_pairs.lm_eval_pairs.lm_task_pairs_generation import (
+from wisent.core.contrastive_pairs.lm_eval_pairs.lm_task_pairs_generation import (
     lm_build_contrastive_pairs,
 )
 

wisent/core/download_full_benchmarks.py

@@ -911,7 +911,7 @@ class FullBenchmarkDownloader:
     def _convert_mbpp_format(self, sample: Dict[str, Any]) -> List[Dict[str, Any]]:
         """Convert MBPP format (programming problems with code)."""
         # Use the benchmark extractor to get contrastive pairs
-        from wisent_guard.core.benchmark_extractors import extract_contrastive_pair
+        from wisent.core.benchmark_extractors import extract_contrastive_pair
 
         try:
             contrastive_data = extract_contrastive_pair("mbpp", sample, None)
@@ -992,7 +992,7 @@ class FullBenchmarkDownloader:
             return []
 
         # Use the HLE extractor to get contrastive pairs
-        from wisent_guard.core.benchmark_extractors import HLEExtractor
+        from wisent.core.benchmark_extractors import HLEExtractor
 
         try:
             extractor = HLEExtractor()

wisent/core/evaluators/oracles/interactive.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import sys
 from typing import Any
 
-from wisent_guard.core.evaluators.core.atoms import BaseEvaluator, EvalResult, EvaluatorError
+from wisent.core.evaluators.core.atoms import BaseEvaluator, EvalResult, EvaluatorError
 
 __all__ = [
     "InteractiveEvaluator",

wisent/core/evaluators/oracles/nlp_evaluator.py

@@ -1,7 +1,7 @@
 import re
 from typing import Any, Mapping
 
-from wisent_guard.core.evaluators.core.atoms import BaseEvaluator, EvalResult
+from wisent.core.evaluators.core.atoms import BaseEvaluator, EvalResult
 
 __all__ = [
     "NLPEvaluator",

wisent/core/evaluators/oracles/user_specified.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 from typing import Any, Optional
 
-from wisent_guard.core.evaluators.core.atoms import BaseEvaluator, EvalResult
+from wisent.core.evaluators.core.atoms import BaseEvaluator, EvalResult
 
 __all__ = [
     "UserSpecifiedEvaluator",

wisent/core/lm_eval_harness_ground_truth.py

@@ -7,8 +7,8 @@ This module provides ground truth evaluation using the lm-eval-harness framework
 import logging
 from typing import Any, Dict
 
-from wisent_guard.core.activations import ActivationAggregationStrategy, Activations
-from wisent_guard.core.layer import Layer
+from wisent.core.activations import ActivationAggregationStrategy, Activations
+from wisent.core.layer import Layer
 
 logger = logging.getLogger(__name__)
 

wisent/core/log_likelihoods_evaluator.py

@@ -10,8 +10,8 @@ directly on each choice to evaluate performance against known ground truth.
 import logging
 from typing import Any, Dict, Optional
 
-from wisent_guard.core.activations import ActivationAggregationStrategy, Activations
-from wisent_guard.core.layer import Layer
+from wisent.core.activations import ActivationAggregationStrategy, Activations
+from wisent.core.layer import Layer
 
 logger = logging.getLogger(__name__)
 
@@ -87,10 +87,10 @@ class LogLikelihoodsEvaluator:
             logger.info(f"Extracted {len(qa_pairs)} QA pairs from {task_name}")
 
             # Use existing contrastive pair creation infrastructure
-            from wisent_guard.core.activations.activation_collection_method import (
+            from wisent.core.activations.activation_collection_method import (
                 ActivationCollectionLogic,
             )
-            from wisent_guard.core.activations.prompts import PromptConstructionStrategy
+            from wisent.core.activations.prompts import PromptConstructionStrategy
 
             collector = ActivationCollectionLogic(model=evaluation_model)
 

wisent/core/models/core/atoms.py

@@ -7,7 +7,7 @@ import torch
 from typing import Mapping
 
 if TYPE_CHECKING:
-    from wisent_guard.core.activations.core.atoms import RawActivationMap 
+    from wisent.core.activations.core.atoms import RawActivationMap 
 
 
 __all__ = [

wisent/core/models/wisent_model.py

@@ -16,12 +16,12 @@ from transformers import (
 
 
 
-from wisent_guard.core.models.core.atoms import SteeringPlan, SteeringVector, HookHandleGroup, GenerationStats, TopLogits
-from wisent_guard.core.activations.core.atoms import RawActivationMap
+from wisent.core.models.core.atoms import SteeringPlan, SteeringVector, HookHandleGroup, GenerationStats, TopLogits
+from wisent.core.activations.core.atoms import RawActivationMap
 
-from wisent_guard.core.prompts.core.atom import ChatMessage
-from wisent_guard.core.utils.device import resolve_default_device, resolve_torch_device
-from wisent_guard.core.contrastive_pairs.diagnostics import run_control_steering_diagnostics
+from wisent.core.prompts.core.atom import ChatMessage
+from wisent.core.utils.device import resolve_default_device, resolve_torch_device
+from wisent.core.contrastive_pairs.diagnostics import run_control_steering_diagnostics
 
 import threading
 

wisent/core/optuna/__init__.py

@@ -21,7 +21,7 @@ Key components:
 
 # Steering optimization components
 # Classifier optimization components
-from wisent_guard.core.optuna.classifier import (
+from wisent.core.optuna.classifier import (
     ActivationGenerator,
     CacheConfig,
     ClassifierCache,
@@ -30,13 +30,13 @@ from wisent_guard.core.optuna.classifier import (
     OptimizationResult,
     OptunaClassifierOptimizer,
 )
-from wisent_guard.core.optuna.steering.metrics import (
+from wisent.core.optuna.steering.metrics import (
     calculate_comprehensive_metrics,
     evaluate_benchmark_performance,
     evaluate_probe_performance,
     generate_performance_summary,
 )
-from wisent_guard.core.optuna.steering.optuna_pipeline import OptimizationConfig, OptimizationPipeline
+from wisent.core.optuna.steering.optuna_pipeline import OptimizationConfig, OptimizationPipeline
 
 __all__ = [
     # Steering optimization

wisent/core/optuna/classifier/activation_generator.py

@@ -15,8 +15,8 @@ from typing import Any, Optional
 import numpy as np
 import torch
 
-from wisent_guard.core.activations.activation_collection_method import ActivationCollectionLogic
-from wisent_guard.core.activations.core import ActivationAggregationStrategy, Activations
+from wisent.core.activations.activation_collection_method import ActivationCollectionLogic
+from wisent.core.activations.core import ActivationAggregationStrategy, Activations
 
 logger = logging.getLogger(__name__)
 

wisent/core/optuna/classifier/classifier_cache.py

@@ -16,7 +16,7 @@ from typing import Any, Optional
 
 import torch
 
-from wisent_guard.core.classifier.classifier import Classifier
+from wisent.core.classifier.classifier import Classifier
 
 logger = logging.getLogger(__name__)
 

wisent/core/optuna/classifier/optuna_classifier_optimizer.py

@@ -16,8 +16,8 @@ import torch
 from optuna.pruners import MedianPruner
 from optuna.samplers import TPESampler
 
-from wisent_guard.core.classifier.classifier import Classifier
-from wisent_guard.core.utils.device import resolve_default_device
+from wisent.core.classifier.classifier import Classifier
+from wisent.core.utils.device import resolve_default_device
 
 from .activation_generator import ActivationData, ActivationGenerator, GenerationConfig
 from .classifier_cache import CacheConfig, ClassifierCache

wisent/core/optuna/steering/bigcode_evaluator_wrapper.py

@@ -8,9 +8,9 @@ evaluation with the optuna optimization pipeline.
 import logging
 from typing import Any, Dict, List, Tuple
 
-from wisent_guard.core.bigcode_extractors import get_bigcode_extractor
-from wisent_guard.core.bigcode_integration import BigCodeEvaluator, is_bigcode_task
-from wisent_guard.parameters.task_config import CODING_TASKS
+from wisent.core.bigcode_extractors import get_bigcode_extractor
+from wisent.core.bigcode_integration import BigCodeEvaluator, is_bigcode_task
+from wisent.parameters.task_config import CODING_TASKS
 
 logger = logging.getLogger(__name__)
 

wisent/core/optuna/steering/data_utils.py

@@ -11,11 +11,11 @@ from tqdm import tqdm
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 # Import LMEvalHarnessGroundTruth for intelligent evaluation (same approach as CLI)
-from wisent_guard.core.lm_eval_harness_ground_truth import LMEvalHarnessGroundTruth
+from wisent.core.lm_eval_harness_ground_truth import LMEvalHarnessGroundTruth
 
 # Import task interface for dynamic task loading
-from wisent_guard.core.task_interface import get_task
-from wisent_guard.core.utils.device import empty_device_cache, preferred_dtype, resolve_default_device
+from wisent.core.task_interface import get_task
+from wisent.core.utils.device import empty_device_cache, preferred_dtype, resolve_default_device
 
 logger = logging.getLogger(__name__)
 

wisent/core/optuna/steering/metrics.py

@@ -8,13 +8,13 @@ from typing import Any, Callable, Dict, List, Optional
 import numpy as np
 from sklearn.metrics import accuracy_score, precision_recall_fscore_support, roc_auc_score
 
-from wisent_guard.core.bigcode_extractors import MBPPExtractor
+from wisent.core.bigcode_extractors import MBPPExtractor
 
 # Import LMEvalHarnessGroundTruth for intelligent evaluation (newer approach used by CLI)
-from wisent_guard.core.lm_eval_harness_ground_truth import LMEvalHarnessGroundTruth
-from wisent_guard.core.task_interface import get_task
-from wisent_guard.core.tasks.file_task import FileTask
-from wisent_guard.parameters.task_config import CODING_TASKS
+from wisent.core.lm_eval_harness_ground_truth import LMEvalHarnessGroundTruth
+from wisent.core.task_interface import get_task
+from wisent.core.tasks.file_task import FileTask
+from wisent.parameters.task_config import CODING_TASKS
 
 from .bigcode_evaluator_wrapper import OptunaBigCodeEvaluator
 

wisent/core/optuna/steering/optuna_pipeline.py

@@ -39,13 +39,13 @@ try:
     WANDB_AVAILABLE = True
 except ImportError:
     WANDB_AVAILABLE = False
-from wisent_guard.core.contrastive_pairs.contrastive_pair import ContrastivePair
-from wisent_guard.core.contrastive_pairs.contrastive_pair_set import ContrastivePairSet
-from wisent_guard.core.optuna.steering import data_utils, metrics
-from wisent_guard.core.response import Response
-from wisent_guard.core.steering_methods.dac import DAC
-from wisent_guard.core.task_interface import get_task
-from wisent_guard.core.utils.device import empty_device_cache, preferred_dtype, resolve_default_device, resolve_device
+from wisent.core.contrastive_pairs.contrastive_pair import ContrastivePair
+from wisent.core.contrastive_pairs.contrastive_pair_set import ContrastivePairSet
+from wisent.core.optuna.steering import data_utils, metrics
+from wisent.core.response import Response
+from wisent.core.steering_methods.dac import DAC
+from wisent.core.task_interface import get_task
+from wisent.core.utils.device import empty_device_cache, preferred_dtype, resolve_default_device, resolve_device
 
 logger = logging.getLogger(__name__)
 
@@ -559,7 +559,7 @@ class OptimizationPipeline:
 
         if method_name == "caa":
             # Create CAA instance
-            from wisent_guard.core.steering_methods.caa import CAA
+            from wisent.core.steering_methods.caa import CAA
 
             caa = CAA(device=self.device)
 

wisent/core/optuna/steering/steering_optimization.py

@@ -14,21 +14,21 @@ from typing import Any, Dict, List, Optional, Tuple
 import torch
 from tqdm import tqdm
 
-from wisent_guard.core.activations.core import ActivationAggregationStrategy
-from wisent_guard.core.classifier.classifier import Classifier
-from wisent_guard.core.contrastive_pairs.contrastive_pair import ContrastivePair
-from wisent_guard.core.contrastive_pairs.contrastive_pair_set import ContrastivePairSet
-from wisent_guard.core.optuna.classifier import (
+from wisent.core.activations.core import ActivationAggregationStrategy
+from wisent.core.classifier.classifier import Classifier
+from wisent.core.contrastive_pairs.contrastive_pair import ContrastivePair
+from wisent.core.contrastive_pairs.contrastive_pair_set import ContrastivePairSet
+from wisent.core.optuna.classifier import (
     CacheConfig,
     ClassifierCache,
     ClassifierOptimizationConfig,
     GenerationConfig,
     OptunaClassifierOptimizer,
 )
-from wisent_guard.core.optuna.steering import data_utils, metrics
-from wisent_guard.core.response import Response
-from wisent_guard.core.steering_methods.dac import DAC
-from wisent_guard.core.task_interface import get_task
+from wisent.core.optuna.steering import data_utils, metrics
+from wisent.core.response import Response
+from wisent.core.steering_methods.dac import DAC
+from wisent.core.task_interface import get_task
 
 logger = logging.getLogger(__name__)
 

wisent/core/prompts/core/prompt_formater.py

@@ -5,7 +5,7 @@ import inspect
 import pkgutil
 from typing import Iterable, Type
 
-from wisent_guard.core.prompts.core.atom import PromptPair, PromptStrategy, UnknownStrategyError
+from wisent.core.prompts.core.atom import PromptPair, PromptStrategy, UnknownStrategyError
 
 __all__ = ["PromptFormatter", "StrategyKey", "UnknownStrategyError"]
 

wisent/core/prompts/prompt_stratiegies/direct_completion.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from wisent_guard.core.prompts.core.atom import ChatMessage, PromptPair, PromptStrategy
+from wisent.core.prompts.core.atom import ChatMessage, PromptPair, PromptStrategy
 
 __all__ = ["DirectCompletionStrategy"]
 

wisent/core/prompts/prompt_stratiegies/instruction_following.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from wisent_guard.core.prompts.core.atom import ChatMessage, PromptPair, PromptStrategy
+from wisent.core.prompts.core.atom import ChatMessage, PromptPair, PromptStrategy
 
 __all__ = ["InstructionFollowingStrategy"]
 

wisent/core/prompts/prompt_stratiegies/multiple_choice.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from wisent_guard.core.prompts.core.atom import ChatMessage, PromptPair, PromptStrategy
+from wisent.core.prompts.core.atom import ChatMessage, PromptPair, PromptStrategy
 
 __all__ = ["MultipleChoiceStrategy"]
 

wisent/core/prompts/prompt_stratiegies/role_playing.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from wisent_guard.core.prompts.core.atom import ChatMessage, PromptPair, PromptStrategy
+from wisent.core.prompts.core.atom import ChatMessage, PromptPair, PromptStrategy
 
 __all__ = ["RolePlayingStrategy"]
 

wisent/core/sample_size_optimizer.py

@@ -13,7 +13,7 @@ import matplotlib.pyplot as plt
 import numpy as np
 from sklearn.metrics import accuracy_score, f1_score, precision_score, recall_score
 
-from wisent_guard.core.classifier.classifier import Classifier
+from wisent.core.classifier.classifier import Classifier
 
 from .activations import ActivationAggregationStrategy
 from .contrastive_pairs import ContrastivePairSet
@@ -171,7 +171,7 @@ class SampleSizeOptimizer:
         logger.info(f"Extracted {len(qa_pairs)} QA pairs")
 
         # Create contrastive pairs from QA pairs
-        from wisent_guard.core.activations.activation_collection_method import ActivationCollectionLogic
+        from wisent.core.activations.activation_collection_method import ActivationCollectionLogic
 
         collector = ActivationCollectionLogic(model=self.model)
 

wisent/core/steering.py

@@ -7,8 +7,8 @@ from typing import Any, Dict, List, Optional, Union
 import torch
 import torch.nn.functional as F
 
-from wisent_guard.core.activations import Activations
-from wisent_guard.core.classifier.classifier import Classifier
+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/steering_methods/core/atoms.py

@@ -7,8 +7,8 @@ import inspect
 
 import torch
 
-from wisent_guard.core.activations.core.atoms import LayerActivations, RawActivationMap, LayerName  
-from wisent_guard.core.contrastive_pairs.core.set import ContrastivePairSet  
+from wisent.core.activations.core.atoms import LayerActivations, RawActivationMap, LayerName  
+from wisent.core.contrastive_pairs.core.set import ContrastivePairSet  
 
 __all__ = [
     "SteeringError",

wisent/core/steering_methods/methods/caa.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 from typing import List
 import torch
 
-from wisent_guard.core.steering_methods.core.atoms import PerLayerBaseSteeringMethod
+from wisent.core.steering_methods.core.atoms import PerLayerBaseSteeringMethod
 
 __all__ = [
     "CAAMethod",

wisent/core/steering_optimizer.py

@@ -806,7 +806,7 @@ class SteeringOptimizer:
         """
         try:
             # Import CLI runner to test configuration
-            from wisent_guard.cli import run_task_pipeline
+            from wisent.cli import run_task_pipeline
             
             # Prepare kwargs with method-specific parameters
             kwargs = {
@@ -1270,7 +1270,7 @@ def get_optimal_steering_params(
 # The following integration points need to be implemented:
 #
 # 1. CAA Integration:
-#    - Load existing CAA implementation from wisent_guard.core.steering_methods.caa
+#    - Load existing CAA implementation from wisent.core.steering_methods.caa
 #    - Implement parameter optimization for CAA vectors
 #    - Measure CAA steering effectiveness
 #