catching use of pyedt

PMEAL · Aug 7, 2024 · aa7b9c0 · aa7b9c0
1 parent 813625a
commit aa7b9c0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/test/integration/test_drainage.py b/test/integration/test_drainage.py
@@ -1,7 +1,10 @@
 import numpy as np
 import porespy as ps
 import matplotlib.pyplot as plt
-from pyedt import edt
+try:
+    from pyedt import edt
+except ModuleNotFoundError:
+    from edt import edt
 
 
 def test_drainage():