ositamiles commited on
Commit
06fb4f5
·
verified ·
1 Parent(s): be1fe7c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -78,7 +78,7 @@ def transformer_encoder(inputs, head_size, num_heads, ff_dim, dropout=0):
78
  return LayerNormalization(epsilon=1e-6)(x + res)
79
 
80
  # RL Environment
81
- class PricingEnv(Env):
82
  def __init__(self, data):
83
  super(PricingEnv, self).__init__()
84
  self.data = data
 
78
  return LayerNormalization(epsilon=1e-6)(x + res)
79
 
80
  # RL Environment
81
+ class PricingEnv(gym.Env):
82
  def __init__(self, data):
83
  super(PricingEnv, self).__init__()
84
  self.data = data