Update src/__init__.py

src/__init__.py

@@ -1,24 +1,22 @@
-"""
-Geometric Bayesian KAN: Uncertainty-Aware 3D Molecular Property Prediction
-"""
-
-__version__ = "0.1.0"
-__author__ = "GB-KAN Team"
-
-from . import layers
-from . import models
-from . import training
-from . import data
-from . import evaluation
-from . import baselines
-from . import utils
-
-__all__ = [
-    "layers",
-    "models",
-    "training",
-    "data",
-    "evaluation",
-    "baselines",
-    "utils",
-]
+"""
+Geometric Bayesian KAN: Uncertainty-Aware 3D Molecular Property Prediction
+"""
+
+__version__ = "0.1.0"
+__author__ = "GB-KAN Team"
+
+from . import layers
+from . import models
+from . import training
+from . import data
+from . import evaluation
+from . import utils
+
+__all__ = [
+    "layers",
+    "models",
+    "training",
+    "data",
+    "evaluation",
+    "utils",
+]