From e2a327baca1e58d5f2217f83cb5f1626558479cf Mon Sep 17 00:00:00 2001
From: SevenLJY <jiayeeliu@gmail.com>
Date: Mon, 1 Apr 2024 16:12:49 -0700
Subject: [PATCH] refractor datasets

---
 datamodules/base.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/datamodules/base.py b/datamodules/base.py
index fb5c62a..284433f 100644
--- a/datamodules/base.py
+++ b/datamodules/base.py
@@ -181,6 +181,7 @@ def _prepare_item(self, idx):
         cond['name'] = self.model_ids[idx]
         cond['obj_cat'] = file['meta']['obj_cat']
         cond['tree_hash'] = file['meta']['tree_hash']
+        return data, cond
     
     def __getitem__(self, index):
         raise NotImplementedError