diff --git a/hydragnn/utils/abstractrawdataset.py b/hydragnn/utils/abstractrawdataset.py
index f5946ba73..3928bee0a 100644
--- a/hydragnn/utils/abstractrawdataset.py
+++ b/hydragnn/utils/abstractrawdataset.py
@@ -373,9 +373,8 @@ def __build_edge(self):
 
         # edge lengths already added manually if using PBC.
         # if spherical coordinates or pair point is set up, then skip directly to edge_transformation
-        if (
-            (not self.periodic_boundary_conditions)
-            and (not hasattr(self, self.edge_feature_transform))
+        if (not self.periodic_boundary_conditions) and (
+            not hasattr(self, self.edge_feature_transform)
         ):
             self.dataset[:] = [compute_edge_lengths(data) for data in self.dataset]