Remove TODO comment.

ricardoV94 · ricardoV94 · commit 8bb2038ddfbd · 2025-02-03T19:06:21.000+01:00
Nothing clever was going on, unknown dims were simply faked as having length 8 for JAX polymorphism export
diff --git a/pytensor/tensor/einsum.py b/pytensor/tensor/einsum.py
@@ -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]
 

Original file line number	Diff line number	Diff line change
`@@ -546,8 +546,6 @@ def einsum(subscripts: str, *operands: "TensorLike", optimize=None) -> TensorVar`
`546`	`546`	`"If you need this functionality open an issue in https://github.com/pymc-devs/pytensor/issues to let us know. "`
`547`	`547`	`)`
`548`	`548`
`549`		`- # TODO: Is this doing something clever about unknown shapes?`
`550`		`- # contract_path = _poly_einsum_handlers.get(ty, _default_poly_einsum_handler)`
`551`	`549`	`tensor_operands = [as_tensor(operand) for operand in operands]`
`552`	`550`	`shapes = [operand.type.shape for operand in tensor_operands]`
`553`	`551`