Skip to content

Commit b66929f

Browse files
author
Maxim Kochurov
committed
add more type hints
1 parent f65dc90 commit b66929f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pymc/data.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,7 @@ class MinibatchIndexRV(IntegersRV):
132132
minibatch_index = MinibatchIndexRV()
133133

134134

135-
def is_minibatch(v):
135+
def is_minibatch(v: TensorVariable) -> bool:
136136
from aesara.tensor.subtensor import AdvancedSubtensor
137137

138138
return (
@@ -142,7 +142,7 @@ def is_minibatch(v):
142142
)
143143

144144

145-
def valid_for_minibatch(v):
145+
def valid_for_minibatch(v: TensorVariable) -> bool:
146146
from aesara.scalar import Cast
147147
from aesara.tensor.elemwise import Elemwise
148148

0 commit comments

Comments
 (0)