Fix mypy

nomad-coe · May 16, 2024 · f4caa3a · f4caa3a
1 parent 0e30cfc
commit f4caa3a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -18,7 +18,7 @@
 
 import os
 import pytest
-from typing import List, Optional
+from typing import List, Optional, Union
 
 from nomad.units import ureg
 from nomad.datamodel import EntryArchive
@@ -221,7 +221,7 @@ def generate_electronic_eigenvalues(
         [0, 1, 1],
         [1, 1, 1],
     ],
-    value: Optional[List[float]] = None,
+    value: Optional[List[Union[List[float], List[int]]]] = None,
     occupation: Optional[List[float]] = None,
     highest_occupied: Optional[float] = None,
     lowest_unoccupied: Optional[float] = None,