From aadd720f1e309191e95bd6a52cedaf67b1b48f61 Mon Sep 17 00:00:00 2001
From: Anastasiia Kutakh <anastasiiakutakh@C02FK59HMD6M.tld>
Date: Fri, 20 Aug 2021 23:46:24 -0400
Subject: [PATCH] .

---
 CDTools/models/base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/CDTools/models/base.py b/CDTools/models/base.py
index 48a33d4..9d8e9eb 100644
--- a/CDTools/models/base.py
+++ b/CDTools/models/base.py
@@ -456,7 +456,7 @@ class CDIModel(t.nn.Module):
         def saveobj(a, filename):
             a = np.abs(a)
             plt.imshow(a)
-            plt.savefig('filename')
+            plt.savefig(filename)
         f = ['base_a.png', 'base_b.png', 'base_c.png', 'base_d.png']
         comp = [a[i, j, :, :] for i, j in zip([0, 0, 1, 1], [0, 1, 0, 1])]
         for i in range(4):