Use idiomatic numpy imports

Author: alexfikl

jitcode/_jitcode.py

@@ -6,7 +6,6 @@
 
 import numpy as np
 import symengine
-from numpy import hstack, log
 
 from jitcxde_common import checker, jitcxde
 from jitcxde_common.helpers import find_dependent_helpers, sort_helpers, sympify_helpers
@@ -729,7 +728,7 @@ def set_initial_value(self, y, time=0.0):
 		for _ in range(self._n_lyap):
 			new_y.append(random_direction(self.n_basic))
 
-		super().set_initial_value(hstack(new_y), time)
+		super().set_initial_value(np.hstack(new_y), time)
 
 	def set_integrator(self,name,interpolate=None,**kwargs):
 		"""
@@ -769,7 +768,7 @@ def integrate(self, *args, **kwargs):
 		super().integrate(*args, **kwargs)
 		delta_t = self.t-old_t
 		norms, tangent_vectors = self.norms()
-		lyaps = log(norms) / delta_t
+		lyaps = np.log(norms) / delta_t
 		self._y[self.n_basic:] = tangent_vectors.flatten()
 		super().set_initial_value(self._y, self.t)
 
@@ -910,7 +909,7 @@ def integrate(self, *args, **kwargs):
 		super().integrate(*args, **kwargs)
 		delta_t = self.t-old_t
 		norm = self.norm()
-		lyap = log(norm) / delta_t
+		lyap = np.log(norm) / delta_t
 		super().set_initial_value(self._y, self.t)
 
 		return self._y[self.main_indices], lyap

jitcode/integrator_tools.py

@@ -1,6 +1,6 @@
 from inspect import signature
 
-from numpy import inf
+import numpy as np
 from scipy.integrate import ode
 from scipy.integrate._ivp.ivp import METHODS as ivp_methods
 from scipy.integrate._ode import find_integrator
@@ -57,7 +57,7 @@ def __init__(
 
 		# Dictionary to be passed as arguments to the integrator and store stuff
 		self.kwargs = {
-				"t_bound": inf,
+				"t_bound": np.inf,
 				"vectorized": False,
 				"fun": self.f
 			}