Commit 5314ea28 authored by Aneesh Pappu's avatar Aneesh Pappu
Browse files

adding robustness

parent 357fc5f5

deepchem/splits/tests/test_splitter.py

+1 −0
Original line number Diff line number Diff line
@@ -103,6 +103,7 @@ class TestSplitters(TestDatasetAPI): 
            column = y_df[col]

            NaN_count = column.isnull().sum()

            if NaN_count == totalRows:

                print("fail -- one column doesn't have results")

                assert NaN_count != totalRows

        print("end of stratified test")

        assert 1 == 1

CRA Git | Maintained and supported by SUSTech CRA and CCSE