Skip to content

GitLab

Commit 2234877e authored by Patrick von Platen's avatar Patrick von Platen
Browse files

fix tests

parent 7ba3130c
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_modeling_utils.py
View file @ 2234877e
...@@ -309,5 +309,5 @@ class PipelineTesterMixin(unittest.TestCase): ...@@ -309,5 +309,5 @@ class PipelineTesterMixin(unittest.TestCase):
image_slice = image[0, -1, -3:, -3:].cpu() image_slice = image[0, -1, -3:, -3:].cpu()
assert image.shape == (1, 3, 32, 32) assert image.shape == (1, 3, 32, 32)
expected_slice = torch.tensor([-0.7688, -0.7690, -0.7597, -0.7660, -0.7713, -0.7531, -0.7009, -0.7098, -0.7350]) expected_slice = torch.tensor([-0.7383, -0.7385, -0.7298, -0.7364, -0.7414, -0.7239, -0.6737, -0.6813, -0.7068])
assert (image_slice.flatten() - expected_slice).abs().max() < 1e-2 assert (image_slice.flatten() - expected_slice).abs().max() < 1e-2
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment