Skip to content

Commit 7f32c30

Browse files
Add type hints for rv arguments
Co-authored-by: Ricardo Vieira <[email protected]>
1 parent e8571b2 commit 7f32c30

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pymc/distributions/logprob.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -290,14 +290,14 @@ def logp(rv: TensorVariable, value) -> TensorVariable:
290290
raise NotImplementedError("PyMC could not infer logp of input variable.") from exc
291291

292292

293-
def logcdf(rv, value) -> TensorVariable:
293+
def logcdf(rv: TensorVariable, value) -> TensorVariable:
294294
"""Return the log-cdf graph of a Random Variable"""
295295

296296
value = at.as_tensor_variable(value, dtype=rv.dtype)
297297
return logcdf_aeppl(rv, value)
298298

299299

300-
def ignore_logprob(rv) -> TensorVariable:
300+
def ignore_logprob(rv: TensorVariable) -> TensorVariable:
301301
"""Return a duplicated variable that is ignored when creating Aeppl logprob graphs
302302
303303
This is used in SymbolicDistributions that use other RVs as inputs but account

0 commit comments

Comments
 (0)