Use float if initial param value is int

vega · Dec 13, 2023 · 8689751 · 8689751
1 parent 11a9056
commit 8689751
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/altair/jupyter/jupyter_chart.py b/altair/jupyter/jupyter_chart.py
@@ -23,9 +23,7 @@ def __init__(self, trait_values):
         super().__init__()
 
         for key, value in trait_values.items():
-            if isinstance(value, int):
-                traitlet_type = traitlets.Int()
-            elif isinstance(value, float):
+            if isinstance(value, (int, float)):
                 traitlet_type = traitlets.Float()
             elif isinstance(value, str):
                 traitlet_type = traitlets.Unicode()