Fix import mistake.

google · Dec 23, 2020 · 53753bb · 53753bb
1 parent e419394
commit 53753bb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/efficientdet/dataloader_test.py b/efficientdet/dataloader_test.py
@@ -18,7 +18,7 @@
 
 import dataloader
 import hparams_config
-from brain_automl.efficientdet import test_util
+import test_util
 
 from keras import anchors
 from object_detection import tf_example_decoder

diff --git a/efficientdet/keras/inference_test.py b/efficientdet/keras/inference_test.py
@@ -17,7 +17,7 @@
 import tempfile
 from absl import logging
 import tensorflow as tf
-from brain_automl.efficientdet import test_util
+import test_util
 from keras import efficientdet_keras
 from keras import inference