Skip to content

Commit

Permalink
Rectify return type hints in logprob module rewrites
Browse files Browse the repository at this point in the history
  • Loading branch information
AryanNanda17 committed Jan 11, 2024
1 parent 1cc27bc commit 4c73bd3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pymc/logprob/tensor.py
Original file line number Diff line number Diff line change
Expand Up @@ -199,7 +199,7 @@ def logprob_join(op, values, axis, *base_rvs, **kwargs):
@node_rewriter([MakeVector, Join])
def find_measurable_stacks(
fgraph, node
) -> Optional[List[TensorVariable, TensorVariable]]:
) -> Optional[List[TensorVariable]]:
r"""Finds `Joins`\s and `MakeVector`\s for which a `logprob` can be computed."""

rv_map_feature: Optional[PreserveRVMappings] = getattr(fgraph, "preserve_rv_mappings", None)
Expand Down

0 comments on commit 4c73bd3

Please sign in to comment.