Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion src/openenv/cli/templates/openenv_env/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@


class __ENV_CLASS_NAME__Env(
EnvClient[__ENV_CLASS_NAME__Action, __ENV_CLASS_NAME__Observation]
EnvClient[__ENV_CLASS_NAME__Action, __ENV_CLASS_NAME__Observation, State]
):
"""
Client for the __ENV_TITLE_NAME__ Environment.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,10 +13,14 @@

from uuid import uuid4

from models import __ENV_CLASS_NAME__Action, __ENV_CLASS_NAME__Observation
from openenv.core.env_server.interfaces import Environment
from openenv.core.env_server.types import State

try:
from ..models import __ENV_CLASS_NAME__Action, __ENV_CLASS_NAME__Observation
except ImportError:
from models import __ENV_CLASS_NAME__Action, __ENV_CLASS_NAME__Observation


class __ENV_CLASS_NAME__Environment(Environment):
"""
Expand Down
10 changes: 6 additions & 4 deletions src/openenv/cli/templates/openenv_env/server/app.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,10 +35,12 @@
"openenv is required for the web interface. Install dependencies with '\n uv sync\n'"
) from e

# Import from local models.py (PYTHONPATH includes /app/env in Docker)
from models import __ENV_CLASS_NAME__Action, __ENV_CLASS_NAME__Observation

from .__ENV_NAME___environment import __ENV_CLASS_NAME__Environment
try:
from ..models import __ENV_CLASS_NAME__Action, __ENV_CLASS_NAME__Observation
from .__ENV_NAME___environment import __ENV_CLASS_NAME__Environment
except ModuleNotFoundError:
from models import __ENV_CLASS_NAME__Action, __ENV_CLASS_NAME__Observation
from server.__ENV_NAME___environment import __ENV_CLASS_NAME__Environment


# Create the app with web interface and README integration
Expand Down
Loading