Reshape returning NoneType

[ManiSadati]

We've experienced an error during testing a CapsuleNet implementation and it gives this error:
ERROR:root:Encountered exception exceptions.TypeError: injectFaultPack() takes exactly 2 arguments (3 given) [[Node: fi_import/primarycap_reshape/Reshape/shape = PyFunc[Tin=[DT_INT32, DT_INT32, DT_INT32], Tout=[DT_INT32], token="pyfunc_359", _device="/job:localhost/replica:0/task:0/cpu:0"](fi_import/primarycap_reshape/strided_slice, import/primarycap_reshape/Reshape/shape/1, import/primarycap_reshape/Reshape/shape/2)]]

see the tensorboard graph of this part:

the code for this part is
outputs = keras.layers.Reshape(target_shape=[-1, 8], name='primarycap_reshape')(output)

we changed the graph to tensorflow graph in order to run it on session for tensorfi using this.

Is there any way we can solve this problem?

[zitaoc]

Based on the error msg, it seems like the error might come from the injectFaultPack() function within injectFault.py. It says the function is accepting 3 arguments but only 2 are written in the code. Note that this function is NOT implemented correctly and there is FIXME notice in the code. So you'll have to implement this function first before running it through.