Difficulties when trying to import DeepExplain to Google Colab

[SebFAU]

Expected behaviour

When using the command:

!pip install -e git+https://github.com/marcoancona/DeepExplain.git#egg=deepexplain

in the interactive session of Google colab, the installation succeeds and you should be able to import the library via:

from deepexplain.tensorflow import DeepExplain

Actual behaviour

The import fails with a "Module not found" error.
That's why I tried the following workaround after solving the environment on Google's Linux-Server:

import sys; sys.path.append("/content/src/deepexplain"); from deepexplain.tensorflow import DeepExplain

which results in a Key Error.

Backtrace

`KeyError Traceback (most recent call last)

in ()
3 import sys
4 sys.path.append("/content/src/deepexplain")
----> 5 from deepexplain.tensorflow import DeepExplain

3 frames

/content/src/deepexplain/deepexplain/tensorflow/init.py in ()
----> 1 from .methods import DeepExplain

/content/src/deepexplain/deepexplain/tensorflow/methods.py in ()
544
545
--> 546 @ops.RegisterGradient("DeepExplainGrad")
547 def deepexplain_grad(op, grad):
548 global _ENABLED_METHOD_CLASS, _GRAD_OVERRIDE_CHECKFLAG

/usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/ops.py in call(self, f)
2438 def call(self, f):
2439 """Registers the function f as gradient function for op_type."""
-> 2440 _gradient_registry.register(f, self._op_type)
2441 return f
2442

/usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/registry.py in register(self, candidate, name)
59 "Registering two %s with name '%s'! "
60 "(Previous registration was in %s %s:%d)" %
---> 61 (self._name, name, frame.name, frame.filename, frame.lineno))
62
63 logging.vlog(1, "Registering %s (%s) in %s.", name, candidate, self._name)

KeyError: "Registering two gradient with name 'DeepExplainGrad'! (Previous registration was in register /usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/registry.py:66)"`

Do you have any suggestion on how to best use your library in colab ?
Maybe the solution is straight forward and I'm just struggling to see it.

[flol]

I had the same problem and found that restarting the runtime after installing DeepExplain usually helps.

So after connecting to a runtime the first cell I run just installs DeepExplain:

import sys
!{sys.executable} -m pip install -e git+https://github.com/marcoancona/DeepExplain.git#egg=deepexplain

Then restart the runtime via Runtime -> restart runtime. After the runtime comes back just continue with importing DeepExplain. If you run into the error module 'tensorflow' has no attribute 'get_default_session' make sure to force loading Tensorflow 1.x, since Google Colab now uses Tensorflow 2.x by default. So my second cell in the notebook (and the first I run after restarting the runtime) is

%tensorflow_version 1.x
from deepexplain.tensorflow import DeepExplain

[timonmerk]

I have the same error in jupyter notebook. Might be an issue with tensorflow 2