Fixed GeospatialScenario running geospatial jobs with auto_templating

NREL · Nov 8, 2024 · 9a087a8 · 9a087a8
1 parent c7a39c4
commit 9a087a8
Show file tree

Hide file tree

Showing 3 changed files with 848 additions and 4,350 deletions.
diff --git a/pvdeg/scenario.py b/pvdeg/scenario.py
@@ -1092,6 +1092,8 @@ def __init__(
         geospatial=False,
         weather_data: xr.Dataset = None,
         meta_data: pd.DataFrame = None,
+        func: Callable = None,
+        template: xr.Dataset = None,
     ):
         super().__init__(
             name=name,
@@ -1106,6 +1108,8 @@ def __init__(
         )
         self.geospatial = geospatial
         self.hpc = hpc
+        self.func = func
+        self.template = template
 
     def __eq__(self, other):
         raise NotImplementedError("""
@@ -1658,7 +1662,7 @@ def addJob(
         if template is None:
 
             # take the weather datapoints specified by metadata and create a template based on them.
-            geo_weather_sub = self.weather_data.sel(gid=self.meta_data.index)
+            self.weather_data = self.weather_data.sel(gid=self.meta_data.index)
             template = pvdeg.geospatial.auto_template(func=func, ds_gids=self.weather_data)
 
         self.template = template
@@ -2184,4 +2188,4 @@ def format_geo_weather(self):
             </div>
             """
 
-        return weather_data_html
+        return weather_data_html