Upload folder using huggingface_hub

server/finenv_environment.py

@@ -73,7 +73,7 @@ class FinenvEnvironment(Environment):
 
         return FinenvObservation(
             stock=self.stock,
-            market=self.market,
+            market=self.market.value,
             price=self.stock_price,
             shares=self.shares,
             cash=self.cash,
@@ -97,7 +97,7 @@ class FinenvEnvironment(Environment):
         
         if action.type == "init":
             self.stock = action.stock
-            self.market = StockExchangeMarket(action.market.lower())
+            self.market = StockExchangeMarket(action.market.value.lower())
 
             self.initial_cash = action.initial_cash or 10000.0
             self.max_steps = action.max_steps or 100
@@ -122,7 +122,7 @@ class FinenvEnvironment(Environment):
 
             return FinenvObservation(
                 stock=self.stock,
-                market=self.market,
+                market=self.market.value,
                 price=self.stock_price,
                 shares=self.shares,
                 cash=self.cash,
@@ -182,7 +182,7 @@ class FinenvEnvironment(Environment):
         
         return FinenvObservation(
             stock=self.stock,
-            market=self.market,
+            market=self.market.value,
             shares=self.shares,
             cash=self.cash,
             price=self.stock_price,