Update __init__.py

PaddlePaddle · Jan 12, 2025 · f6679d6 · f6679d6
1 parent aab1943
commit f6679d6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ppsci/data/dataset/__init__.py b/ppsci/data/dataset/__init__.py
@@ -27,6 +27,7 @@
 from ppsci.data.dataset.cylinder_dataset import MeshCylinderDataset
 from ppsci.data.dataset.darcyflow_dataset import DarcyFlowDataset
 from ppsci.data.dataset.dgmr_dataset import DGMRDataset
+from ppsci.data.dataset.drivaernet_dataset import DrivAerNetDataset
 from ppsci.data.dataset.drivaernetplusplus_dataset import DrivAerNetPlusPlusDataset
 from ppsci.data.dataset.enso_dataset import ENSODataset
 from ppsci.data.dataset.era5_dataset import ERA5Dataset
@@ -88,6 +89,7 @@
     "build_dataset",
     "CGCNNDataset",
     "FWIDataset",
+    "DrivAerNetDataset",
     "DrivAerNetPlusPlusDataset",
 ]