Upload folder using huggingface_hub
Browse files- .gitignore +9 -0
- Dockerfile +3 -5
- README.md +2 -2
- __init__.py +2 -0
- client.py +6 -0
- models.py +23 -0
- openenv.yaml +8 -0
- server/Dockerfile +10 -0
- server/__init__.py +2 -0
- server/app.py +10 -0
- server/requirements.txt +4 -0
.gitignore
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
__pycache__/
|
| 2 |
+
*.pyc
|
| 3 |
+
.venv/
|
| 4 |
+
env/
|
| 5 |
+
.env
|
| 6 |
+
hf-staging/
|
| 7 |
+
.DS_Store
|
| 8 |
+
|
| 9 |
+
|
Dockerfile
CHANGED
|
@@ -3,13 +3,11 @@ FROM ghcr.io/meta-pytorch/openenv-base:latest
|
|
| 3 |
|
| 4 |
WORKDIR /app
|
| 5 |
|
|
|
|
| 6 |
|
| 7 |
-
# Copy only what's needed for this environment
|
| 8 |
-
COPY src/core/ /app/src/core/
|
| 9 |
-
COPY src/envs/my_env/ /app/src/envs/my_env/
|
| 10 |
ENV ENABLE_WEB_INTERFACE=true
|
| 11 |
-
ENV PYTHONPATH=/app
|
| 12 |
|
| 13 |
-
CMD ["uvicorn", "
|
| 14 |
|
| 15 |
|
|
|
|
| 3 |
|
| 4 |
WORKDIR /app
|
| 5 |
|
| 6 |
+
COPY . /app
|
| 7 |
|
|
|
|
|
|
|
|
|
|
| 8 |
ENV ENABLE_WEB_INTERFACE=true
|
| 9 |
+
ENV PYTHONPATH=/app
|
| 10 |
|
| 11 |
+
CMD ["uvicorn", "server.app:app", "--host", "0.0.0.0", "--port", "8000"]
|
| 12 |
|
| 13 |
|
README.md
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
---
|
| 2 |
title: My_env Environment Server
|
| 3 |
-
emoji:
|
| 4 |
-
colorFrom:
|
| 5 |
colorTo: indigo
|
| 6 |
sdk: docker
|
| 7 |
pinned: false
|
|
|
|
| 1 |
---
|
| 2 |
title: My_env Environment Server
|
| 3 |
+
emoji: 🦛
|
| 4 |
+
colorFrom: green
|
| 5 |
colorTo: indigo
|
| 6 |
sdk: docker
|
| 7 |
pinned: false
|
__init__.py
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
|
client.py
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""Client stub for my_env environment."""
|
| 2 |
+
|
| 3 |
+
class my_envEnv:
|
| 4 |
+
pass
|
| 5 |
+
|
| 6 |
+
|
models.py
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from dataclasses import dataclass, field
|
| 2 |
+
from typing import Any
|
| 3 |
+
|
| 4 |
+
|
| 5 |
+
@dataclass
|
| 6 |
+
class Action:
|
| 7 |
+
"""Base Action for my_env."""
|
| 8 |
+
pass
|
| 9 |
+
|
| 10 |
+
|
| 11 |
+
@dataclass
|
| 12 |
+
class Observation:
|
| 13 |
+
"""Base Observation for my_env."""
|
| 14 |
+
message: str = ""
|
| 15 |
+
|
| 16 |
+
|
| 17 |
+
@dataclass
|
| 18 |
+
class State:
|
| 19 |
+
"""Episode state."""
|
| 20 |
+
episode_id: str = ""
|
| 21 |
+
step_count: int = 0
|
| 22 |
+
|
| 23 |
+
|
openenv.yaml
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
spec_version: 1
|
| 2 |
+
name: my_env
|
| 3 |
+
type: space
|
| 4 |
+
runtime: fastapi
|
| 5 |
+
app: envs.my_env.server.app:app
|
| 6 |
+
port: 8000
|
| 7 |
+
|
| 8 |
+
|
server/Dockerfile
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
ARG BASE_IMAGE=ghcr.io/meta-pytorch/openenv-base:latest
|
| 2 |
+
FROM ${BASE_IMAGE}
|
| 3 |
+
|
| 4 |
+
WORKDIR /app
|
| 5 |
+
|
| 6 |
+
COPY . /app
|
| 7 |
+
|
| 8 |
+
CMD ["uvicorn", "envs.my_env.server.app:app", "--host", "0.0.0.0", "--port", "8000"]
|
| 9 |
+
|
| 10 |
+
|
server/__init__.py
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
|
server/app.py
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from fastapi import FastAPI
|
| 2 |
+
|
| 3 |
+
app = FastAPI()
|
| 4 |
+
|
| 5 |
+
|
| 6 |
+
@app.get("/health")
|
| 7 |
+
def health() -> dict:
|
| 8 |
+
return {"status": "healthy"}
|
| 9 |
+
|
| 10 |
+
|
server/requirements.txt
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
fastapi
|
| 2 |
+
uvicorn
|
| 3 |
+
|
| 4 |
+
|