Fix mypy

mggg · Nov 16, 2023 · 370e497 · 370e497
1 parent 6921dbe
commit 370e497
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/votekit/elections/election_types.py b/src/votekit/elections/election_types.py
@@ -114,9 +114,12 @@ def run_step(self, step: Optional[Union[int, None]] = None) -> ElectionState:
                     self.run_step()
 
                 while self.state.curr_round > step:
-                    self.state = self.state.previous
+                    if self.state.previous:
+                        self.state = self.state.previous
+                    else:
+                        raise ValueError("Previous state is None type.") 
 
-                return(self.state)
+            return(self.state)
 
         # run next step   
         else: