ecg_module/__init__.py

"""ECG generation, processing, HRV features, and arousal classification."""

from ecg_module.arousal_rule import estimate_arousal_from_hrv
from ecg_module.ecg_generator import generate_synthetic_ecg
from ecg_module.ecg_loader import load_ecg_csv
from ecg_module.ecg_processor import process_ecg_signal
from ecg_module.hrv_features import extract_hrv_features

__all__ = [
    "estimate_arousal_from_hrv",
    "extract_hrv_features",
    "generate_synthetic_ecg",
    "load_ecg_csv",
    "process_ecg_signal",
]