Uncomment other tests

nicolay-r · Aug 15, 2021 · 2b12ce7 · 2b12ce7
1 parent 8434e3c
commit 2b12ce7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/contrib/networks/test_tf_mi_feed.py b/tests/contrib/networks/test_tf_mi_feed.py
@@ -5,10 +5,12 @@
 
 sys.path.append('../../../')
 
-from tests.contrib.networks.labels import TestNeutralLabel, TestThreeLabelScaler
+from tests.contrib.networks.labels import TestThreeLabelScaler
 from tests.contrib.networks.test_tf_ctx_feed import TestContextNetworkFeeding
 from tests.contrib.networks.tf_networks.supported import get_supported
 
+from arekit.contrib.networks.multi.configurations.att_self import AttSelfOverSentencesConfig
+from arekit.contrib.networks.multi.architectures.att_self import AttSelfOverSentences
 from arekit.contrib.networks.core.feeding.bags.bag import Bag
 from arekit.contrib.networks.core.feeding.batch.multi import MultiInstanceMiniBatch
 from arekit.contrib.networks.multi.configurations.max_pooling import MaxPoolingOverSentencesConfig
@@ -42,7 +44,7 @@ def __create_minibatch(config, labels_scaler):
     def multiinstances_supported(ctx_config, ctx_network):
         return [
             (MaxPoolingOverSentencesConfig(ctx_config), MaxPoolingOverSentences(ctx_network)),
-            # (AttSelfOverSentencesConfig(ctx_config), AttSelfOverSentences(ctx_network))
+            (AttSelfOverSentencesConfig(ctx_config), AttSelfOverSentences(ctx_network))
         ]
 
     def test(self):