fix: test_architecture.py — use os.path.dirname(__file__) instead of hardcoded /app for sys.path

test_architecture.py

@@ -11,10 +11,14 @@ Tests the complete forward pass with synthetic data to verify:
 7. Loss function variants (smooth_l1, mse, cosine)
 8. Anti-collapse regularizations (SIGReg, VICReg)
 9. Parameter counting is correct
+
+Run from repo root:  python test_architecture.py
 """
 
+import os
 import sys
-sys.path.insert(0, '/app')
+# Ensure the repo root is on the path (where mr_jepa/ package lives)
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
 
 import torch
 import torch.nn as nn
@@ -256,9 +260,6 @@ def test_ablation_no_jepa():
     """no_jepa: model forward should skip JEPA entirely."""
     print("\n=== Ablation: --no_jepa ===")
     D, K, B, N_s = 256, 3, 2, 8
-    # The train_mrjepa.py handles this at model level: when use_jepa=False,
-    # the model skips target_encoder forward and returns task_loss only.
-    # Here we verify the JEPALoss still computes (it's the model that decides whether to call it).
     cfg = JEPAObjectiveConfig(use_sigreg=True, sigreg_weight=0.1)
     loss_fn = JEPALoss(cfg, D)
     pred = torch.randn(B, K + 1, N_s, D, requires_grad=True)