Solve #377 in 0.2

mathLab · Nov 7, 2024 · 03710e0 · 03710e0
1 parent 02addbf
commit 03710e0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/pina/domain/operation_interface.py b/pina/domain/operation_interface.py
@@ -66,7 +66,7 @@ def _check_dimensions(self, geometries):
         :type geometries: list[Location]
         """
         for geometry in geometries:
-            if geometry.variables != geometries[0].variables:
+            if sorted(geometry.variables) != sorted(geometries[0].variables):
                 raise NotImplementedError(
                     f"The geometries need to have same dimensions and labels."
                 )
diff --git a/pina/domain/union_domain.py b/pina/domain/union_domain.py
@@ -108,10 +108,9 @@ def sample(self, n, mode="random", variables="all"):
             sampled_points.append(
                 geometry.sample(
                     num_points + int(i < remainder), mode, variables
-                )
+                ).sort_labels()
             )
             # in case number of sampled points is smaller than the number of geometries
             if len(sampled_points) >= n:
                 break
-
-        return LabelTensor(torch.cat(sampled_points), labels=self.variables)
+        return LabelTensor.cat(sampled_points)