Update allowed renewable parameter values

openclimatefix · Dec 19, 2024 · 8c62a54 · 8c62a54
1 parent b7923f1
commit 8c62a54
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pvnet/data/datamodule.py b/pvnet/data/datamodule.py
@@ -93,14 +93,14 @@ def __init__(
         )
 
     def _get_streamed_samples_dataset(self, start_time, end_time) -> Dataset:
-        if self.configuration.renewable == "pv":
+        if self.configuration.renewable == "uk_pv":
             return PVNetUKRegionalDataset(
                 self.configuration, start_time=start_time, end_time=end_time
             )
-        elif self.configuration.renewable in ["wind", "pv_india", "pv_site"]:
+        elif self.configuration.renewable == "site":
             return SitesDataset(self.configuration, start_time=start_time, end_time=end_time)
         else:
-            raise ValueError(f"Unknown renewable: {self.configuration.renewable}")
+            raise ValueError(f"Unknown renewable: {self.configuration.renewable}, renewable value should either be uk_pv or site")
 
     def _get_premade_samples_dataset(self, subdir) -> Dataset:
         split_dir = f"{self.sample_dir}/{subdir}"