remove plot load from slaterjastrow

NLESC-JCER · Oct 5, 2021 · 31fb635 · 31fb635
1 parent d762715
commit 31fb635
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qmctorch/wavefunction/slater_jastrow.py b/qmctorch/wavefunction/slater_jastrow.py
@@ -3,7 +3,6 @@
 import torch
 from scipy.optimize import curve_fit
 from copy import deepcopy
-import matplotlib.pyplot as plt
 import numpy as np
 from torch import nn
 import operator
@@ -656,6 +655,7 @@ def sto(x, norm, alpha):
 
             # plot if necessary
             if plot:
+                import matplotlib.pyplot as plt
                 plt.plot(xdata, ydata)
                 plt.plot(xdata, sto(xdata, *popt))
                 plt.show()