scripts/utils.py

import pandas as pd
from pathlib import Path

SCRIPTS_DIR = Path(__file__).parent
ROOT_DIR = SCRIPTS_DIR.parent
DATA_DIR = ROOT_DIR / "data"
TMP_DIR = ROOT_DIR / "tmp"


def get_traders_family(row: pd.DataFrame) -> str:
    if row.staking == "non_agent":
        return "non_agent"
    elif row.market_creator == "pearl":
        return "pearl_agent"
    # quickstart
    return "quickstart_agent"