[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
deepmodeling · Apr 3, 2024 · e56d0b3 · e56d0b3
1 parent 1a34fc9
commit e56d0b3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/dpdata/pymatgen/structure.py b/dpdata/pymatgen/structure.py
@@ -1,7 +1,7 @@
 import numpy as np
 
 try:
-    from pymatgen.core import Structure # noqa: F401
+    from pymatgen.core import Structure  # noqa: F401
 except ImportError:
     pass
 
@@ -19,6 +19,6 @@ def from_system_data(structure) -> dict:
         "atom_numbs": atom_numbs,
         "atom_types": atom_types,
         "coords": np.array([coords]),
-        "cells": np.array([cells])
+        "cells": np.array([cells]),
     }
     return info_dict
diff --git a/tests/test_from_pymatgen.py b/tests/test_from_pymatgen.py
@@ -1,7 +1,7 @@
 import os
 import unittest
 
-from comp_sys import CompSys, IsPBC
+from comp_sys import CompSys
 from context import dpdata
 
 try: