filter on forecast_uuid too (#100)

* filter on forecast_uuid too * fix * format
openclimatefix · Nov 19, 2024 · e3e60bb · e3e60bb
1 parent 84e25bc
commit e3e60bb
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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()