ckp: fix flake8 assertions

devitocodes · Apr 18, 2022 · 90c0b7f · 90c0b7f
1 parent ac9e40a
commit 90c0b7f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/pyrevolve/pyrevolve.py b/pyrevolve/pyrevolve.py
@@ -187,7 +187,7 @@ def op_capo(self):
 
     @property
     def next_op_capo(self):
-        _op_capo = (self.scheduler.capo +1) * self.block_size
+        _op_capo = (self.scheduler.capo + 1) * self.block_size
         return _op_capo if _op_capo < self.op_timesteps else self.op_timesteps
 
     def apply_forward(self):
@@ -455,7 +455,8 @@ def reload_scheduler(self, uf=1, ub=1, up=1):
             self.up = up
             self.arch = Architecture(self.storage_list)
             self.scheduler = HRevolve(
-                self.n_checkpoints, self.cp_timesteps, self.arch, self.uf, self.ub, self.up
+                self.n_checkpoints, self.cp_timesteps, self.arch, self.uf, self.ub,
+                self.up
             )
         else:
             raise ValueError(

diff --git a/tests/test_multilevel.py b/tests/test_multilevel.py
@@ -30,7 +30,7 @@ def test_forward_nt(nt, mwd, mrd, dwd, drd, uf, ub, singlefile, block_size):
     )
     st_list = [npStorage, dkStorage]
     rev = MultiLevelRevolver(cp, f, b, nt, storage_list=st_list, uf=uf, ub=ub,
-        block_size=block_size)
+                             block_size=block_size)
     assert f.counter == 0
     rev.apply_forward()
     assert f.counter == nt
@@ -60,7 +60,7 @@ def test_reverse_nt(nt, mwd, mrd, dwd, drd, uf, ub, singlefile, block_size):
     )
     st_list = [npStorage, dkStorage]
     rev = MultiLevelRevolver(cp, f, b, nt, storage_list=st_list, uf=uf, ub=ub,
-        block_size=block_size)
+                             block_size=block_size)
 
     rev.apply_forward()
     assert f.counter == nt
@@ -89,7 +89,7 @@ def test_num_loads_and_saves(nt, mwd, mrd, dwd, drd, uf, ub, singlefile, block_s
     )
     st_list = [npStorage, dkStorage]
     rev = MultiLevelRevolver(cp, f, b, nt, storage_list=st_list, uf=uf, ub=ub,
-        block_size=block_size)
+                             block_size=block_size)
 
     rev.apply_forward()
     assert cp.load_counter == 0