fix chain id check in atomic edges #199

a-r-j · Jul 27, 2022 · 409e515 · 409e515
1 parent 95735aa
commit 409e515
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/graphein/protein/edges/atomic.py b/graphein/protein/edges/atomic.py
@@ -134,7 +134,7 @@ def add_atomic_edges(G: nx.Graph, tolerance: float = 0.56) -> nx.Graph:
             continue
 
         # Check atoms are in the same chain
-        if not (chain_1 and chain_2):
+        if not (chain_1 == chain_2):
             continue
 
         if G.has_edge(node_1, node_2):