Update stokes_cylinder

nbouziani · Jan 21, 2024 · 4f2339f · 4f2339f
1 parent 85b716d
commit 4f2339f
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 1 deletion.
diff --git a/data/datasets/stokes_cylinder/test_data.h5 b/data/datasets/stokes_cylinder/test_data.h5
diff --git a/data/datasets/stokes_cylinder/train_data.h5 b/data/datasets/stokes_cylinder/train_data.h5
diff --git a/physics_driven_ml/dataset_processing/pde_dataset.py b/physics_driven_ml/dataset_processing/pde_dataset.py
@@ -95,7 +95,7 @@ def load_dataset(self, fname: str):
             # Load adjacency list
             edge_index = np.array(afile.h5pyfile["edge_index"])
             # Load mesh
-            mesh = afile.load_mesh("mesh")
+            mesh = afile.load_mesh()
             # Load data
             for i in range(n):
                 f = afile.load_function(mesh, "f", idx=i)