Skip to content

Commit

Permalink
Fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
CokeDong committed Oct 10, 2022
1 parent 9b626ec commit 0847982
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions mmcv/ops/prroi_pool.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,10 +33,9 @@ def forward(ctx,
output_size: Tuple,
spatial_scale: float = 1.0) -> torch.Tensor:
if features.dtype() != torch.float or rois.dtype() != torch.float:
raise ValueError(
'Precise RoI Pooling only takes float input, got '
f'{features.dtype()} for features and'
f'{rois.dtype()} for rois.')
raise ValueError('Precise RoI Pooling only takes float input, got '
f'{features.dtype()} for features and'
f'{rois.dtype()} for rois.')

pooled_height = int(output_size[0])
pooled_width = int(output_size[1])
Expand Down

0 comments on commit 0847982

Please sign in to comment.