Merge branch 'main' of https://github.com/kenko911/m3gnet

materialsvirtuallab · Sep 27, 2024 · c038d44 · c038d44
2 parents 82a1ab0 + cdbf7a9
commit c038d44
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/m3gnet/graph/_batch.py b/m3gnet/graph/_batch.py
@@ -144,11 +144,13 @@ def _concatenate(list_of_arrays: List, name: AnyStr) -> Optional[np.ndarray]:
 
 
 @overload
-def assemble_material_graph(graphs: List[MaterialGraph]) -> MaterialGraph: ...
+def assemble_material_graph(graphs: List[MaterialGraph]) -> MaterialGraph:
+    ...
 
 
 @overload
-def assemble_material_graph(graphs: List[List]) -> List: ...
+def assemble_material_graph(graphs: List[List]) -> List:
+    ...
 
 
 def assemble_material_graph(graphs):