fix: correct type hint for operators variable in ExpressionGenerator

tum-ai · Dec 20, 2024 · 1874f8f · 1874f8f
1 parent 503d6ae
commit 1874f8f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/data/arithmetics_dataset/expression_generator.py b/data/arithmetics_dataset/expression_generator.py
@@ -1,4 +1,5 @@
 import random
+from typing import Optional
 
 
 class ExpressionGenerator:
@@ -8,7 +9,7 @@ class ExpressionGenerator:
 
     def __init__(
         self,
-        operators: list[str] = None,
+        operators: Optional[list[str]] = None,
         negative_probability: float = 0.5,
         parentheses_probability: float = 1.0,
     ):