fix test

megnet/data/molecule.py

@@ -357,7 +357,7 @@ def create_bond_feature(mol, bid: int, eid: int) -> Dict:
         return {"a_idx": bid,
                 "b_idx": eid,
                 "bond_type": 0,
-                "same_ring": same_ring,
+                "same_ring": bool(same_ring),
                 "spatial_distance": a1.GetDistance(a2)}
 
     def get_pair_feature(self, mol, bid: int,
@@ -390,7 +390,7 @@ def get_pair_feature(self, mol, bid: int,
         return {"a_idx": bid,
                 "b_idx": eid,
                 "bond_type": 4 if bond.IsAromatic() else bond.GetBondOrder(),
-                "same_ring": same_ring,
+                "same_ring": bool(same_ring),
                 "spatial_distance": a1.GetDistance(a2)}
 
     @staticmethod

pylintrc

@@ -7,7 +7,7 @@ extension-pkg-whitelist=numpy, rdkit, pybel, openbabel
 
 # Add files or directories to the blacklist. They should be base names, not
 # paths.
-ignore=R0201
+ignore=
 
 # Add files or directories matching the regex patterns to the blacklist. The
 # regex matches against base names, not paths.
@@ -159,7 +159,8 @@ disable=print-statement,
         R0901,
         C0415,
         C0330,
-        C0302
+        C0302,
+        R0201
 
 # Enable the message, report, category or checker with the given id(s). You can
 # either give multiple identifier separated by comma (,) or put this option