fix: mypy error in colormap.py

yzhou377 · Jun 20, 2019 · 8e62aca · 8e62aca
1 parent 69d5422
commit 8e62aca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/argoverse/visualization/colormap.py b/argoverse/visualization/colormap.py
@@ -23,7 +23,7 @@
 import numpy as np
 
 
-def colormap(rgb=False):
+def colormap(rgb: bool = False) -> np.ndarray:
     color_list = np.array(
         [
             0.000, 0.447, 0.741,