Fix dtype of Geometry.boundary_normal

Author: lululxvi

deepxde/geometry/geometry_2d.py

@@ -8,12 +8,14 @@
 from .geometry import Geometry
 from .geometry_nd import Hypercube
 from .sampler import sample
+from .. import config
 from ..utils import vectorize
 
 
 class Disk(Geometry):
     def __init__(self, center, radius):
-        self.center, self.radius = np.array(center), radius
+        self.center = np.array(center, dtype=config.real(np))
+        self.radius = radius
         super(Disk, self).__init__(
             2, (self.center - radius, self.center + radius), 2 * radius
         )
@@ -181,9 +183,9 @@ def __init__(self, x1, x2, x3):
             self.area = -self.area
             x2, x3 = x3, x2
 
-        self.x1 = np.array(x1)
-        self.x2 = np.array(x2)
-        self.x3 = np.array(x3)
+        self.x1 = np.array(x1, dtype=config.real(np))
+        self.x2 = np.array(x2, dtype=config.real(np))
+        self.x3 = np.array(x3, dtype=config.real(np))
 
         self.v12 = self.x2 - self.x1
         self.v23 = self.x3 - self.x2
@@ -338,7 +340,7 @@ class Polygon(Geometry):
     """
 
     def __init__(self, vertices):
-        self.vertices = np.array(vertices)
+        self.vertices = np.array(vertices, dtype=config.real(np))
         if len(vertices) == 3:
             raise ValueError("The polygon is a triangle. Use Triangle instead.")
         if Rectangle.is_valid(self.vertices):

deepxde/geometry/geometry_nd.py

@@ -41,7 +41,9 @@ def on_boundary(self, x):
         return np.logical_and(self.inside(x), _on_boundary)
 
     def boundary_normal(self, x):
-        _n = np.isclose(x, self.xmin) * -1.0 + np.isclose(x, self.xmax) * 1.0
+        _n = -np.isclose(x, self.xmin).astype(config.real(np)) + np.isclose(
+            x, self.xmax
+        )
         # For vertices, the normal is averaged for all directions
         idx = np.count_nonzero(_n, axis=-1) > 1
         if np.any(idx):
@@ -97,7 +99,8 @@ def periodic_point(self, x, component):
 
 class Hypersphere(Geometry):
     def __init__(self, center, radius):
-        self.center, self.radius = np.array(center), radius
+        self.center = np.array(center, dtype=config.real(np))
+        self.radius = radius
         super(Hypersphere, self).__init__(
             len(center), (self.center - radius, self.center + radius), 2 * radius
         )
@@ -144,7 +147,7 @@ def random_points(self, n, random="pseudo"):
     def random_boundary_points(self, n, random="pseudo"):
         """http://mathworld.wolfram.com/HyperspherePointPicking.html"""
         if random == "pseudo":
-            X = np.random.normal(size=(n, self.dim))
+            X = np.random.normal(size=(n, self.dim)).astype(config.real(np))
         else:
             U = sample(n, self.dim, random)
             X = stats.norm.ppf(U)