We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
type
drug_descriptor_type
1 parent a6a85f8 commit 0aa9f23Copy full SHA for 0aa9f23
1 file changed
coderdata/dataset/dataset.py
@@ -602,14 +602,16 @@ def format(
602
raise ValueError(
603
f"'shape' has to be one of '{legal_shapes}'"
604
)
605
- type = kwargs.get('type', None)
+ drug_descriptor_type = kwargs.get('drug_descriptor_type', None)
606
607
- if type is None:
+ if drug_descriptor_type is None:
608
tmp = data.drug_descriptors
609
else:
610
# TODO: potentially allow for list of columns to retain
611
tmp = data.drug_descriptors[
612
- data.drug_descriptors['structural_descriptor'] == type
+ data.drug_descriptors[
613
+ 'structural_descriptor'
614
+ ] == drug_descriptor_type
615
]
616
if shape == 'long':
617
ret = tmp
0 commit comments