Skip to content

Commit

Permalink
Lint with black
Browse files Browse the repository at this point in the history
  • Loading branch information
mkhorton authored Sep 28, 2024
1 parent e461bbe commit ee3201d
Showing 1 changed file with 18 additions and 5 deletions.
23 changes: 18 additions & 5 deletions emmet-core/emmet/core/symmetry.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,12 @@
from pydantic import BaseModel, Field
from pymatgen.core import Structure
from pymatgen.core.structure import Molecule
from pymatgen.symmetry.analyzer import PointGroupAnalyzer, SpacegroupAnalyzer, spglib, SymmetryUndeterminedError
from pymatgen.symmetry.analyzer import (
PointGroupAnalyzer,
SpacegroupAnalyzer,
spglib,
SymmetryUndeterminedError,
)

from emmet.core.settings import EmmetSettings
from emmet.core.utils import ValueEnum
Expand Down Expand Up @@ -145,7 +150,7 @@ class SymmetryData(BaseModel):
angle_tolerance: Optional[float] = Field(
None,
title="Angle Tolerance",
description="Angle tolerance provided to spglib to determine the symmetry of this structure."
description="Angle tolerance provided to spglib to determine the symmetry of this structure.",
)

version: Optional[str] = Field(None, title="spglib version")
Expand All @@ -162,16 +167,24 @@ def from_structure(cls, structure: Structure) -> "SymmetryData":
"hall": None,
"version": spglib.__version__,
"symprec": SETTINGS.SYMPREC,
"angle_tolerance": SETTINGS.ANGLE_TOL
"angle_tolerance": SETTINGS.ANGLE_TOL,
}

try:
sg = SpacegroupAnalyzer(structure, symprec=symmetry["symprec"], angle_tolerance=symmetry["angle_tolerance"])
sg = SpacegroupAnalyzer(
structure,
symprec=symmetry["symprec"],
angle_tolerance=symmetry["angle_tolerance"],
)
except SymmetryUndeterminedError:
try:
symmetry["symprec"] = 1e-3
symmetry["angle_tolerance"] = 1
sg = SpacegroupAnalyzer(structure, symprec=symmetry["symprec"], angle_tolerance=symmetry["angle_tolerance"])
sg = SpacegroupAnalyzer(
structure,
symprec=symmetry["symprec"],
angle_tolerance=symmetry["angle_tolerance"],
)
except SymmetryUndeterminedError:
return SymmetryData(**symmetry)

Expand Down

0 comments on commit ee3201d

Please sign in to comment.