notebooks' review

firedrakeproject · Sep 7, 2023 · ea60a16 · ea60a16
1 parent b45fa28
commit ea60a16
Show file tree

Hide file tree

Showing 3 changed files with 421 additions and 143 deletions.
diff --git a/checkpoint_schedules/basic_schedules.py b/checkpoint_schedules/basic_schedules.py
@@ -125,7 +125,7 @@ def _iterator(self):
         while True:
             while self._r < self._max_n:
                 n1 = self._max_n - self._r
-                n0 = ((n1 - 1) // sys.maxsize) * sys.maxsize
+                n0 = n1 - 1
 
                 self._n = n0
                 if self._move_data:

diff --git a/checkpoint_schedules/schedule.py b/checkpoint_schedules/schedule.py
@@ -17,7 +17,7 @@
         "EndForward",
         "EndReverse",
         "CheckpointSchedule",
-        "StorageType"
+        "StorageType",
     ]