I'd like to use the debug probe mentioned here but if I try to import the module as shown in the talk I get a syntax error as follows:
from tensorflow-determinism import probe
File "<ipython-input-13-e6323193779e>", line 1
from tensorflow-determinism import probe
^
SyntaxError: invalid syntax
I can't find anything in the repo about it. I'm using TF 2.1.0 and tensorflow-determinism 0.3.0.
What is the cause of this error? or is there a different way I should be accessing the probe?