diff --git a/teleop/__init__.py b/teleop/__init__.py index eb36af4..efd88e6 100644 --- a/teleop/__init__.py +++ b/teleop/__init__.py @@ -4,7 +4,7 @@ import logging from typing import Callable, List import uvicorn -from fastapi import FastAPI, WebSocket, WebSocketDisconnect +from fastapi import APIRouter, FastAPI, WebSocket, WebSocketDisconnect from fastapi.responses import FileResponse from fastapi.staticfiles import StaticFiles import transforms3d as t3d @@ -192,6 +192,9 @@ def __init__( logging.getLogger("uvicorn.access").setLevel(logging.WARNING) self.__setup_routes() + def include_router(self, router: APIRouter, **kwargs) -> None: + self.__app.include_router(router, **kwargs) + def set_pose(self, pose: np.ndarray) -> None: """ Set the current pose of the end-effector.