more importlib hacks

docs/scripts/_stats.py

@@ -1,10 +1,29 @@
 """Statistical analysis of the results."""
 
+import sys
+from importlib.util import module_from_spec, spec_from_file_location
+from pathlib import Path
+
 import matplotlib.pyplot as plt
 import pandas as pd
 import seaborn as sns
 
-from ._plotting import melt_and_process
+# Import plotting module using full path
+CURRENT_DIR = Path(__file__).parent
+plotting_path = CURRENT_DIR / "_plotting.py"
+
+if "_plotting" not in sys.modules:
+    spec = spec_from_file_location("_plotting", plotting_path)
+    if spec is None or spec.loader is None:
+        msg = "Could not load _plotting module"
+        raise ImportError(msg)
+    plotting = module_from_spec(spec)
+    sys.modules["_plotting"] = plotting
+    spec.loader.exec_module(plotting)
+else:
+    plotting = sys.modules["_plotting"]
+
+melt_and_process = plotting.melt_and_process
 
 
 def calculate_stats(overview: pd.DataFrame) -> None:

docs/scripts/hooks.py

@@ -10,26 +10,34 @@
 
 import pandas as pd
 
+# Get the current directory
+CURRENT_DIR = Path(__file__).parent
 
-# Dynamically import local modules
+# Dynamically import local modules with full paths
 def import_local_module(module_name: str) -> None:
     """Import a local module."""
-    module_path = Path(__file__).parent / f"_{module_name}.py"
+    module_path = CURRENT_DIR / f"_{module_name}.py"
+    module = None
+
+    # If module is already loaded, return it
+    if f"_{module_name}" in sys.modules:
+        return sys.modules[f"_{module_name}"]
+
     spec = spec_from_file_location(f"_{module_name}", module_path)
     if spec is None or spec.loader is None:
         msg = f"Could not load module {module_name}"
         raise ImportError(msg)
     module = module_from_spec(spec)
-    sys.modules[spec.name] = module  # This makes the module available for other imports
+    sys.modules[module.__name__] = module
     spec.loader.exec_module(module)
     return module
 
-# Import the local modules
+# Import the modules
 plotting = import_local_module("plotting")
 preprocess = import_local_module("preprocess")
 stats = import_local_module("stats")
 
-# Extract the needed functions from the modules
+# Extract the needed functions
 plot_accuracy_per_model = plotting.plot_accuracy_per_model
 plot_accuracy_per_quantisation = plotting.plot_accuracy_per_quantisation
 plot_accuracy_per_task = plotting.plot_accuracy_per_task