Add missing Optional marker in argument typing

reinfer · Jun 17, 2022 · a36063b · a36063b
1 parent a71d783
commit a36063b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/ctranslate2/specs/model_spec.py b/python/ctranslate2/specs/model_spec.py
@@ -7,7 +7,7 @@
 import shutil
 import struct
 
-from typing import Dict, List
+from typing import Dict, List, Optional
 
 import numpy as np
 
@@ -185,7 +185,7 @@ def _quantize(spec, name, value):
 
         self._visit(_quantize)
 
-    def optimize(self, quantization: str = None) -> None:
+    def optimize(self, quantization: Optional[str] = None) -> None:
         """Recursively applies some optimizations to this layer:
 
         * Alias variables with the same shape and value.