| 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" | |