Update pyrazine.py

binggu56 · Mar 9, 2022 · f9e42e7 · f9e42e7
1 parent 3e8cf36
commit f9e42e7
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/lime/models/pyrazine.py b/lime/models/pyrazine.py
@@ -299,6 +299,8 @@ def plot3d():
 
     #data = [go.Surface(z=apes)]
     #fig = go.Figure(data = data)
+    import matplotlib.pyplot as plt 
+
     fig = plt.figure(figsize=(5,4))
     set_style(fontsize=14)
 
@@ -419,7 +421,7 @@ def mayavi(surfaces):
     rcParams['xtick.major.pad']='2'
     rcParams['ytick.major.pad']='2'
 
-    #mayavi()
-    contour()
+    # mayavi()
+    # contour()
     #cut()
-    #plot3d()
+    plot3d()