Upload folder using huggingface_hub

models.py

@@ -11,29 +11,15 @@ from typing import Any, Dict, Optional
 
 from openenv.core.env_server.types import Action, Observation
 
-TASKS = {
-    "easy": {
-        "goal": "Make profit in 5 steps",
-        "target_profit": 0.01
-    },
-    "medium": {
-        "goal": "Maximize portfolio value",
-        "target_profit": 0.05
-    },
-    "hard": {
-        "goal": "Achieve 10% return",
-        "target_return": 0.1
-    }
-}
+from typing import Optional, Literal
+
+
 
 class StockExchangeMarket(Enum):
     BSE = "bse"
     NSE = "nse"
 
-class task_type(str, Enum):
-    Easy = "easy"
-    Medium = "medium"
-    Hard = "hard"
+
 
 class FinenvAction(Action):
     """
@@ -51,7 +37,7 @@ class FinenvAction(Action):
     # Trading fields
     quantity: Optional[int] = 0
 
-    task_type: Optional[task_type] = None
+    task_type: Optional[Literal["easy", "medium", "hard"]] = None
 
     # Initialization fields
     stock: Optional[str] = None