import pandas as pd
from src.outlier_detection import OutlierDetector  
from zenml import step


@step
def outlier_detection_step(data: pd.DataFrame) -> pd.DataFrame:
    
    """
    A ZenML step to detect outliers in a given DataFrame using the OutlierDetector class.

    Parameters
    data : pd.DataFrame
        The DataFrame containing the data to detect outliers in.

    Returns
    
    pd.DataFrame
        The DataFrame with outliers removed.
    """

    detector = OutlierDetector(data)
    cleaned_data = detector.run_outlier_detection()
    return cleaned_data