Fix dtype conversion

tum-pbs · Nov 28, 2023 · 26b1918 · 26b1918
1 parent a70c465
commit 26b1918
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/phiml/backend/_numpy_backend.py b/phiml/backend/_numpy_backend.py
@@ -514,7 +514,7 @@ def matrix_solve_least_squares(self, matrix: TensorType, rhs: TensorType) -> Ten
 
     def solve_triangular_dense(self, matrix, rhs, lower: bool, unit_diagonal: bool):
         if matrix.ndim == 2:
-            return scipy.linalg.solve_triangular(matrix, rhs, lower=lower, unit_diagonal=unit_diagonal).astype(matrix.dtype)
+            return scipy.linalg.solve_triangular(matrix, rhs, lower=lower, unit_diagonal=unit_diagonal)
         else:
             batch_size = matrix.shape[0]
             return np.stack([self.solve_triangular(matrix[b], rhs[b], lower, unit_diagonal) for b in range(batch_size)])