fix(Classifier): Removed deprecated two_dim parameter from unittesting

frankkramer-lab · Jun 8, 2022 · c8ce20d · c8ce20d
1 parent d0544a3
commit c8ce20d
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/test_architectures_classifier.py b/tests/test_architectures_classifier.py
@@ -136,8 +136,7 @@ def test_functionality_base(self):
                                          activation_output="softmax")
         model_input = Input(shape=(32,32,3))
         model = classification_head.build(model_input=model_input,
-                                          model_output=model_input,
-                                          two_dim=True)
+                                          model_output=model_input)
         try : model.summary()
         except : raise Exception()
 
@@ -150,7 +149,6 @@ def test_functionality_metadata(self):
                                          meta_variables=10)
         model_input = Input(shape=(32,32,3))
         model = classification_head.build(model_input=model_input,
-                                          model_output=model_input,
-                                          two_dim=True)
+                                          model_output=model_input)
         try : model.summary()
         except : raise Exception()