Skip to content

Commit 2a5844a

Browse files
Vivek Miglanifacebook-github-bot
authored andcommitted
Fix utils/model pyre fixme issues (#1606)
Summary: Pull Request resolved: #1606 Fixing unresolved pyre fixme issues in corresponding file Reviewed By: sarahtranfb Differential Revision: D76737258 fbshipit-source-id: 7e6dd861f2707f77e8321360842affd4cbb5e206
1 parent 525e811 commit 2a5844a

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)