File size: 398 Bytes
efabdf9
1ed82ec
 
 
 
 
 
efabdf9
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
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"