Skip to content

Commit

Permalink
Set dtype to float32 in reference metric
Browse files Browse the repository at this point in the history
  • Loading branch information
baskrahmer committed Sep 24, 2024
1 parent 8541d97 commit 90ad414
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/unittests/segmentation/test_hausdorff_distance.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@
# Wrapper that converts to numpy to avoid Torch-to-numpy functional issues
def torch_skimage_hausdorff_distance(p: torch.Tensor, t: torch.Tensor) -> torch.Tensor:
out = skimage_hausdorff_distance(p.numpy(), t.numpy())
return torch.tensor([out])
return torch.tensor([out], dtype=torch.float32)


@pytest.mark.parametrize(
Expand Down

0 comments on commit 90ad414

Please sign in to comment.