Already merged, but this type-hint is wrong: It should be TensorVariable _Originally posted by @ricardoV94 in https://github.com/pymc-labs/pymc-marketing/pull/397#discussion_r1413865810_