Skip to content

Commit 02244cb

Browse files
committed
fix(CalcRiskPeriod): fixes problem with undefined group_id
1 parent 4921528 commit 02244cb

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

climada/trajectories/riskperiod.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -134,8 +134,16 @@ def __init__(
134134
self.calc_residual = calc_residual
135135
self.measure = None # Only possible to set with apply_measure to make sure snapshots are consistent
136136

137-
self._group_id_E0 = self.snapshot0.exposure.gdf["group_id"].values
138-
self._group_id_E1 = self.snapshot1.exposure.gdf["group_id"].values
137+
self._group_id_E0 = (
138+
self.snapshot0.exposure.gdf["group_id"].values
139+
if "group_id" in self.snapshot0.exposure.gdf.columns
140+
else np.array([])
141+
)
142+
self._group_id_E1 = (
143+
self.snapshot1.exposure.gdf["group_id"].values
144+
if "group_id" in self.snapshot1.exposure.gdf.columns
145+
else np.array([])
146+
)
139147

140148
def _reset_impact_data(self):
141149
self._impacts_arrays = None

0 commit comments

Comments
 (0)