From 2e5d808072605d9a147b3926eb1080275303e497 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E7=90=9B=E6=B6=B5?= <wchpub@163.com>
Date: Wed, 13 Nov 2024 20:17:19 +0800
Subject: [PATCH] data

---
 data/data.py      | 2 ++
 utils/__init__.py | 0
 2 files changed, 2 insertions(+)
 create mode 100644 utils/__init__.py

diff --git a/data/data.py b/data/data.py
index e0b8a53..54ef3fb 100644
--- a/data/data.py
+++ b/data/data.py
@@ -95,6 +95,8 @@ def test_dataset():
             print(batch['brep_features'].shape)
             print(batch['sdf'].shape)
             break
+    except Exception as e:
+        logger.error(f"Error in test_dataset: {str(e)}")
 
 if __name__ == '__main__':
     test_dataset()
\ No newline at end of file
diff --git a/utils/__init__.py b/utils/__init__.py
new file mode 100644
index 0000000..e69de29