formatting

ORNL · Aug 25, 2023 · 9c05b9c · 9c05b9c
1 parent 1aa4695
commit 9c05b9c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hydragnn/utils/abstractrawdataset.py b/hydragnn/utils/abstractrawdataset.py
@@ -374,7 +374,7 @@ 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, 'edge_feature_transform')
+            not hasattr(self, "edge_feature_transform")
         ):
             self.dataset[:] = [compute_edge_lengths(data) for data in self.dataset]
 
@@ -397,7 +397,7 @@ def __build_edge(self):
                 data.edge_attr = data.edge_attr / max_edge_length
 
         # Descriptors about topology of the local environment
-        elif hasattr(self, 'edge_feature_transform'):
+        elif hasattr(self, "edge_feature_transform"):
             self.dataset[:] = [
                 self.edge_feature_transform(data) for data in self.dataset
             ]