fix: Properly handle NaN and Inf default values (#61)

niivue · Apr 19, 2024 · c937533 · c937533
1 parent f00f972
commit c937533
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/scripts/generate_options_mixin.py b/scripts/generate_options_mixin.py
@@ -42,10 +42,10 @@ def type_hint(value: typing.Any):
 
 
 def get_value(value: typing.Any):
-    if isinstance(value, float) and math.isnan(value):
-        return 'float("nan")'
     if value == float("inf"):
         return 'float("inf")'
+    if isinstance(value, float) and math.isnan(value):
+        return 'float("nan")'
     if isinstance(value, SliceType):
         return f"SliceType.{value.name}"
     if isinstance(value, MuliplanarType):
@@ -66,7 +66,7 @@ def generate_mixin(options: typing.Dict[str, typing.Any]):
         "",
         "import typing",
         "",
-        "from ._constants import SliceType, MuliplanarType, DragMode",
+        "from ._constants import DragMode, MuliplanarType, SliceType",
         "",
         '__all__ = ["OptionsMixin"]',
         "",

diff --git a/src/ipyniivue/_options_mixin.py b/src/ipyniivue/_options_mixin.py
@@ -8,7 +8,6 @@
 
 __all__ = ["OptionsMixin"]
 
-
 class OptionsMixin:
     @property
     def text_height(self) -> float: