Fix EntryArchive

nomad-coe · Mar 12, 2024 · 5c56b90 · 5c56b90
1 parent a9d1c28
commit 5c56b90
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/nomad_simulations/utils/utils.py b/src/nomad_simulations/utils/utils.py
@@ -22,6 +22,7 @@
 from structlog.stdlib import BoundLogger
 
 from nomad.datamodel.data import ArchiveSection
+from nomad.datamodel import EntryArchive
 
 
 def get_sibling_section(
@@ -128,12 +129,12 @@ def is_not_representative(model_system, logger: BoundLogger = None):
     return False
 
 
-def check_archive(archive: ArchiveSection, logger: BoundLogger = None):
+def check_archive(archive: EntryArchive, logger: BoundLogger = None):
     """
-    Checks if the given `ArchiveSection` is empty and logs a warning.
+    Checks if the given `EntryArchive` is empty and logs a warning.
 
     Args:
-        archive (ArchiveSection): The `ArchiveSection` to check.
+        archive (EntryArchive): The `EntryArchive` to check.
         logger (BoundLogger): The logger to log messages.
 
     Returns: