Merge pull request #1 from 1024064118/1024064118_bugfix

Create env_stocktrading.py
1024064118 · May 14, 2021 · 3708324 · 3708324
2 parents dc20b0b + c0fe534
commit 3708324
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/finrl/env/env_stocktrading.py b/finrl/env/env_stocktrading.py
@@ -132,7 +132,7 @@ def _buy_stock(self, index, action):
         def _do_buy():
             if self.state[index+1]>0: 
                 #Buy only if the price is > 0 (no missing data in this particular date)       
-                available_amount = self.state[0] // self.state[index+1]
+                available_amount = self.state[0] // ((self.state[index+1])*(1+ self.buy_cost_pct))
                 # print('available_amount:{}'.format(available_amount))
 
                 #update balance
@@ -377,4 +377,4 @@ def _seed(self, seed=None):
     def get_sb_env(self):
         e = DummyVecEnv([lambda: self])
         obs = e.reset()
-        return e, obs
+        return e, obs