port functional and consistency tests

pmeier · Feb 17, 2023 · 085097e · 085097e
1 parent a9448c3
commit 085097e
Show file tree

Hide file tree

Showing 11 changed files with 693 additions and 718 deletions.
diff --git a/test/builtin_dataset_mocks.py b/test/builtin_dataset_mocks.py
@@ -18,7 +18,8 @@
 import numpy as np
 import pytest
 import torch
-from datasets_utils import combinations_grid, create_image_file, create_image_folder, make_tar, make_zip
+from common_utils import combinations_grid
+from datasets_utils import create_image_file, create_image_folder, make_tar, make_zip
 from torch.nn.functional import one_hot
 from torch.testing import make_tensor as _make_tensor
 from torchvision.prototype import datasets