# -*- coding: utf-8 -*-
from pathlib import Path

import importlib.util
import pandas as pd


ROOT = Path(__file__).resolve().parents[2]
DATA = ROOT / "local_batch_api" / "output_full" / "training_dataset.csv"


def main():
    df = pd.read_csv(DATA)
    print(f"shape={df.shape}")
    print("columns:")
    for col in df.columns:
        print(col)
    print("head_json:")
    print(df.head(2).to_json(force_ascii=False, orient="records"))
    print("libraries:")
    for name in ["pandas", "numpy", "sklearn", "xgboost", "lightgbm"]:
        print(f"{name}={bool(importlib.util.find_spec(name))}")


if __name__ == "__main__":
    main()