Merge pull request #2225 from bluewave-labs/develop-saas

Develop saas -> Master saas (26 Sept)

Author: HarshP4585

BiasAndFairnessModule/run_full_evaluation.py

@@ -15,6 +15,8 @@
 import pandas as pd
 import numpy as np
 import json
+import traceback
+import re
 
 # Add the src directory to Python path
 current_dir = Path(__file__).parent
@@ -282,7 +284,6 @@ def _normalize_predictions_to_tokens(config) -> None:
         print(f"   ❌ Prediction normalization skipped: file not found at {results_path}")
         return
 
-    import re
 
     def extract_numeric_value(text: str) -> float:
         """Extract a numeric value from text; understands '55K' and '50,000' forms.
@@ -411,7 +412,6 @@ def main():
         return results
     except Exception as e:
         print(f"\n❌ Pipeline execution failed: {str(e)}")
-        import traceback
         traceback.print_exc()
         raise
 

BiasAndFairnessModule/src/__init__.py

@@ -8,17 +8,11 @@
 from .core import ConfigManager, Config
 from .model_loader import ModelLoader, load_sklearn_model
 from .dataset_loader import DataLoader
-from .inference import ModelInferencePipeline, run_comprehensive_evaluation
+from .inference import ModelInferencePipeline
 from .eval_engine import (
-    FairnessEvaluator,
-    FairnessCompassEngine,
-    route_metric,
     PostProcessor,
     summarize_results,
-    clean_predictions,
-    register_metric,
-    get_metric,
-    list_metrics
+    clean_predictions
 )
 
 __version__ = "0.1.0"
@@ -38,18 +32,9 @@
 
     # Inference
     'ModelInferencePipeline',
-    'run_comprehensive_evaluation',
 
     # Evaluation
-    'FairnessEvaluator',
-    'FairnessCompassEngine',
-    'route_metric',
     'PostProcessor',
     'summarize_results',
-    'clean_predictions',
-    
-    # Metrics
-    'register_metric',
-    'get_metric',
-    'list_metrics'
+    'clean_predictions'
 ]

BiasAndFairnessModule/src/core/__init__.py

@@ -4,6 +4,5 @@
 
 from .config import ConfigManager, Config
 from .common import read_yaml
-from .cli import setup_logging, run_prompt_evaluation, run_predict_evaluation
 
-__all__ = ['ConfigManager', 'Config', 'read_yaml', 'setup_logging', 'run_prompt_evaluation', 'run_predict_evaluation']
+__all__ = ['ConfigManager', 'Config', 'read_yaml']