Skip to content

Commit

Permalink
fix: type hint
Browse files Browse the repository at this point in the history
Signed-off-by: Saurav Maheshkar <[email protected]>
  • Loading branch information
SauravMaheshkar committed Jul 25, 2023
1 parent a2dab60 commit 2303b30
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion monai/losses/cldice.py
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ def soft_skel(img: torch.Tensor, iter_: int) -> torch.Tensor:
return skel


def soft_dice(y_true: torch.Tensor, y_pred: torch.Tensor, smooth: int = 1) -> torch.Tensor:
def soft_dice(y_true: torch.Tensor, y_pred: torch.Tensor, smooth: float = 1.0) -> torch.Tensor:
"""
Function to compute soft dice loss
Expand Down

0 comments on commit 2303b30

Please sign in to comment.