fix tests

openAGI · Dec 1, 2021 · 64a8926 · 64a8926
1 parent 93f514d
commit 64a8926
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/__init__.py b/tests/__init__.py
diff --git a/tests/test_loader.py b/tests/test_loader.py
@@ -19,14 +19,14 @@
 from absl.testing import absltest
 
 from datum.reader.loader import load
-from tests import utils
+from tests.utils import _test_create_det_records
 
 
 class TestDataset(absltest.TestCase):
 
   def setUp(self):
     self.tempdir = tempfile.mkdtemp()
-    utils._test_create_det_records(self.tempdir)
+    _test_create_det_records(self.tempdir)
     self._dataset = load(self.tempdir)
     dataset_configs = self._dataset.dataset_configs
     dataset_configs.batch_size_train = 1