Remove top-level imports in tests

This is a pre-requisite for cleaning up our public API.
kiudee · Apr 17, 2020 · 719cc7a · 719cc7a
1 parent 006d227
commit 719cc7a
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/csrank/tests/test_fate.py b/csrank/tests/test_fate.py
@@ -5,8 +5,8 @@
 from keras.regularizers import l2
 import numpy as np
 
-from csrank import FATENetworkCore
-from csrank import FATEObjectRanker
+from csrank.core import FATENetworkCore
+from csrank.objectranking import FATEObjectRanker
 from csrank.tests.test_ranking import optimizer
 
 

diff --git a/csrank/tests/test_ranking.py b/csrank/tests/test_ranking.py
@@ -5,7 +5,6 @@
 import pytest
 import tensorflow as tf
 
-from csrank import PairedCombinatorialLogit
 from csrank.constants import CMPNET
 from csrank.constants import ERR
 from csrank.constants import FATE_RANKER
@@ -15,6 +14,7 @@
 from csrank.constants import LISTNET
 from csrank.constants import RANKNET
 from csrank.constants import RANKSVM
+from csrank.discretechoice import PairedCombinatorialLogit
 from csrank.metrics_np import zero_one_accuracy_np
 from csrank.metrics_np import zero_one_rank_loss_for_scores_ties_np
 from csrank.objectranking import *

diff --git a/csrank/tests/test_util.py b/csrank/tests/test_util.py
@@ -3,7 +3,7 @@
 import pytest
 import tensorflow as tf
 
-from csrank import SyntheticIterator
+from csrank.dataset_reader import SyntheticIterator
 from csrank.tensorflow_util import tensorify
 from csrank.tuning import check_learner_class