From 96609df28073c721c4ea9222b1a4f4e6f520e7c2 Mon Sep 17 00:00:00 2001
From: 1andrin <115493865+1andrin@users.noreply.github.com>
Date: Fri, 25 Oct 2024 00:46:59 +0200
Subject: [PATCH] linter issues

Signed-off-by: 1andrin <115493865+1andrin@users.noreply.github.com>
---
 notebooks/RunExperiments/experiment_runner.py | 18 ++++++++----------
 1 file changed, 8 insertions(+), 10 deletions(-)

diff --git a/notebooks/RunExperiments/experiment_runner.py b/notebooks/RunExperiments/experiment_runner.py
index 49a33c3f..98e28a2b 100644
--- a/notebooks/RunExperiments/experiment_runner.py
+++ b/notebooks/RunExperiments/experiment_runner.py
@@ -1,10 +1,5 @@
 import os
 import sys
-
-# Ensure CausalTune is in the Python path
-root_path = os.path.realpath("../../../..")
-sys.path.append(os.path.join(root_path, "causaltune"))
-
 import pickle
 import numpy as np
 import matplotlib
@@ -17,11 +12,14 @@
 
 warnings.filterwarnings("ignore")
 
-from causaltune import CausalTune
-from causaltune.datasets import (
-    load_dataset,
-)
-from causaltune.models.passthrough import passthrough_model
+# Ensure CausalTune is in the Python path
+root_path = os.path.realpath("../../../..")
+sys.path.append(os.path.join(root_path, "causaltune"))  # noqa: E402
+
+# Import CausalTune and other custom modules after setting up the path
+from causaltune import CausalTune  # noqa: E402
+from causaltune.datasets import load_dataset  # noqa: E402
+from causaltune.models.passthrough import passthrough_model  # noqa: E402
 
 
 def parse_arguments():