Skip to content

Commit a563c78

Browse files
committed
Add __repr__ for Numba and JAX linkers
1 parent 07b3f8e commit a563c78

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

pytensor/link/jax/linker.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -91,3 +91,6 @@ def create_thunk_inputs(self, storage_map):
9191
thunk_inputs.append(sinput)
9292

9393
return thunk_inputs
94+
95+
def __repr__(self):
96+
return "JAXLinker()"

pytensor/link/numba/linker.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,3 +52,6 @@ def create_thunk_inputs(self, storage_map):
5252
thunk_inputs.append(sinput)
5353

5454
return thunk_inputs
55+
56+
def __repr__(self):
57+
return "NumbaLinker()"

0 commit comments

Comments
 (0)