fix the variable name

segasai · Nov 5, 2024 · 8d63a1d · 8d63a1d
1 parent 471e9c2
commit 8d63a1d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/py/rvspecfit/nn/NNInterpolator.py b/py/rvspecfit/nn/NNInterpolator.py
@@ -11,14 +11,14 @@ def __init__(self,
                  width=None,
                  npc=None,
                  npix=None,
-                 widthbn=True):
+                 withbn=True):
         super(NNInterpolator, self).__init__()
         self.indim = indim
         self.nlayers = nlayers
         self.width = width
         self.npc = npc
         self.npix = npix
-        self.widthbn = widthbn
+        self.withbn = withbn
         self.initLayers()
 
     def initLayers(self):
@@ -51,7 +51,7 @@ def initLayers(self):
         batchnorm_after_nl = True
 
         for i in range(len(shapes)):
-            withbn = self.widthbn
+            withbn = self.withbn
             if i == 0 or i == len(shapes) - 1:
                 withbn = False
             if batchnorm_after_nl: