Skip to content

Commit

Permalink
.
Browse files Browse the repository at this point in the history
  • Loading branch information
William Blum committed Aug 6, 2024
1 parent c0eecc9 commit 5b8c683
Show file tree
Hide file tree
Showing 4 changed files with 5 additions and 5 deletions.
2 changes: 1 addition & 1 deletion .vscode/settings.json
Original file line number Diff line number Diff line change
Expand Up @@ -23,4 +23,4 @@
"files.trimFinalNewlines": true,
"editor.defaultFormatter": "",
"files.trimTrailingWhitespace": true,
}
}
4 changes: 2 additions & 2 deletions cyberbattle/agents/baseline/notebooks/notebook_dql_debug.py
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,7 @@

current_o, info = ctf_env.reset()
wrapped_env = AgentWrapper(ctf_env, ActionTrackingStateAugmentation(ep, current_o))
l = dqn_learning_run["learner"]
_l = dqn_learning_run["learner"]

# %%
# Use the trained agent to run the steps one by one
Expand All @@ -89,7 +89,7 @@
h = []
for i in range(max_steps):
# run the suggested action
_, next_action, _ = l.exploit(wrapped_env, current_o)
_, next_action, _ = _l.exploit(wrapped_env, current_o)
h.append(
(
ctf_env.get_explored_network_node_properties_bitmap_as_numpy(current_o),
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -3,4 +3,4 @@ line-length = 200
exclude = [
"typings",
"venv"
]
]
2 changes: 1 addition & 1 deletion pyrightconfig.json
Original file line number Diff line number Diff line change
Expand Up @@ -21,4 +21,4 @@
"reportIncompatibleVariableOverride": "information",
"strictParameterNoneValue": true,
"strict": []
}
}

0 comments on commit 5b8c683

Please sign in to comment.