Update server/finenv_environment.py

server/finenv_environment.py

@@ -21,13 +21,15 @@ from uuid import uuid4
 
 from openenv.core.env_server.interfaces import Environment
 from openenv.core.env_server.types import State
-from reward import reward_message
+
 try:
     from ..models import FinenvAction, FinenvObservation, StockExchangeMarket
     from .stock_price import get_stock_price
+    from .reward import reward_message
 except ImportError:
     from models import FinenvAction, FinenvObservation, StockExchangeMarket
     from stock_price import get_stock_price
+    from reward import reward_message
 
 
 class FinenvEnvironment(Environment):