forgot to copy drug features here

JudithBernett · JudithBernett · commit 24084d6fe6a2 · 2025-06-23T10:50:01.000+02:00
diff --git a/drevalpy/experiment.py b/drevalpy/experiment.py
@@ -993,6 +993,7 @@ def train_and_predict(
         )
 
     if len(prediction_dataset) > 0:
+        drug_input = drug_features.copy() if drug_features is not None else None
         prediction_dataset._predictions = model.predict(
             cell_line_ids=prediction_dataset.cell_line_ids,
             drug_ids=prediction_dataset.drug_ids,

Original file line number	Diff line number	Diff line change
`@@ -993,6 +993,7 @@ def train_and_predict(`
`993`	`993`	`)`
`994`	`994`
`995`	`995`	`if len(prediction_dataset) > 0:`
	`996`	`+ drug_input = drug_features.copy() if drug_features is not None else None`
`996`	`997`	`prediction_dataset._predictions = model.predict(`
`997`	`998`	`cell_line_ids=prediction_dataset.cell_line_ids,`
`998`	`999`	`drug_ids=prediction_dataset.drug_ids,`