type hint

delphi-suite · Feb 5, 2024 · 0ae352b · 0ae352b
1 parent 49bc620
commit 0ae352b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/delphi/train/shuffle.py b/src/delphi/train/shuffle.py
@@ -12,7 +12,7 @@ def __init__(self, seed: int):
         assert 0 <= seed < self.m
         self.state = seed
 
-    def __call__(self):
+    def __call__(self) -> int:
         self.state = (self.state * self.a) % self.m
         return self.state