burtenshaw HF Staff commited on
Commit
2c1bf6b
·
verified ·
1 Parent(s): 488af5f

Upload folder using huggingface_hub

Browse files
src/envs/wildfire_env/server/app.py CHANGED
@@ -1,10 +1,10 @@
1
  # server/app.py
2
  import os
3
- from core.env_server import create_fastapi_app
4
  from ..models import WildfireAction, WildfireObservation
5
  from .wildfire_environment import WildfireEnvironment
6
 
7
  W = int(os.getenv("WILDFIRE_WIDTH", "16"))
8
  H = int(os.getenv("WILDFIRE_HEIGHT", "16"))
9
  env = WildfireEnvironment(width=W, height=H)
10
- app = create_fastapi_app(env, WildfireAction, WildfireObservation)
 
1
  # server/app.py
2
  import os
3
+ from core.env_server import create_app
4
  from ..models import WildfireAction, WildfireObservation
5
  from .wildfire_environment import WildfireEnvironment
6
 
7
  W = int(os.getenv("WILDFIRE_WIDTH", "16"))
8
  H = int(os.getenv("WILDFIRE_HEIGHT", "16"))
9
  env = WildfireEnvironment(width=W, height=H)
10
+ app = create_app(env, WildfireAction, WildfireObservation, env_name='wildfire_env')