Create launch_gym_server.py

Author: TarrySingh

deep-learning/Deep-Reinforcement-Learning-Complete-Collection/PyTorch-cpp/launch_gym_server.py

@@ -0,0 +1,41 @@
+#!/usr/bin/env python
+"""
+Pytorch-cpp-rl OpenAI gym server main script.
+"""
+import logging
+
+from gym_server.server import Server
+from gym_server.zmq_client import ZmqClient
+
+
+def main():
+    """
+    Host the server.
+    """
+    # If anything is logged during imports, it messes up our logging so we
+    # reset the logging module here
+    root_logger = logging.getLogger()
+    if root_logger.handlers:
+        for handler in root_logger.handlers:
+            root_logger.removeHandler(handler)
+    logging.basicConfig(level=logging.DEBUG,
+                        format=('%(asctime)s %(funcName)s '
+                                '[%(levelname)s]: %(message)s'),
+                        datefmt='%Y%m%d %H:%M:%S')
+    logging.info("Initializing gym server")
+
+    zmq_client = ZmqClient(10201)
+    logging.info("Connecting to client")
+    zmq_client.send("Connection established")
+    logging.info("Connected")
+    server = Server(zmq_client)
+
+    try:
+        server.serve()
+    except:  # pylint: disable=bare-except
+        import pdb
+        pdb.post_mortem()
+
+
+if __name__ == '__main__':
+    main()