(mypy) Replace Union[None,...] with Optional[...] for function params

TeamHG-Memex · Jul 11, 2019 · e8a34f1 · e8a34f1
1 parent 913d415
commit e8a34f1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/eli5/keras/gradcam.py b/eli5/keras/gradcam.py
@@ -164,7 +164,7 @@ def _calc_gradient(ys, xs):
 
 
 def _get_target_prediction(targets, estimator):
-    # type: (Union[None, list], Model) -> K.variable
+    # type: (Optional[list], Model) -> K.variable
     """
     Get a prediction ID based on ``targets``, 
     from the model ``estimator`` (with a rank 2 tensor for its final layer).