functions.py

from zeno import distill, metric, ZenoOptions
from pandas import DataFrame


@distill
def length(df: DataFrame, ops: ZenoOptions):
    return df["prompt"].str.len()


@metric
def avg_image_nswf(df: DataFrame, ops: ZenoOptions):
    return float(df["image_nsfw"].dropna().mean())


@metric
def avg_prompt_nsfw(df: DataFrame, ops: ZenoOptions):
    return float(df["prompt_nsfw"].dropna().mean())