torch-agent/torch_agent/__main__.py

import sys
from torch_agent import main

if __name__ == '__main__':
  sys.exit(main())