Skip to content

Commit

Permalink
mend
Browse files Browse the repository at this point in the history
  • Loading branch information
amva13 committed Mar 12, 2024
1 parent 49da597 commit 10d1fc0
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 7 deletions.
6 changes: 3 additions & 3 deletions tdc/test/dev_tests/utils_tests/test_misc_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,9 +31,9 @@ def test_neg_sample(self):

@unittest.skip("this is a visual test and should only be run locally")
def test_label_distribution(self):
from tdc.single_pred import ADME
data = ADME(name='Caco2_Wang')
x = data.label_distribution()
from tdc.single_pred import ADME
data = ADME(name='Caco2_Wang')
x = data.label_distribution()

def test_get_label_map(self):
from tdc.multi_pred import DDI
Expand Down
3 changes: 2 additions & 1 deletion tdc/test/test_resources.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,8 @@ def test_get_gene_metadata(self):
# assert isinstance(varpyarrow, SparseCOOTensor)

@unittest.skip(
"this test takes up too much mem for GH worker but works locally.. skip for now")
"this test takes up too much mem for GH worker but works locally.. skip for now"
)
def test_get_measurement_matrix(self):
Xslice = self.resource.get_measurement_matrix(upper=5,
lower=0,
Expand Down
6 changes: 3 additions & 3 deletions tdc/utils/label.py
Original file line number Diff line number Diff line change
Expand Up @@ -222,7 +222,7 @@ def NegSample(df, column_names, frac, two_types):
neg_list_val = []
for i in neg_list:
neg_list_val.append([i[0], id2seq[i[0]], i[1], id2seq[i[1]], 0])

concat_frames = pd.DataFrame(neg_list_val).rename(columns={
0: id1,
1: x1,
Expand Down Expand Up @@ -280,6 +280,6 @@ def NegSample(df, column_names, frac, two_types):
4: "Y"
})
],
ignore_index=True,
sort=False)
ignore_index=True,
sort=False)
return df

0 comments on commit 10d1fc0

Please sign in to comment.