Commit 4a927322 authored by jansen31's avatar jansen31
Browse files

return handles for plots

parent aff72aa5
......@@ -143,8 +143,8 @@ class DegenerateOrbital(PlanarMolecule):
u2 = self.output['u2']
change = self.output['stats']['changes']
self.plot_guess(u1, name='Best approximation: physical constraint satisfied', show=False)
self.plot_guess(u2, name='Best approximation: Fourier constraint satisfied', show=False)
f1 = self.plot_guess(u1, name='Best approximation: physical constraint satisfied', show=False)
f2 = self.plot_guess(u2, name='Best approximation: Fourier constraint satisfied', show=False)
# f, ((ax1, ax2), (ax3, ax4)) = plt.subplots(2, 2, figsize=(9, 6))
# im = ax1.imshow(abs(u1), cmap='gray')
......@@ -188,7 +188,8 @@ class DegenerateOrbital(PlanarMolecule):
h.tight_layout()
plt.show()
self.output['plots'].append(f)
self.output['plots'].append(f1)
self.output['plots'].append(f2)
self.output['plots'].append(g)
self.output['plots'].append(h)
......@@ -207,3 +208,4 @@ class DegenerateOrbital(PlanarMolecule):
plt.tight_layout()
if show:
plt.show()
return fig
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment