unexpected argument error in graph_parser_network.py

[aminianmary]

Hello,

Thanks for providing the Parser. I'm trying to train an SDP model using your parser but keep getting the following error message:

Traceback (most recent call last):
  File "main.py", line 284, in <module>
    main()
  File "main.py", line 107, in main
    kwargs.pop('action')(**kwargs)
  File "main.py", line 169, in train
    network.train(load=load, noscreen=noscreen)
  File "/home/aminian/Parser-v3/parser/base_network.py", line 141, in train
    train_graph = self.build_graph(input_network_outputs=input_network_outputs, reuse=False)
  File "/home/aminian/Parser-v3/parser/graph_parser_network.py", line 64, in build_graph
    n_nonzero = tf.to_float(tf.count_nonzero(layer, axis=-1, keepdims=True))
TypeError: count_nonzero() got an unexpected keyword argument 'keepd

I'm using the DM18.cfg file with slight changes in the directory paths. Any thought to resolve the issue is appreciated.

Thanks