one columns for final visual

ubermag · Dec 16, 2024 · 42678b9 · 42678b9
1 parent c17eba8
commit 42678b9
Showing 1 changed file with 16 additions and 9 deletions.
diff --git a/micromagneticdata/mode_trajectory.py b/micromagneticdata/mode_trajectory.py
@@ -73,15 +73,22 @@ def calculate_fields(self):
         ]
 
     def final_visual(self):
-        return self.get_top_layer() * self.get_top_core_positions(
-            ["first_skyrmion_top_region", "second_skyrmion_top_region"]
-        ) * self.get_top_trajectory(
-            ["first_skyrmion_top_region", "second_skyrmion_top_region"]
-        ) + self.get_bottom_layer() * self.get_bottom_core_positions(
-            ["first_skyrmion_bottom_region", "second_skyrmion_bottom_region"]
-        ) * self.get_bottom_trajectory(
-            ["first_skyrmion_bottom_region", "second_skyrmion_bottom_region"]
-        )
+        return (
+            self.get_top_layer()
+            * self.get_top_core_positions(
+                ["first_skyrmion_top_region", "second_skyrmion_top_region"]
+            )
+            * self.get_top_trajectory(
+                ["first_skyrmion_top_region", "second_skyrmion_top_region"]
+            )
+            + self.get_bottom_layer()
+            * self.get_bottom_core_positions(
+                ["first_skyrmion_bottom_region", "second_skyrmion_bottom_region"]
+            )
+            * self.get_bottom_trajectory(
+                ["first_skyrmion_bottom_region", "second_skyrmion_bottom_region"]
+            )
+        ).cols(1)
 
     def get_layers(self):
         return self.get_top_layer() + self.get_bottom_layer()