Fix mypy errors.

materialsproject · Apr 11, 2024 · b30b9e7 · b30b9e7
1 parent 5e7ecf5
commit b30b9e7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/emmet-core/emmet/core/classical_md/openmm/tasks.py b/emmet-core/emmet/core/classical_md/openmm/tasks.py
@@ -9,8 +9,8 @@
 from emmet.core.vasp.task_valid import TaskState  # type: ignore[import-untyped]
 from pydantic import BaseModel, Field
 
-from emmet.core.classical_md import ClassicalMDTaskDocument
-from emmet.core.classical_md.tasks import HexBytes
+from emmet.core.classical_md import ClassicalMDTaskDocument  # type: ignore[import-untyped]
+from emmet.core.classical_md.tasks import HexBytes  # type: ignore[import-untyped]
 
 
 class CalculationInput(BaseModel, extra="allow"):  # type: ignore[call-arg]

diff --git a/emmet-core/emmet/core/classical_md/tasks.py b/emmet-core/emmet/core/classical_md/tasks.py
@@ -15,10 +15,11 @@
     PlainValidator,
     PlainSerializer,
     WithJsonSchema,
+    errors,
 )
 from monty.json import MSONable
 
-from emmet.core.vasp.task_valid import TaskState
+from emmet.core.vasp.task_valid import TaskState  # type: ignore[import-untyped]
 
 
 def hex_bytes_validator(o: Any) -> bytes:
@@ -28,6 +29,7 @@ def hex_bytes_validator(o: Any) -> bytes:
         return bytes(o)
     elif isinstance(o, str):
         return zlib.decompress(bytes.fromhex(o))
+    raise errors.BytesError()
 
 
 def hex_bytes_serializer(b: bytes) -> str: