diff --git a/database-cleanup/database_cleanup/save.py b/database-cleanup/database_cleanup/save.py index 0c19128..b804746 100644 --- a/database-cleanup/database_cleanup/save.py +++ b/database-cleanup/database_cleanup/save.py @@ -73,7 +73,11 @@ def save_forecast_and_values( fs.mkdir(directory) if site_uuids is not None: - stmt = sa.select(ForecastSQL.forecast_uuid).filter(ForecastSQL.site_uuid.in_(site_uuids)) + stmt = ( + sa.select(ForecastSQL.forecast_uuid) + .filter(ForecastSQL.site_uuid.in_(site_uuids)) + .filter(ForecastSQL.forecast_uuid.in_(forecast_uuids)) + ) forecast_uuids = session.scalars(stmt).all()