From ede525b694ce58784428500a626ba570c872a079 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Sun, 11 Feb 2024 17:27:30 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 aiida_cp2k/utils/input_generator.py | 6 +++---
 aiida_cp2k/workchains/base.py       | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/aiida_cp2k/utils/input_generator.py b/aiida_cp2k/utils/input_generator.py
index 6e1b4a73..dc17e6f4 100644
--- a/aiida_cp2k/utils/input_generator.py
+++ b/aiida_cp2k/utils/input_generator.py
@@ -207,11 +207,11 @@ def add_wfn_restart_section(input_dict, is_kpoints):
 
 
 @calcfunction
-def add_ext_restart_section(input_dict,first_snapshot=None):
+def add_ext_restart_section(input_dict, first_snapshot=None):
     """Add external restart section to the input dictionary."""
     params = input_dict.get_dict()
     if first_snapshot is not None:
-         params["MOTION"]["MD"]["REFTRAJ"]["FIRST_SNAPSHOT"] = first_snapshot
+        params["MOTION"]["MD"]["REFTRAJ"]["FIRST_SNAPSHOT"] = first_snapshot
     # overwrite the complete EXT_RESTART section if present
     params["EXT_RESTART"] = {"RESTART_FILE_NAME": "./parent_calc/aiida-1.restart"}
     if params["GLOBAL"]["RUN_TYPE"] == "MD":
@@ -221,5 +221,5 @@ def add_ext_restart_section(input_dict,first_snapshot=None):
         params["EXT_RESTART"]["RESTART_VEL"] = ".TRUE."
         params["EXT_RESTART"]["RESTART_CELL"] = ".TRUE."
         params["EXT_RESTART"]["RESTART_THERMOSTAT"] = ".TRUE."
-        params["EXT_RESTART"]["RESTART_CONSTRAINT"] = ".FALSE."    
+        params["EXT_RESTART"]["RESTART_CONSTRAINT"] = ".FALSE."
     return Dict(params)
diff --git a/aiida_cp2k/workchains/base.py b/aiida_cp2k/workchains/base.py
index 05073e58..4bd420fd 100644
--- a/aiida_cp2k/workchains/base.py
+++ b/aiida_cp2k/workchains/base.py
@@ -94,7 +94,7 @@ def restart_incomplete_calculation(self, calc):
             try:
                 first_snapshot = int(params['MOTION']['MD']['REFTRAJ']['FIRST_SNAPSHOT']) + calc.outputs.output_trajectory.get_shape('positions')[0]
             except KeyError:
-                pass                
+                pass
             params = add_ext_restart_section(params,first_snapshot=first_snapshot)
 
         self.ctx.inputs.parameters = params  # params (new or old ones) that include the necessary restart information.