From 054c5b8a07756650004a4e609a63ab9455fdcab1 Mon Sep 17 00:00:00 2001
From: mrava87 <matteoravasi@gmail.com>
Date: Fri, 9 Aug 2024 23:00:24 +0200
Subject: [PATCH] minor: fix linting

---
 pylops/jaxoperator.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pylops/jaxoperator.py b/pylops/jaxoperator.py
index bf00a604..5d5c40ed 100644
--- 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)