jampuramprem commited on
Commit
77c9435
·
1 Parent(s): 5f214e4

Updated the environments as per the task requirements

Browse files
Files changed (1) hide show
  1. 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