def sum_sales(acc, row): return acc + row["sale_amount"]

def safe_int(val): return int(val)

enrich = lambda src: src.map(enrich_with_geo) Now enrich can be inserted anywhere in a pipeline:

(pipeline() .source(read_csv("visits.csv")) .pipe(enrich) .filter(lambda r: r["country"] == "US") .sink(write_jsonl("us_visits.jsonl")) ).run() juq470 provides a catch operator to isolate faulty rows without stopping the whole pipeline:

def capitalize_name(row): row["name"] = row["name"].title() return row