UI module for Icon

atst/app.py

@@ -15,7 +15,7 @@ from atst.handlers.dev import Dev
 from atst.home import home
 from atst.api_client import ApiClient
 from atst.sessions import RedisSessions
-from atst.ui import UI
+from atst import ui_modules
 
 ENV = os.getenv("TORNADO_ENV", "dev")
 
@@ -100,11 +100,11 @@ def make_app(config, deps, **kwargs):
         static_path=home.child("static"),
         cookie_secret=config["default"]["COOKIE_SECRET"],
         debug=config["default"].getboolean("DEBUG"),
+        ui_modules=ui_modules,
         **kwargs,
     )
     app.config = config
     app.sessions = deps["sessions"]
-    app.ui = UI
     return app