From 4c73bd3bcde15ab17ddbcaee6770e1dd24629232 Mon Sep 17 00:00:00 2001 From: Aryan Date: Fri, 12 Jan 2024 00:39:25 +0530 Subject: [PATCH] Rectify return type hints in logprob module rewrites --- pymc/logprob/tensor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pymc/logprob/tensor.py b/pymc/logprob/tensor.py index ffc69f88be4..9c8379cdd99 100644 --- a/pymc/logprob/tensor.py +++ b/pymc/logprob/tensor.py @@ -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)