We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f65dc90 commit b66929fCopy full SHA for b66929f
pymc/data.py
@@ -132,7 +132,7 @@ class MinibatchIndexRV(IntegersRV):
132
minibatch_index = MinibatchIndexRV()
133
134
135
-def is_minibatch(v):
+def is_minibatch(v: TensorVariable) -> bool:
136
from aesara.tensor.subtensor import AdvancedSubtensor
137
138
return (
@@ -142,7 +142,7 @@ def is_minibatch(v):
142
)
143
144
145
-def valid_for_minibatch(v):
+def valid_for_minibatch(v: TensorVariable) -> bool:
146
from aesara.scalar import Cast
147
from aesara.tensor.elemwise import Elemwise
148
0 commit comments