minor: fix jax import in test_jaxoperator

Author: mrava87

pytests/test_jaxoperator.py

@@ -7,7 +7,7 @@
 from pylops import JaxOperator, MatrixMult
 from pylops.utils import deps
 
-jax_message = deps.devito_import("the jax module")
+jax_message = deps.jax_import("the jax module")
 
 if jax_message is None:
     import jax