fix import

GraphRouter_eval/__init__.py

@@ -0,0 +1 @@
+# GraphRouter_eval package 

GraphRouter_eval/data_processing/__init__.py

@@ -0,0 +1 @@
+# data_processing package 

GraphRouter_eval/model/__init__.py

@@ -0,0 +1 @@
+# model package 

demo.py

@@ -70,7 +70,17 @@ sys.path.append(os.path.join(os.path.dirname(__file__), 'GraphRouter_eval/model'
 sys.path.append(os.path.join(os.path.dirname(__file__), 'GraphRouter_eval/data_processing'))
 sys.path.append(os.path.join(os.path.dirname(__file__), 'GraphRouter_eval'))
 try:
-    from multi_task_graph_router import graph_router_prediction
+    # Import the GraphRouter_eval package
+    import sys
+    import os
+    
+    # Add the parent directory to Python path so we can import GraphRouter_eval as a package
+    current_dir = os.path.dirname(__file__)
+    if current_dir not in sys.path:
+        sys.path.insert(0, current_dir)
+    
+    # Import the required modules
+    from GraphRouter_eval.model.multi_task_graph_router import graph_router_prediction
     GRAPH_ROUTER_AVAILABLE = True
     print("✅ Graph router successfully imported")
 except ImportError as e: