Skip to content

Commit 96fa707

Browse files
committed
fix linter
1 parent 7cdec90 commit 96fa707

File tree

2 files changed

+5
-7
lines changed

2 files changed

+5
-7
lines changed

examples/training/pomdp_solve/stopping_pomdp_defender/run_vs_random_attacker_v_001.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
import numpy as np
2-
from gym_csle_stopping_game.dao.stopping_game_config import StoppingGameConfig
32
from csle_common.metastore.metastore_facade import MetastoreFacade
43
from gym_csle_stopping_game.util.stopping_game_util import StoppingGameUtil
54
from csle_tolerance.util.pomdp_solve_parser import PomdpSolveParser
@@ -21,15 +20,16 @@
2120
simulation_env_config.simulation_env_input_config.stopping_game_config.L = 1
2221
simulation_env_config.simulation_env_input_config.stopping_game_config.O = list(range(10))
2322
simulation_env_config.simulation_env_input_config.stopping_game_config.Z = \
24-
StoppingGameUtil.observation_tensor(n=10-1)
23+
StoppingGameUtil.observation_tensor(n=10 - 1)
2524
gamma = 0.95
2625
pi2 = np.array([
2726
[0.9, 0.1],
2827
[1, 0],
2928
[1, 0]
3029
])
31-
pomdp_solve_file_str = StoppingGameUtil.pomdp_solver_file(config=simulation_env_config.simulation_env_input_config.stopping_game_config,
32-
discount_factor=gamma, pi2=pi2)
30+
pomdp_solve_file_str = StoppingGameUtil.pomdp_solver_file(
31+
config=simulation_env_config.simulation_env_input_config.stopping_game_config,
32+
discount_factor=gamma, pi2=pi2)
3333
with open("/home/kim/thesis/paper_1.pomdp", 'w') as f:
3434
f.write(pomdp_solve_file_str)
3535

simulation-system/libs/gym-csle-stopping-game/src/gym_csle_stopping_game/util/stopping_game_util.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -384,7 +384,5 @@ def pomdp_solver_file(config: StoppingGameConfig, discount_factor: float, pi2: n
384384
for s_prime in config.S:
385385
for o in config.O:
386386
r = config.R[0][a1][0][s]
387-
file_str = (
388-
file_str + f"R: {a1} : {s} : {s_prime} : {o} {r:.80f}\n"
389-
)
387+
file_str = file_str + f"R: {a1} : {s} : {s_prime} : {o} {r:.80f}\n"
390388
return file_str

0 commit comments

Comments
 (0)