Small optimization.

GAA-UAM · Mar 30, 2023 · 53dfa29 · 53dfa29
1 parent 546f360
commit 53dfa29
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/skfda/_utils/_utils.py b/skfda/_utils/_utils.py
@@ -551,7 +551,11 @@ def map_function(
 
 def _int_to_real(array: Union[NDArrayInt, NDArrayFloat]) -> NDArrayFloat:
     """Convert integer arrays to floating point."""
-    return array + 0.0
+    if np.issubdtype(array.dtype, np.integer):
+        return array.astype(np.float64)
+
+    assert np.issubdtype(array.dtype, np.floating)
+    return cast(NDArrayFloat, array)
 
 
 def _check_array_key(array: NDArrayAny, key: Any) -> Any:

diff --git a/skfda/representation/basis/_fdatabasis.py b/skfda/representation/basis/_fdatabasis.py
@@ -852,7 +852,7 @@ def _mul_scalar(
     ) -> T:
         """Multiplication by scalar."""
         try:
-            vector = np.atleast_1d(other)
+            vector = _int_to_real(np.atleast_1d(other))
         except Exception:
             return NotImplemented