Function f

analysis/wandb_api.py:55–58

(df)

def df_larger_than(**kwargs) -> DF_MAPPING:
    def f(df) -> pd.DataFrame:
        for k, v in kwargs.items():
            df = df.loc[getattr(df, k) > v]
        return df

    return f