diff --git a/reVX/setbacks/base.py b/reVX/setbacks/base.py index d85e187f5..8001c8a04 100644 --- a/reVX/setbacks/base.py +++ b/reVX/setbacks/base.py @@ -275,8 +275,8 @@ def _parse_regulations(self, regulations_fpath): logger.error(msg) raise RuntimeError(msg) - for non_nan_col in ["Feature Type", "Value Type", "Value", "FIPS"]: - regulations = regulations[~regulations[non_nan_col].isna()] + for col in ["Feature Type", "Value Type", "Value", "FIPS"]: + regulations = regulations[~regulations[col].isna()] feature_types = regulations['Feature Type'].str.strip().str.lower() regulations['Feature Type'] = feature_types