Ramp’s ML platform, rebuilt for the agent era

1from prefect import flow, task
2from prefect_aws.ecs import ecs
3
4
5@task(retries=2)
6def load_features(date: str) -> list[dict]:
7    return warehouse.fetch(date)
8
9
10@task
11def predict(features: list[dict]) -> list[float]:
12    return model.score(features)
13
14
15# One decorator: production-like permissions + bigger compute.
16@ecs(cpu=4096, memory=16384)
17@flow(name="daily-predict")
18def daily_predict(date: str):
19    features = load_features(date)
20    return predict(features)