Skip to content

Commit 31957e5

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent f4d43ba commit 31957e5

File tree

1 file changed

+13
-6
lines changed

1 file changed

+13
-6
lines changed

crystal_toolkit/renderables/migrationgraph.py

Lines changed: 13 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
1-
from turtle import pos
2-
import numpy as np
31

42
from pymatgen.analysis.diffusion.neb.full_path_mapper import MigrationGraph
5-
from crystal_toolkit.core.scene import Scene, Cylinders
3+
4+
from crystal_toolkit.core.scene import Cylinders, Scene
5+
66

77
def get_migrationgraph_scene(
88
self,
@@ -19,13 +19,20 @@ def get_migrationgraph_scene(
1919
result_scene = self.structure.get_scene()
2020
hop_contents = []
2121
for one_hop in self.unique_hops.values():
22-
hop_cyl = Cylinders(positionPairs=[[list(one_hop["ipos_cart"]), list(one_hop["epos_cart"])]], radius=0.2, visible=True)
22+
hop_cyl = Cylinders(
23+
positionPairs=[[list(one_hop["ipos_cart"]), list(one_hop["epos_cart"])]],
24+
radius=0.2,
25+
visible=True,
26+
)
2327
hop_contents.append(hop_cyl)
2428

2529
result_scene.contents.append(
26-
Scene(name="hops", origin=result_scene.contents[0].origin, contents=hop_contents)
30+
Scene(
31+
name="hops", origin=result_scene.contents[0].origin, contents=hop_contents
32+
)
2733
)
2834
print(result_scene.contents[-1])
2935
return result_scene
3036

31-
MigrationGraph.get_scene = get_migrationgraph_scene
37+
38+
MigrationGraph.get_scene = get_migrationgraph_scene

0 commit comments

Comments
 (0)