Spaces:
Sleeping
Sleeping
Commit ·
77c9435
1
Parent(s): 5f214e4
Updated the environments as per the task requirements
Browse files- environment.py +16 -0
environment.py
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from typing import Dict, Any, Tuple
|
| 2 |
+
from models import Action, Observation, Reward
|
| 3 |
+
|
| 4 |
+
|
| 5 |
+
class EmailSortingEnvironment:
|
| 6 |
+
def __init__(self) -> None:
|
| 7 |
+
pass
|
| 8 |
+
|
| 9 |
+
def reset(self, task_id: str = "email_classification") -> Observation:
|
| 10 |
+
pass
|
| 11 |
+
|
| 12 |
+
def step(self, action: Action) -> Tuple[Observation, Reward, bool, Dict]:
|
| 13 |
+
pass
|
| 14 |
+
|
| 15 |
+
def state(self) -> Dict[str, Any]:
|
| 16 |
+
pass
|