Skip to content

Commit eb8ea76

Browse files
Vivek Miglanifacebook-github-bot
authored andcommitted
Fix utils/model pyre fixme issues
Summary: Fixing unresolved pyre fixme issues in corresponding file Differential Revision: D76737258
1 parent 5d223d6 commit eb8ea76

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

captum/_utils/models/model.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,7 @@ class Model(ABC):
2222
def fit(
2323
self,
2424
train_data: DataLoader,
25-
# pyre-fixme[2]: Parameter must be annotated.
26-
**kwargs,
25+
**kwargs: object,
2726
) -> Optional[Dict[str, Union[int, float, Tensor]]]:
2827
r"""
2928
Override this method to actually train your model.

0 commit comments

Comments
 (0)