black

Renumics · May 24, 2024 · 662f8ea · 662f8ea
1 parent 91d9ec9
commit 662f8ea
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 10 deletions.
diff --git a/mesh2vec/mesh2vec_cae.py b/mesh2vec/mesh2vec_cae.py
@@ -254,7 +254,9 @@ def from_d3plot_shell(
         return Mesh2VecCae(distance, mesh, element_info, calc_strategy=calc_strategy)
 
     @staticmethod
-    def from_keyfile_shell(distance: int, keyfile: Path, partid="", calc_strategy="bfs") -> "Mesh2VecCae":
+    def from_keyfile_shell(
+        distance: int, keyfile: Path, partid="", calc_strategy="bfs"
+    ) -> "Mesh2VecCae":
         """
         Read the given keyfile and use the shell elements to generate a hypergraph, using mesh
         nodes as hyperedges, and adjacent elements as hypervertices.

diff --git a/mesh2vec/mesh_features.py b/mesh2vec/mesh_features.py
@@ -265,34 +265,45 @@ def parse_contents(file_contents):
             for line in lines:
                 if line.startswith("*"):
                     current_section = line.split()[0].upper()
-                    current_section_options = set(current_section.split('_')[1:])
+                    current_section_options = set(current_section.split("_")[1:])
                     current_section_lines_per_entry = 1
                     current_section_lineno = 0
                     continue
                 if line.startswith("$"):  # comment
                     continue
                 if current_section == "*NODE":
                     try:
-                        point_coordinates.append([float(line[8+i*16:8+(i+1)*16]) for i in range(3)])
+                        point_coordinates.append(
+                            [float(line[8 + i * 16 : 8 + (i + 1) * 16]) for i in range(3)]
+                        )
                         pnt_ids.append(line[:8].strip())
                     except:
                         pass
                 elif current_section.startswith("*ELEMENT_SHELL"):
-
 
                     if current_section_lineno % current_section_lines_per_entry == 0:
                         if partid == "" or partid == line[8:16].strip():
-                            node_ids = [line[16+i*8:16+(i+1)*8].strip() for i in range(8)]
-                            node_ids = [node_id for node_id in node_ids if len(node_id) > 0 and node_id != "0"]
+                            node_ids = [
+                                line[16 + i * 8 : 16 + (i + 1) * 8].strip() for i in range(8)
+                            ]
+                            node_ids = [
+                                node_id
+                                for node_id in node_ids
+                                if len(node_id) > 0 and node_id != "0"
+                            ]
                             # TODO: Check for unhandled options, e.g. COMPOSITE, DOF
                             if current_section_lineno == 0:
                                 if len(current_section_options & thickcard_options_set) > 0:
-                                    current_section_lines_per_entry += 1 # skip thickness card
+                                    current_section_lines_per_entry += 1  # skip thickness card
                                     if len(node_ids) > 4:
-                                        current_section_lines_per_entry += 1 # skip additional thickness card for mid-side nodes
+                                        current_section_lines_per_entry += (
+                                            1  # skip additional thickness card for mid-side nodes
+                                        )
                                 if "OFFSET" in current_section_options:
-                                        current_section_lines_per_entry += 1 # skip offset card
-                            elem_node_ids.append([node_id for node_id in node_ids if len(node_id) > 0])
+                                    current_section_lines_per_entry += 1  # skip offset card
+                            elem_node_ids.append(
+                                [node_id for node_id in node_ids if len(node_id) > 0]
+                            )
                             if node_ids[0] == 1.0:
                                 print("HERE")
                             elem_ids.append(line[:8].strip())