linting

Author: VicTrqt

src/atomate2/abinit/run.py

@@ -67,7 +67,6 @@ def run_abinit(
                     process.terminate()
 
         process.wait()
-    return
 
 
 def run_mrgddb(
@@ -114,7 +113,6 @@ def run_mrgddb(
                     process.terminate()
 
         process.wait()
-    return
 
 
 def run_anaddb(
@@ -162,4 +160,3 @@ def run_anaddb(
                     process.terminate()
 
         process.wait()
-    return

src/atomate2/abinit/schemas/calculation.py

@@ -251,7 +251,7 @@ def from_abinit_files(
         abinit_log_file: Path | str = LOG_FILE_NAME,
         abinit_abort_file: Path | str = MPIABORTFILE,
         abinit_out_file: Path | str = OUTPUT_FILE_NAME,
-    ) -> tuple[Calculation, dict[AbinitObject, dict]]:
+    ) -> tuple[Self, dict[AbinitObject, dict]]:
         """
         Create an Abinit calculation document from a directory and file paths.
 
@@ -319,15 +319,14 @@ def from_abinit_files(
             msg = f"{cls} exception while parsing event_report:\n{exc}"
             logger.critical(msg)
 
-        return (
-            cls(
+        instance = cls(
                 dir_name=str(dir_name),
                 task_name=task_name,
                 abinit_version=abinit_out.version,
                 has_abinit_completed=has_abinit_completed,
                 completed_at=completed_at,
                 output=output_doc,
                 event_report=report,
-            ),
-            None,  # abinit_objects,
-        )
+            )
+
+        return instance, None  # abinit_objects,

src/atomate2/abinit/schemas/task.py

@@ -14,6 +14,7 @@
 from pydantic import BaseModel, Field
 from pymatgen.core import Structure
 from pymatgen.io.abinit.pseudos import AbinitPseudo
+from typing_extensions import Self
 
 from atomate2.abinit.files import load_abinit_input
 from atomate2.abinit.schemas.calculation import AbinitObject, Calculation, TaskState