Skip to content

Commit 0aa9f23

Browse files
committed
renamed variable type to drug_descriptor_type
1 parent a6a85f8 commit 0aa9f23

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

coderdata/dataset/dataset.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -602,14 +602,16 @@ def format(
602602
raise ValueError(
603603
f"'shape' has to be one of '{legal_shapes}'"
604604
)
605-
type = kwargs.get('type', None)
605+
drug_descriptor_type = kwargs.get('drug_descriptor_type', None)
606606

607-
if type is None:
607+
if drug_descriptor_type is None:
608608
tmp = data.drug_descriptors
609609
else:
610610
# TODO: potentially allow for list of columns to retain
611611
tmp = data.drug_descriptors[
612-
data.drug_descriptors['structural_descriptor'] == type
612+
data.drug_descriptors[
613+
'structural_descriptor'
614+
] == drug_descriptor_type
613615
]
614616
if shape == 'long':
615617
ret = tmp

0 commit comments

Comments
 (0)