Cast high/lo in RBF (#326)

ur-whitelab · Aug 3, 2021 · 04ca434 · 04ca434
1 parent 49eac84
commit 04ca434
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/htf/layers.py b/htf/layers.py
@@ -29,7 +29,8 @@ def __init__(self, low, high, count):
         super(RBFExpansion, self).__init__(name='rbf-layer')
         self.low = low
         self.high = high
-        self.centers = tf.cast(tf.linspace(low, high, count), dtype=tf.float32)
+        self.centers = tf.cast(tf.linspace(
+            float(low), float(high), count), dtype=tf.float32)
         self.gap = self.centers[1] - self.centers[0]
 
     def get_config(self):