File size: 1,044 Bytes
67f71c2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 | """
FastAPI application for the shader environment.
Usage:
uvicorn server.app:app --reload --host 0.0.0.0 --port 8000
"""
try:
from openenv.core.env_server.http_server import create_app
except Exception as e:
raise ImportError(
"openenv-core is required. Install with: pip install openenv-core[core]"
) from e
try:
from ..models import ShaderAction, ShaderObservation
from .environment import ShaderEnvironment
except ImportError:
from models import ShaderAction, ShaderObservation
from server.environment import ShaderEnvironment
app = create_app(
ShaderEnvironment,
ShaderAction,
ShaderObservation,
env_name="shader",
max_concurrent_envs=4,
)
def main() -> None:
import argparse
import uvicorn
parser = argparse.ArgumentParser()
parser.add_argument("--host", default="0.0.0.0")
parser.add_argument("--port", type=int, default=8000)
args = parser.parse_args()
uvicorn.run(app, host=args.host, port=args.port)
if __name__ == "__main__":
main()
|