typo

Author: kumiori

src/irrevolutions/utils/viz.py

@@ -2,7 +2,8 @@
 from dolfinx.plot import vtk_mesh as compute_topology
 import matplotlib.tri as tri
 import matplotlib.pyplot as plt
-from pyvista.plotting.utilities import xvfbimport pyvista
+from pyvista.plotting.utilities import xvfb
+import pyvista
 from mpi4py import MPI
 import logging
 import sys
@@ -24,6 +25,7 @@
 # Start Xvfb for PyVista (for offscreen rendering)
 # xvfb.start_xvfb(wait=0.05)
 
+
 def plot_vector(u, plotter, subplot=None, scale=1.0):
     """
     Plots a vector field using PyVista with glyph representation.
@@ -64,6 +66,7 @@ def plot_vector(u, plotter, subplot=None, scale=1.0):
     plotter.set_background("white")
     return plotter
 
+
 def plot_scalar(u, plotter, subplot=None, lineproperties={}):
     """
     Plots a scalar field using PyVista.
@@ -98,7 +101,17 @@ def plot_scalar(u, plotter, subplot=None, lineproperties={}):
     plotter.set_background("white")
     return plotter
 
-def plot_profile(u, points, plotter, subplot=None, lineproperties={}, fig=None, ax=None, subplotnumber=1):
+
+def plot_profile(
+    u,
+    points,
+    plotter,
+    subplot=None,
+    lineproperties={},
+    fig=None,
+    ax=None,
+    subplotnumber=1,
+):
     """
     Plots a profile of the solution at given points.
 
@@ -131,6 +144,7 @@ def plot_profile(u, points, plotter, subplot=None, lineproperties={}, fig=None,
     plt.legend()
     return plt, (points_on_proc[:, 0], u_values)
 
+
 def plot_mesh(mesh, ax=None):
     """
     Plots the mesh using Matplotlib.
@@ -151,6 +165,7 @@ def plot_mesh(mesh, ax=None):
     ax.triplot(tria, color="k")
     return ax
 
+
 def get_datapoints(u, points):
     """
     Retrieves data points for a given solution field evaluated at specific points.
@@ -182,6 +197,7 @@ def get_datapoints(u, points):
 
     return points_on_proc, u_values
 
+
 def plot_perturbations(comm, Lx, prefix, β, v, bifurcation, stability, i_t):
     """
     Plots the perturbations using PyVista.
@@ -249,6 +265,7 @@ def plot_perturbations(comm, Lx, prefix, β, v, bifurcation, stability, i_t):
 
     return plotter
 
+
 def plot_matrix(M):
     """
     Plots the matrix as a heatmap using Matplotlib.
@@ -269,4 +286,4 @@ def plot_matrix(M):
             c = _M[j, i]
             ax.text(i, j, f"{c:.3f}", va="center", ha="center")
 
-    return fig
+    return fig