minor: fix linting

mrava87 · Aug 9, 2024 · 054c5b8 · 054c5b8
1 parent 310b2a9
commit 054c5b8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pylops/jaxoperator.py b/pylops/jaxoperator.py
@@ -18,8 +18,8 @@
         'the jaxoperator module run "pip install jax" or'
         '"conda install -c conda-forge jax".'
     )
-    jaxarrayin_type = None
-    jaxarrayout_type = None
+    jaxarrayin_type = Any
+    jaxarrayout_type = Any
 
 JaxTypeIn = NewType("JaxTypeIn", jaxarrayin_type)
 JaxTypeOut = NewType("JaxTypeOut", jaxarrayout_type)