refactor: fix identifiers

gdrplatform · Aug 21, 2024 · 874ffff · 874ffff
1 parent 3ca6a64
commit 874ffff
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/R/data_type.R b/R/data_type.R
@@ -406,8 +406,14 @@ process_perturbations <- function(dt,
           new_column_name <- unique_drugs[1]
           dt[, (new_column_name) := dt[[conc_col]]]
 
+          drug_order <- gsub(".*\\_", "", drug_col)
+          drug_cols <- 
+            get_relevant_ids(paste0(
+              c("drug", "drug_name", "drug_moa", "concentration"), drug_order),
+              dt)
+
           # Remove the current drug and concentration columns
-          dt[, (c(drug_col, conc_col)) := NULL]
+          dt[, (drug_cols) := NULL]
         }
       }
     }