refractor datasets

3dlg-hcvc · Apr 1, 2024 · e2a327b · e2a327b
1 parent a100ca8
commit e2a327b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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