data

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()