Attribute error when trying to visualize RL-optimized custom network

[wjdunlop]

Bug Description

Tried to use the "visualizer_rllib.py" from tutorial 4 to visualize a rl-optimized custom network. Received an an AttributeError with the name of the class I defined the custom network in.

Bug Reproduce

File in which the bug occurred: visualizer_rllib.py

The exact command that you did run that caused the bug:
python visualizer_rllib.py /path/to/my/checkpoint

Output Log

Full error output log:

Traceback (most recent call last):
  File "../flow/visualize/visualizer_rllib.py", line 386, in <module>
    visualizer_rllib(args)
  File "../flow/visualize/visualizer_rllib.py", line 70, in visualizer_rllib
    flow_params = get_flow_params(config)
  File "/home/rrishi/flow/flow/utils/rllib.py", line 181, in get_flow_params
    net_instance = getattr(net_module, network)
AttributeError: module '__main__' has no attribute 'myNetwork'