bugfix

stefanch · stefanch · commit c591909a3edc · 2022-08-15T20:12:12.000+02:00
diff --git a/sgdml/__init__.py b/sgdml/__init__.py
@@ -22,7 +22,7 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
-__version__ = '0.5.2.dev1'
+__version__ = '0.5.2.dev2'
 
 MAX_PRINT_WIDTH = 100
 LOG_LEVELNAME_WIDTH = 7  # do not modify
diff --git a/sgdml/torchtools.py b/sgdml/torchtools.py
@@ -364,7 +364,7 @@ def forward(self, J_indx):
                             )
 
                         elif _n_perm_batches < self.n_perms:
-                            self.n_perm_batches = _next_batch_size(
+                            _n_perm_batches = _next_batch_size(
                                 self.n_perms, _n_perm_batches
                             )
 

Original file line number	Diff line number	Diff line change
`@@ -364,7 +364,7 @@ def forward(self, J_indx):`
`364`	`364`	`)`
`365`	`365`
`366`	`366`	`elif _n_perm_batches < self.n_perms:`
`367`		`- self.n_perm_batches = _next_batch_size(`
	`367`	`+ _n_perm_batches = _next_batch_size(`
`368`	`368`	`self.n_perms, _n_perm_batches`
`369`	`369`	`)`
`370`	`370`