From 437bd7d616885c89220f342a66e0004d524b2de8 Mon Sep 17 00:00:00 2001
From: Void Agent <void@jayrup.hermes>
Date: Wed, 29 Jul 2026 19:48:28 +0100
Subject: jlens_v2: fix import path for src/ -> root layout

---
 src/jlens_v2.py | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'src')

diff --git a/src/jlens_v2.py b/src/jlens_v2.py
index 1bacbc6..c9b7019 100644
--- a/src/jlens_v2.py
+++ b/src/jlens_v2.py
@@ -7,6 +7,9 @@ import torch.nn as nn
 import numpy as np
 import pickle
 import os
+import os, sys
+# jlens_v2.py is in src/, model.py is in repo root
+sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 from collections import defaultdict
 
 def load_model(checkpoint_path, device='cuda'):
-- 
cgit v1.2.3