spidey121 commited on
Commit
c0ff0fb
·
1 Parent(s): 5e4a461

fix round 1

Browse files
env/attacker.py CHANGED
@@ -94,8 +94,16 @@ def directory_traversal():
94
 
95
 
96
  def simulate_attack():
97
- brute_force()
98
- port_scan()
99
- credential_stuffing()
100
- sql_injection()
101
- directory_traversal()
 
 
 
 
 
 
 
 
 
94
 
95
 
96
  def simulate_attack():
97
+
98
+ attacks = [
99
+ brute_force,
100
+ port_scan,
101
+ credential_stuffing,
102
+ sql_injection,
103
+ directory_traversal
104
+ ]
105
+
106
+ selected = random.sample(attacks, k=3)
107
+
108
+ for attack in selected:
109
+ attack()
tasks/easy/grader.py CHANGED
@@ -1,3 +1,5 @@
1
  def grade(rewards):
 
 
2
  score = sum(rewards) / len(rewards)
3
  return min(max(score, 0.0), 1.0)
 
1
  def grade(rewards):
2
+ if not rewards:
3
+ return 0.0
4
  score = sum(rewards) / len(rewards)
5
  return min(max(score, 0.0), 1.0)
tasks/hard/grader.py CHANGED
@@ -1,4 +1,7 @@
1
  def grade(rewards):
 
 
 
2
  score = sum(rewards) / len(rewards)
3
  score = score * 0.8
4
  return min(max(score, 0.0), 1.0)
 
1
  def grade(rewards):
2
+ if not rewards:
3
+ return 0.0
4
+
5
  score = sum(rewards) / len(rewards)
6
  score = score * 0.8
7
  return min(max(score, 0.0), 1.0)
tasks/medium/grader.py CHANGED
@@ -1,4 +1,7 @@
1
  def grade(rewards):
 
 
 
2
  score = sum(rewards) / len(rewards)
3
  score = score * 0.9
4
  return min(max(score, 0.0), 1.0)
 
1
  def grade(rewards):
2
+ if not rewards:
3
+ return 0.0
4
+
5
  score = sum(rewards) / len(rewards)
6
  score = score * 0.9
7
  return min(max(score, 0.0), 1.0)