Add __repr__ for Numba and JAX linkers

ricardoV94 · ricardoV94 · commit a563c7845e2b · 2024-06-07T19:24:18.000+02:00
diff --git a/pytensor/link/jax/linker.py b/pytensor/link/jax/linker.py
@@ -91,3 +91,6 @@ def create_thunk_inputs(self, storage_map):
             thunk_inputs.append(sinput)
 
         return thunk_inputs
+
+    def __repr__(self):
+        return "JAXLinker()"
diff --git a/pytensor/link/numba/linker.py b/pytensor/link/numba/linker.py
@@ -52,3 +52,6 @@ def create_thunk_inputs(self, storage_map):
             thunk_inputs.append(sinput)
 
         return thunk_inputs
+
+    def __repr__(self):
+        return "NumbaLinker()"
