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