Actions updated

Author: Abeshith

main.py

@@ -1,7 +1,3 @@
-#!/usr/bin/env python3
-"""
-Main entry point for running ML pipeline stages
-"""
 import sys
 from pathlib import Path
 

tests/test_app.py

@@ -1,6 +1,11 @@
 import sys
 import os
-sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+import sys
+from pathlib import Path
+
+# Add src to Python path
+project_root = Path(__file__).parent
+sys.path.insert(0, str(project_root))
 
 from app import app