run both black and ruff

Author: adsharma

src_py/query_result.py

@@ -272,7 +272,8 @@ def get_schema(self) -> dict[str, str]:
         return dict(
             zip(
                 self._query_result.getColumnNames(),
-                self._query_result.getColumnDataTypes(), strict=False,
+                self._query_result.getColumnDataTypes(),
+                strict=False,
             )
         )
 

src_py/torch_geometric_result_converter.py

@@ -135,15 +135,13 @@ def __extract_properties_from_node(
                 # Check if the shape of the property is consistent
                 if (
                     label in self.nodes_dict and prop_name in self.nodes_dict[label]
-                ):
-                    # If the shape is inconsistent, then mark the property as unconverted
-                    if curr_value.shape != self.nodes_dict[label][prop_name][0].shape:
-                        self.warning_messages.add(
-                            f"Property {label}.{prop_name} has an inconsistent shape. The property is marked as unconverted."
-                        )
-                        self.__mark_property_unconverted(label, prop_name)
-                        pos = self.__add_unconverted_property(node, label, prop_name)
-                        continue
+                ) and curr_value.shape != self.nodes_dict[label][prop_name][0].shape:
+                    self.warning_messages.add(
+                        f"Property {label}.{prop_name} has an inconsistent shape. The property is marked as unconverted."
+                    )
+                    self.__mark_property_unconverted(label, prop_name)
+                    pos = self.__add_unconverted_property(node, label, prop_name)
+                    continue
 
             # Create the dictionary for the label if it does not exist
             if label not in self.nodes_dict:

test/test_arrow.py

@@ -599,7 +599,8 @@ def _test_node(_conn: lb.Connection) -> None:
                 ground_truth.TINY_SNB_PERSONS_GROUND_TRUTH[8],
                 ground_truth.TINY_SNB_PERSONS_GROUND_TRUTH[9],
                 ground_truth.TINY_SNB_PERSONS_GROUND_TRUTH[10],
-            ], strict=False,
+            ],
+            strict=False,
         ):
             _test_node_helper(a, b)
 
@@ -620,7 +621,8 @@ def _test_node_rel(_conn: lb.Connection) -> None:
                 ground_truth.TINY_SNB_PERSONS_GROUND_TRUTH[3],
                 ground_truth.TINY_SNB_PERSONS_GROUND_TRUTH[5],
                 ground_truth.TINY_SNB_PERSONS_GROUND_TRUTH[7],
-            ], strict=False,
+            ],
+            strict=False,
         ):
             _test_node_helper(a, b)
         for a, b in zip(
@@ -653,7 +655,8 @@ def _test_node_rel(_conn: lb.Connection) -> None:
                     "rating": 9.2,
                     "year": 2015,
                 },
-            ], strict=False,
+            ],
+            strict=False,
         ):
             _test_node_helper(a, b)
 
@@ -663,7 +666,8 @@ def _test_node_rel(_conn: lb.Connection) -> None:
                 ground_truth.TINY_SNB_ORGANISATIONS_GROUND_TRUTH[4],
                 ground_truth.TINY_SNB_ORGANISATIONS_GROUND_TRUTH[6],
                 ground_truth.TINY_SNB_ORGANISATIONS_GROUND_TRUTH[6],
-            ], strict=False,
+            ],
+            strict=False,
         ):
             _test_node_helper(a, b)
 

test/type_aliases.py

@@ -1,4 +1,3 @@
-
 from typing import TypeAlias
 
 from real_ladybug import Connection, Database