Skip to content

Commit a77c9ee

Browse files
committed
Add __repr__ for Numba and JAX linkers
1 parent 0b56ed9 commit a77c9ee

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
@@ -87,3 +87,6 @@ def create_thunk_inputs(self, storage_map):
8787
thunk_inputs.append(sinput)
8888

8989
return thunk_inputs
90+
91+
def __repr__(self):
92+
return "JAXLinker()"

pytensor/link/numba/linker.py

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

3737
return thunk_inputs
38+
39+
def __repr__(self):
40+
return "NumbaLinker()"

0 commit comments

Comments
 (0)