From 0fac8cdfec96c8ac1c28d982b2ae86f2b6161e53 Mon Sep 17 00:00:00 2001
From: Sarthak Kapoor <sarthakkapoor@hotmail.com>
Date: Thu, 2 Apr 2026 17:16:22 +0200
Subject: [PATCH] Linting

---
 src/nomad_material_processing/general.py        | 17 +++++++++--------
 .../solution/general.py                         |  4 ++--
 2 files changed, 11 insertions(+), 10 deletions(-)

diff --git a/src/nomad_material_processing/general.py b/src/nomad_material_processing/general.py
index e4e0f02..82008b7 100644
--- a/src/nomad_material_processing/general.py
+++ b/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',
diff --git a/src/nomad_material_processing/solution/general.py b/src/nomad_material_processing/solution/general.py
index a6e873f..aadf564 100644
--- a/src/nomad_material_processing/solution/general.py
+++ b/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.