From e971015af45190d78ce7e10a56be15c907515d06 Mon Sep 17 00:00:00 2001 From: Peng Zheng Date: Fri, 5 Jan 2024 14:01:39 -0800 Subject: [PATCH] 0.0.5 (#6) * fix the bug in the pre_selected_covs processing * update version to 0.0.5 * pine dependency on mrtool to 0.1.4 --- pyproject.toml | 4 ++-- src/bopforge/continuous_pipeline/__main__.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 71e1042..e5f6a4c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta" [project] name = "bopforge" -version = "0.0.4" +version = "0.0.5" description = "Pipelines for Burden of Proof (BoP) analyses" readme = "REDME.md" requires-python = ">=3.10" @@ -12,7 +12,7 @@ license = { file = "LICENSE" } authors = [ { name = "IHME Math Sciences", email = "ihme.math.sciences@gmail.com" }, ] -dependencies = ["numpy", "scipy", "pandas", "matplotlib", "mrtool", "pplkit"] +dependencies = ["numpy", "scipy", "pandas", "matplotlib", "mrtool==0.1.4", "pplkit"] [project.optional-dependencies] test = ["pytest"] diff --git a/src/bopforge/continuous_pipeline/__main__.py b/src/bopforge/continuous_pipeline/__main__.py index 7963f57..739a248 100644 --- a/src/bopforge/continuous_pipeline/__main__.py +++ b/src/bopforge/continuous_pipeline/__main__.py @@ -26,8 +26,8 @@ def pre_processing(dataif: DataInterface) -> None: df.drop(columns=covs_to_remove, inplace=True) # remove from settings - all_covs = set(all_covs) - covs_to_remove = set(covs_to_remove) + all_covs = set("em" + col[3:] for col in all_covs) + covs_to_remove = set("em" + col[3:] for col in covs_to_remove) pre_selected_covs = set(settings["pre_selected_covs"]) pre_selected_covs = pre_selected_covs & all_covs pre_selected_covs = pre_selected_covs - covs_to_remove