sim = asr.Simulate(
    fm,
    labels=project.data_store["included"],
    classifier=load_extension("models.classifiers", "svm")(),
    query_strategy=load_extension("models.query", "max_random")(),
    balance_strategy=load_extension("models.balance", "balanced")(),
    feature_extraction=feature_model,
    n_query=lambda x: 2,
)
sim.review()

assert sim._results["training_set"].to_list() == [None, None, 2, 2, 4, 4]

def n_query(x):
    return len(x) // 2

sim = asr.Simulate(
    fm,
    labels=project.data_store["included"],
    classifier=load_extension("models.classifiers", "svm")(),
    query_strategy=load_extension("models.query", "max_random")(),
    balance_strategy=load_extension("models.balance", "balanced")(),
    feature_extraction=feature_model,
    n_query=n_query,
)
sim.review()

assert sim._results["training_set"].to_list() == [None, None, 2, 3, 4, 4]