Linting

src/nomad_material_processing/general.py

@@ -1,13 +1,5 @@
 from typing import TYPE_CHECKING
 
-if TYPE_CHECKING:
-    from nomad.datamodel.datamodel import (
-        EntryArchive,
-    )
-    from structlog.stdlib import (
-        BoundLogger,
-    )
-
 import numpy as np
 from nomad.config import config
 from nomad.datamodel.data import ArchiveSection, EntryData
@@ -38,6 +30,15 @@
     SubSection,
 )
 
+if TYPE_CHECKING:
+    from nomad.datamodel.datamodel import (
+        EntryArchive,
+    )
+    from structlog.stdlib import (
+        BoundLogger,
+    )
+
+
 m_package = SchemaPackage(
     aliases=[
         'nomad_material_processing',

src/nomad_material_processing/solution/general.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Union
+from typing import TYPE_CHECKING
 
 from nomad.datamodel.data import (
     ArchiveSection,
@@ -247,7 +247,7 @@ class SolutionComponent(PureSubstanceComponent, BaseSolutionComponent):
     molar_concentration = SubSection(section_def=MolarConcentration)
     pure_substance = SubSection(section_def=PubChemPureSubstanceSection)
 
-    def _calculate_moles(self, logger: 'BoundLogger' = None) -> Union[Quantity, None]:
+    def _calculate_moles(self, logger: 'BoundLogger' = None) -> Quantity | None:
         """
         A private method to calculate the moles of a component in the solution.