spidey121 commited on
Commit
8cf7800
·
1 Parent(s): 3e66129

fix startup order

Browse files
Files changed (1) hide show
  1. env/fake_server.py +6 -3
env/fake_server.py CHANGED
@@ -115,14 +115,17 @@ def reset():
115
 
116
  # ---------------- Step API (IMPORTANT) ----------------
117
 
118
- @app.route("/step", methods=["GET","POST"])
119
  def step():
120
 
121
- action = request.json.get("action")
 
 
 
 
122
 
123
  reward = 0.0
124
  done = False
125
-
126
  # Detect brute force
127
  if action == "detect_bruteforce":
128
  if logs["failed_logins"] > 3:
 
115
 
116
  # ---------------- Step API (IMPORTANT) ----------------
117
 
118
+ @app.route("/step", methods=["GET", "POST"])
119
  def step():
120
 
121
+ if request.method == "POST":
122
+ data = request.get_json(silent=True) or {}
123
+ action = data.get("action", None)
124
+ else:
125
+ action = request.args.get("action")
126
 
127
  reward = 0.0
128
  done = False
 
129
  # Detect brute force
130
  if action == "detect_bruteforce":
131
  if logs["failed_logins"] > 3: