Fix mypy

nomad-coe · Jan 17, 2025 · bf7aa7c · bf7aa7c
1 parent efe1486
commit bf7aa7c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/electronicparsers/quantumespresso/parser.py b/electronicparsers/quantumespresso/parser.py
@@ -21,6 +21,7 @@
 import re
 from datetime import datetime
 import os
+from typing import Optional
 
 from nomad.units import ureg
 from nomad.parsing.file_parser.text_parser import TextParser, Quantity, DataTextParser
@@ -2800,7 +2801,7 @@ def __init__(self):
         }
         self._re_label = re.compile(r'([A-Z][a-z]?)')
 
-    def get_n_electrons_safe(self) -> float:
+    def get_n_electrons_safe(self) -> Optional[float]:
         n_electrons = self.out_parser.get('run', [])
         if n_electrons:
             n_electrons = n_electrons[0].get_header('number_of_electrons', {})
@@ -2811,6 +2812,7 @@ def get_n_electrons_safe(self) -> float:
                     'Number of electrons not found. Using spin up + down.'
                 )
                 return up + down
+        return None
 
     def parse_scc(self, run, calculation):
         sec_run = self.archive.run[-1]