adapt tests

tests/test_model_extention.py

@@ -59,7 +59,8 @@ class TestBaseClass:
                 for node in range(mlp_specs["output_neurons"]):
                     # get singe node results
                     tc = self.model.get_tc(
-                        x_data=x_data,
+                        "x",
+                        forward_kwargs={"x": x_data},
                         index_list=[index],
                         node=node,
                         eval_max_node_only=eval_max_node_only,
@@ -68,7 +69,8 @@ class TestBaseClass:
                     node_outputs.append(tc)
                 # get all-node results
                 tc = self.model.get_tc(
-                    x_data=x_data,
+                    "x",
+                    forward_kwargs={"x": x_data},
                     index_list=[index],
                     node=None,
                     eval_max_node_only=eval_max_node_only,

tests/test_tc_computation.py

@@ -65,7 +65,8 @@ class TestTCComputation(unittest.TestCase):
             ):
                 # compute TC
                 tc = _model.get_tc(
-                    x_data=_point,
+                    "point",
+                    forward_kwargs={"point": _point},
                     index_list=[combination],
                     features_axis=_feature_axis,
                 )