take ruff's suggestions

Author: yaoyi92

src/atomate2/forcefields/jobs.py

@@ -16,9 +16,9 @@
 from atomate2.ase.jobs import AseRelaxMaker
 from atomate2.forcefields import MLFF, _get_formatted_ff_name
 from atomate2.forcefields.schemas import (
-    ForceFieldStructureTaskDocument, 
     ForceFieldMoleculeTaskDocument, 
-    ForceFieldTaskDocument
+    ForceFieldStructureTaskDocument, 
+    ForceFieldTaskDocument,
 )
 from atomate2.forcefields.utils import ase_calculator, revert_default_dtype
 

src/atomate2/forcefields/md.py

@@ -16,8 +16,8 @@
     _FORCEFIELD_DATA_OBJECTS,
 )
 from atomate2.forcefields.schemas import (
-    ForceFieldStructureTaskDocument, 
     ForceFieldMoleculeTaskDocument, 
+    ForceFieldStructureTaskDocument, 
     ForceFieldTaskDocument,
 )
 from atomate2.forcefields.utils import ase_calculator, revert_default_dtype

src/atomate2/forcefields/schemas.py

@@ -11,9 +11,9 @@
 from pymatgen.core import Structure, Molecule
 
 from atomate2.ase.schemas import (
+    AseMoleculeTaskDoc, 
     AseObject, 
     AseResult, 
-    AseMoleculeTaskDoc, 
     AseStructureTaskDoc, 
     AseTaskDoc,
     _task_doc_translation_keys,