Fix missing denominator column in ALR

Previously, if denominator column was not in selected columns, it was removed before the transformation.
GispoCoding · Jan 7, 2025 · 0026856 · 0026856
1 parent 1bf85a9
commit 0026856
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/eis_toolkit/transformations/coda/alr.py b/eis_toolkit/transformations/coda/alr.py
@@ -62,7 +62,12 @@ def alr_transform(
         if invalid_columns:
             raise InvalidColumnException(f"The following columns were not found in the dataframe: {invalid_columns}.")
         columns_to_transform = columns
-        df = df[columns_to_transform]
+
+        if denominator_column not in columns_to_transform:
+            df = df[columns_to_transform + [denominator_column]]
+        else:
+            df = df[columns_to_transform]
+
     else:
         columns_to_transform = df.columns.to_list()