search

pytorch / captum

Model interpretability and understanding for PyTorch
https://captum.ai
BSD 3-Clause "New" or "Revised" License
4.95k stars 499 forks source link

Fix pyre error in _generate_baseline_single_dict_feature #1441

Closed jsawruk closed 1 week ago

jsawruk commented 1 week ago

Summary: Fix the pyre error in _generate_baseline_single_dict_feature

The issue was that the device parameter didn't have a type annotation. It appears that the correct type is torch.device.

Also changed the call to .to() by adding the device= named parameter. If a named parameter is not used in this case, torch may assume this is a dtype (see: https://pytorch.org/docs/stable/generated/torch.Tensor.to.html). Adding the named parameter also clarifies the intent.

Reviewed By: jjuncho

Differential Revision: D65914877

facebook-github-bot commented 1 week ago

This pull request was exported from Phabricator. Differential Revision: D65914877

facebook-github-bot commented 1 week ago

This pull request has been merged in pytorch/captum@49b1ed4fd98e491f4b22e39712af3165b0f3dc6e.