diff --git a/run.py b/run.py
index 2327de83..341a99ee 100755
--- a/run.py
+++ b/run.py
@@ -113,11 +113,7 @@ def launch_hdf5(config: DictConfig):
 
     log.info(f"Task: {task_name}")
 
-    if task_name in [
-        TASK_NAMES.FIT.value,
-        TASK_NAMES.TEST.value,
-        TASK_NAMES.FINETUNE.value,
-    ]:
+    if task_name in [TASK_NAMES.FIT.value, TASK_NAMES.TEST.value, TASK_NAMES.FINETUNE.value]:
         # load environment variables from `.env` file if it exists
         # recursively searches for `.env` in all folders starting from work dir
         dotenv.load_dotenv(override=True)