fix import

openclimatefix · Dec 13, 2023 · 3c46248 · 3c46248
1 parent a804428
commit 3c46248
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/end2end/test_metnet_training.py b/tests/end2end/test_metnet_training.py
@@ -8,7 +8,7 @@
 from datetime import timedelta
 
 from ocf_datapipes.select import (
-    DropGSP,
+    SelectGSPIDs,
     LocationPicker,
     SelectLiveT0Time,
     SelectLiveTimeSlice,
@@ -43,7 +43,7 @@ def test_metnet_production(
     #####################################
     # Normalize GSP and PV on whole dataset here
     pv_datapipe = passiv_datapipe
-    gsp_datapipe, gsp_loc_datapipe = DropGSP(gsp_datapipe, gsps_to_keep=[0]).fork(2)
+    gsp_datapipe, gsp_loc_datapipe = SelectGSPIDs(gsp_datapipe, gsps_to_keep=[0]).fork(2)
     gsp_datapipe = Normalize(gsp_datapipe, normalize_fn=lambda x: x / x.installedcapacity_mwp)
     topo_datapipe = ReprojectTopography(topo_datapipe)
     sat_hrv_datapipe = AddT0IdxAndSamplePeriodDuration(