Tensorflow and sonnet versions?

[bkj]

I'm trying to run w/ the following versions in python==3.6.5

>>> sonnet.__version__
'1.27'
>>> tensorflow.__version__
'1.12.0'

and getting the following error:

$ python train.py --problem=mnist --save_path=./mnist
WARNING:tensorflow:From /root/.anaconda/lib/python3.6/site-packages/tensorflow/contrib/learn/python/learn/datasets/mnist.py:290: DataSet.__init__ (from tensorflow.contrib.learn.python.learn.datasets.mnist) is deprecated and will be removed in a future version.
Instructions for updating:
Please use alternatives such as official/mnist/dataset.py from tensorflow/models.
Optimizee variables
['mlp/linear_0/w:0', 'mlp/linear_0/b:0', 'mlp/linear_1/w:0', 'mlp/linear_1/b:0']
Problem variables
[]
Traceback (most recent call last):
  File "/root/.anaconda/lib/python3.6/site-packages/tensorflow/python/util/nest.py", line 179, in assert_same_structure
    _pywrap_tensorflow.AssertSameStructure(nest1, nest2, check_types)
TypeError: The two structures don't have the same nested structure.

Presumably this is a version issue -- does anyone know tensorflow/sonnet versions where this repo will run?

Thanks!
~ Ben

[d-li14]

It should work after modifying certain nested structures, refer to #22. @bkj