Upload folder using huggingface_hub

server/finenv_environment.py

@@ -100,7 +100,19 @@ class FinenvEnvironment(Environment):
                 raise ValueError("Stock and market required for init")
             
             self.stock = action.stock
-            self.market = StockExchangeMarket(action.market.lower())
+            market=action.market
+            if isinstance(market, str):
+                market = market.lower()
+                if market == "nse":
+                    self.market = StockExchangeMarket.NSE
+                elif market == "bse":
+                    self.market = StockExchangeMarket.BSE
+                else:
+                    raise ValueError("Unsupported market. Use 'NSE' or 'BSE'.")
+            elif isinstance(market, StockExchangeMarket):
+                self.market = market
+            else:
+                raise ValueError("Invalid market type. Use string or StockExchangeMarket enum.")
 
             
 
@@ -161,7 +173,7 @@ class FinenvEnvironment(Environment):
         # hold → no action
 
        
-        index = self._state.step_count % len(self.price_history)
+        index = min(self._state.step_count, len(self.price_history) - 1)
         self.stock_price = self.price_history[index]
 
         new_value = self.cash + self.shares * self.stock_price