Skip to content

Commit 86db600

Browse files
craymichaelfacebook-github-bot
authored andcommitted
Fix typing error in stochastic gates base (#1584)
Summary: Pull Request resolved: #1584 Fixes https://github.com/pytorch/captum/actions/runs/15593782761/job/43919109271 Reviewed By: sarahtranfb Differential Revision: D76604756 fbshipit-source-id: 94fa39c4c0eef6cd69cacca4de87b50d306204e1
1 parent e3f3330 commit 86db600

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

captum/module/stochastic_gates_base.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,8 @@ class StochasticGatesBase(Module, ABC):
3030
extend this class and implement the distribution specific functions.
3131
"""
3232

33+
mask: Optional[Tensor]
34+
3335
def __init__(
3436
self,
3537
n_gates: int,

0 commit comments

Comments
 (0)