add imports to init

RasmusOrsoe · Sep 13, 2024 · bf3fc6e · bf3fc6e
1 parent 50f9a35
commit bf3fc6e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/graphnet/data/__init__.py b/src/graphnet/data/__init__.py
@@ -8,4 +8,9 @@
 from .pre_configured import I3ToParquetConverter
 from .pre_configured import I3ToSQLiteConverter
 from .datamodule import GraphNeTDataModule
-from .curated_datamodule import CuratedDataset, ERDAHostedDataset
+from .curated_datamodule import (
+    CuratedDataset,
+    ERDAHostedDataset,
+    SecretDataset,
+    PublicBenchmarkDataset,
+)