Skip to content

Commit

Permalink
Deleted EntryArchive from imports
Browse files Browse the repository at this point in the history
  • Loading branch information
JosePizarro3 committed Jan 22, 2024
1 parent 4eda8c8 commit 0c1b1ea
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 5 deletions.
3 changes: 1 addition & 2 deletions simulationdataschema/general.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,6 @@
# m_env.m_add_sub_section(Environment.packages, system.m_package)
import numpy as np

from nomad.datamodel import EntryArchive
from nomad.datamodel.data import EntryData
from nomad.datamodel.metainfo.basesections import Computation as BaseComputation
from nomad.metainfo import SubSection
Expand Down Expand Up @@ -84,7 +83,7 @@ def _set_system_tree_index(
system_child.tree_index = tree_index + 1
self._set_system_tree_index(system_child, tree_index + 1)

def normalize(self, archive: EntryArchive, logger) -> None:
def normalize(self, archive, logger) -> None:
super(EntryData, self).normalize(archive, logger)

# Finding which is the representative system of a calculation: typically, we will
Expand Down
6 changes: 3 additions & 3 deletions simulationdataschema/system.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@
from nomad.atomutils import Formula, get_normalized_wyckoff, search_aflow_prototype

from nomad.datamodel.data import ArchiveSection
from nomad.datamodel import EntryArchive

from nomad.metainfo import Quantity, SubSection, SectionProxy, MEnum
from nomad.datamodel.metainfo.basesections import System, RealSpace
from nomad.datamodel.metainfo.annotations import ELNAnnotation
Expand Down Expand Up @@ -180,7 +180,7 @@ class AtomicCell(RealSpace):
""",
)

def normalize(self, archive: EntryArchive, logger) -> None:
def normalize(self, archive, logger) -> None:
# Check if AtomicCell section exists
if self is None:
logger.error("Could not find the basic System.atomic_cell information.")
Expand Down Expand Up @@ -750,7 +750,7 @@ def _resolve_bulk_symmetry(self, ase_atoms: ase.Atoms) -> None:
sec_symmetry.m_set(val, symmetry.get(key))
sec_symmetry.atomic_cell_ref = self.atomic_cell[-1]

def normalize(self, archive: EntryArchive, logger) -> None:
def normalize(self, archive, logger) -> None:
super().normalize(archive, logger)
self.logger = logger

Expand Down

0 comments on commit 0c1b1ea

Please sign in to comment.