Linting

sedos-project · Jul 18, 2024 · 91b6efd · 91b6efd
1 parent 6e07cfc
commit 91b6efd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/data_adapter_oemof/calculations.py b/data_adapter_oemof/calculations.py
@@ -309,7 +309,9 @@ def find_and_replace_irrelevant_data(group_df: pd.DataFrame) -> pd.DataFrame:
             fill_indices += group_df[invest_col[0]] == 0
 
         # Fill indices
-        group_df.loc[fill_indices] = group_df.fillna(group_df.mean(numeric_only=True))
+        group_df.loc[fill_indices] = group_df.fillna(
+            group_df.mean(numeric_only=True)
+        ).loc[fill_indices]
 
         return group_df