From 660c94143d95cad7a00c7a5890be39e4ed16595e Mon Sep 17 00:00:00 2001
From: Ashir Mumtaz <110593054+ashirmumtaz@users.noreply.github.com>
Date: Sat, 15 Feb 2025 16:44:38 +0500
Subject: [PATCH] Update train.py

---
 bin/train.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/bin/train.py b/bin/train.py
index 7938b24..866ffc2 100644
--- a/bin/train.py
+++ b/bin/train.py
@@ -27,7 +27,8 @@
 from pytorch_lightning.strategies.ddp import DDPStrategy
 
 from transformers import BertConfig
-
+# Add parent directory to Python path
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 from foldingdiff import datasets
 from foldingdiff import modelling
 from foldingdiff import losses