From abcf352883ae4ba19bcfd6ffa56d5cf736d9e038 Mon Sep 17 00:00:00 2001
From: Ameya Daigavane <ameya.d.98@gmail.com>
Date: Sun, 19 May 2024 11:28:25 -0700
Subject: [PATCH] Fix path.

---
 moleculib/protein/dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/moleculib/protein/dataset.py b/moleculib/protein/dataset.py
index 0fd07e3..1825f9c 100644
--- a/moleculib/protein/dataset.py
+++ b/moleculib/protein/dataset.py
@@ -490,7 +490,7 @@ def __init__(
         split: str = "train",
         tau: int = 1,
     ):
-        base = "/mas/projects/molecularmachines/db/timewarp/"
+        base = "/mas/projects/molecularmachines/db/timewarp2/"
         self.base_path = os.path.join(base, dataset, split)
         self.counter = 0
         self.tau = tau