updated relative import paths in diagnostics.py

ray-chew · May 20, 2024 · 1d815da · 1d815da
1 parent 82bc4f5
commit 1d815da
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/wrappers/diagnostics.py b/wrappers/diagnostics.py
@@ -3,8 +3,8 @@
 """
 
 import numpy as np
-from src import physics
-from vis import plotter
+from ..src import physics
+from ..vis import plotter
 from copy import deepcopy
 
 import matplotlib.pyplot as plt
@@ -357,3 +357,4 @@ def show(
                 plt.savefig(self.output_dir + fn + ".pdf", dpi=200, bbox_inches="tight")
 
             plt.show()
+