Skip to content

Commit f1f46d0

Browse files
committed
asserting in input, causes silent bugs otherwise
1 parent 68f2aa8 commit f1f46d0

1 file changed

Lines changed: 9 additions & 0 deletions

File tree

drevalpy/models/drp_model.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -213,6 +213,15 @@ def get_concatenated_features(
213213
cell_line_input=cell_line_input,
214214
drug_input=drug_input,
215215
)
216+
if drug_view is not None:
217+
if drug_view not in inputs:
218+
raise ValueError(f"Expected drug_view '{drug_view}' to be in inputs, but it was not. Inputs: {inputs}")
219+
if cell_line_view is not None:
220+
if cell_line_view not in inputs:
221+
raise ValueError(
222+
f"Expected cell_line_view '{cell_line_view}' to be in inputs, but it was not. Inputs: {inputs}"
223+
)
224+
216225
cell_line_features = None if cell_line_view is None else inputs.get(cell_line_view)
217226
drug_features = None if drug_view is None else inputs.get(drug_view)
218227

0 commit comments

Comments
 (0)