From 138d3098275c7f80e5686f93fd080729e2b1a9f3 Mon Sep 17 00:00:00 2001
From: Orhidea Shatri <140660448+scrappie1@users.noreply.github.com>
Date: Tue, 29 Sep 2026 13:46:39 +0200
Subject: [PATCH] Fix SQLiteWriter optional LEGO import

---
 SQLiteWriter.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/SQLiteWriter.py b/SQLiteWriter.py
index 7b074c2..7c6d672 100644
--- a/SQLiteWriter.py
+++ b/SQLiteWriter.py
@@ -1,11 +1,16 @@
+from __future__ import annotations
+
 import os
 import sqlite3
+from typing import TYPE_CHECKING
 
 import pandas as pd
 import pyomo.core.base.set
 import pyomo.environ as pyo
 from InOutModule.printer import Printer
-from LEGO.LEGO import LEGO
+
+if TYPE_CHECKING:
+    from LEGO.LEGO import LEGO
 
 printer = Printer.getInstance()