type fixes

cgevans · Jun 28, 2024 · cf04f7e · cf04f7e
1 parent 4fe792d
commit cf04f7e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/src/alhambra_mixes/quantitate.py b/src/alhambra_mixes/quantitate.py
@@ -50,7 +50,7 @@
 import pint
 from pint import Quantity
 
-from .units import Q_, DecimalQuantity, _parse_vol_optional, nmol, normalize, uM, ureg
+from .units import NAN_AMOUNT, NAN_VOL, Q_, DecimalQuantity, _parse_vol_optional, nmol, normalize, uM, ureg
 
 
 def parse_vol(vol: Union[float, int, str, DecimalQuantity]) -> DecimalQuantity:

diff --git a/src/alhambra_mixes/util.py b/src/alhambra_mixes/util.py
@@ -1,6 +1,9 @@
 from __future__ import annotations
 
-from typing import TypeVar
+from typing import TYPE_CHECKING, TypeVar
+
+if TYPE_CHECKING:
+    from kithairon.picklists import PickList
 
 T = TypeVar("T")
 

diff --git a/tests/test_mixes.py b/tests/test_mixes.py
@@ -493,7 +493,7 @@ def assert_close(
 ) -> None:
     # This helps with comparing Decimal quantities, which cannot be multiplied by floats, which is
     # what happens with the default rtol parameter of pint.testing.assert_allclose.
-    pint.testing.assert_allclose(actual, expected, rtol, atol, msg)
+    pint.testing.assert_allclose(actual, expected, rtol, atol, msg) # type: ignore
 
 
 def test_split_mix():