diff --git a/megnet/data/molecule.py b/megnet/data/molecule.py index 818d677e0..9139e2dd2 100644 --- a/megnet/data/molecule.py +++ b/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 diff --git a/pylintrc b/pylintrc index f7d3c95b8..9316dfb31 100644 --- a/pylintrc +++ b/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