diff --git a/src/atomate2/forcefields/jobs.py b/src/atomate2/forcefields/jobs.py index 661c546d09..39ebb6f44c 100644 --- a/src/atomate2/forcefields/jobs.py +++ b/src/atomate2/forcefields/jobs.py @@ -64,7 +64,7 @@ def make(self, structure: Structure): structure, relax_cell=self.relax_cell, steps=self.steps, **self.relax_kwargs ) - ff_task_doc = ForceFieldTaskDocument.from_chgnet_result( + return ForceFieldTaskDocument.from_chgnet_result( result, self.relax_cell, self.steps, @@ -73,8 +73,6 @@ def make(self, structure: Structure): **self.task_document_kwargs, ) - return ff_task_doc - @dataclass class CHGNetStaticMaker(CHGNetRelaxMaker):