Skip to content

Commit

Permalink
Remove TODO comment.
Browse files Browse the repository at this point in the history
Nothing clever was going on, unknown dims were simply faked as having length 8 for JAX polymorphism export
  • Loading branch information
ricardoV94 committed Feb 3, 2025
1 parent c4a3444 commit 8bb2038
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions pytensor/tensor/einsum.py
Original file line number Diff line number Diff line change
Expand Up @@ -546,8 +546,6 @@ def einsum(subscripts: str, *operands: "TensorLike", optimize=None) -> TensorVar
"If you need this functionality open an issue in https://github.com/pymc-devs/pytensor/issues to let us know. "
)

# TODO: Is this doing something clever about unknown shapes?
# contract_path = _poly_einsum_handlers.get(ty, _default_poly_einsum_handler)
tensor_operands = [as_tensor(operand) for operand in operands]
shapes = [operand.type.shape for operand in tensor_operands]

Expand Down

0 comments on commit 8bb2038

Please sign in to comment.