Skip to content
Snippets Groups Projects
Commit ea4f6151 authored by Artur Monsch's avatar Artur Monsch
Browse files

fixed failing tests

parent 8076fe56
Branches
No related tags found
1 merge request!7Adding basic functionality to specify the feature axis of the input tensor...
Pipeline #3433 passed
......@@ -255,11 +255,11 @@ class BaseTaylorAnalysis(object):
functions = [self._first_order, self._second_order, self._third_order]
try:
return functions[len(indices) - 1](
x_data=x_data,
node=node,
eval_max_node_only=eval_max_node_only,
forward_kwargs=forward_kwargs,
features_axis=features_axis,
x_data,
node,
eval_max_node_only,
forward_kwargs,
features_axis,
*indices,
)
except KeyError:
......@@ -308,11 +308,11 @@ class BaseTaylorAnalysis(object):
raise ValueError("index_list must be a list of tuples!")
# get TCs
out = self._calculate_tc(
x_data=x_data,
node=node,
eval_max_node_only=eval_max_node_only,
forward_kwargs=forward_kwargs,
features_axis=features_axis,
x_data,
node,
eval_max_node_only,
forward_kwargs,
features_axis,
*ind,
)
# apply reduce function
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment