Skip to content

Commit

Permalink
updated tests
Browse files Browse the repository at this point in the history
Signed-off-by: Jakub Mitura <[email protected]>
  • Loading branch information
jakubMitura14 committed Apr 13, 2022
1 parent 5606491 commit 53d75bb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_hausdorff_distance_morphologic.py
Original file line number Diff line number Diff line change
Expand Up @@ -123,7 +123,7 @@ def test_value(self, input_data, expected_value):
if(not version_leq(f"{torch.version.cuda}", "10.100") and not version_leq(f"{torch.version.cuda}", "10.200")):
[y_pred, y, percentt, compare_values] = input_data
hd_metric = MorphologicalHausdorffDistanceMetric(
compare_values.to(device), percentt, True
compare_values.to(device), percentt, False
) # True only for tests
result = hd_metric._compute_tensor(y_pred.to(device), y.to(device))
np.testing.assert_allclose(expected_value, result, rtol=1e-7)
Expand Down

0 comments on commit 53d75bb

Please sign in to comment.