fix: handle new/ removed variables (#26)

lkstrp · web-flow · commit 9fcf1c5aad54 · 2024-11-07T16:21:37.000+01:00
diff --git a/src/comment_components.py b/src/comment_components.py
@@ -211,7 +211,9 @@ def variables_deviation_df(self) -> pd.DataFrame:
             :, [col for col in vars2.columns if str(col).replace(".", "", 1).isdigit()]
         ]
 
-        assert vars1.index.equals(vars2.index)
+        # Filter any new/ removed variables        
+        vars1 = vars1[vars1.index.isin(vars2.index)]
+        vars2 = vars2[vars2.index.isin(vars1.index)]
 
         deviation_df = self.get_deviation_df(
             vars1, vars2, nrmse_normalization_method=self.NRMSE_NORMALIZATION_METHOD

Original file line number	Diff line number	Diff line change
`@@ -211,7 +211,9 @@ def variables_deviation_df(self) -> pd.DataFrame:`
`211`	`211`	`:, [col for col in vars2.columns if str(col).replace(".", "", 1).isdigit()]`
`212`	`212`	`]`
`213`	`213`
`214`		`- assert vars1.index.equals(vars2.index)`
	`214`	`+ # Filter any new/ removed variables`
	`215`	`+ vars1 = vars1[vars1.index.isin(vars2.index)]`
	`216`	`+ vars2 = vars2[vars2.index.isin(vars1.index)]`
`215`	`217`
`216`	`218`	`deviation_df = self.get_deviation_df(`
`217`	`219`	`vars1, vars2, nrmse_normalization_method=self.NRMSE_NORMALIZATION_METHOD`