fix linter

Limmen · Jul 19, 2024 · 96fa707 · 96fa707
1 parent 7cdec90
commit 96fa707
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/examples/training/pomdp_solve/stopping_pomdp_defender/run_vs_random_attacker_v_001.py b/examples/training/pomdp_solve/stopping_pomdp_defender/run_vs_random_attacker_v_001.py
@@ -1,5 +1,4 @@
 import numpy as np
-from gym_csle_stopping_game.dao.stopping_game_config import StoppingGameConfig
 from csle_common.metastore.metastore_facade import MetastoreFacade
 from gym_csle_stopping_game.util.stopping_game_util import StoppingGameUtil
 from csle_tolerance.util.pomdp_solve_parser import PomdpSolveParser
@@ -21,15 +20,16 @@
     simulation_env_config.simulation_env_input_config.stopping_game_config.L = 1
     simulation_env_config.simulation_env_input_config.stopping_game_config.O = list(range(10))
     simulation_env_config.simulation_env_input_config.stopping_game_config.Z = \
-        StoppingGameUtil.observation_tensor(n=10-1)
+        StoppingGameUtil.observation_tensor(n=10 - 1)
     gamma = 0.95
     pi2 = np.array([
         [0.9, 0.1],
         [1, 0],
         [1, 0]
     ])
-    pomdp_solve_file_str = StoppingGameUtil.pomdp_solver_file(config=simulation_env_config.simulation_env_input_config.stopping_game_config,
-                                       discount_factor=gamma, pi2=pi2)
+    pomdp_solve_file_str = StoppingGameUtil.pomdp_solver_file(
+        config=simulation_env_config.simulation_env_input_config.stopping_game_config,
+        discount_factor=gamma, pi2=pi2)
     with open("/home/kim/thesis/paper_1.pomdp", 'w') as f:
         f.write(pomdp_solve_file_str)
 

diff --git a/...-system/libs/gym-csle-stopping-game/src/gym_csle_stopping_game/util/stopping_game_util.py b/...-system/libs/gym-csle-stopping-game/src/gym_csle_stopping_game/util/stopping_game_util.py
@@ -384,7 +384,5 @@ def pomdp_solver_file(config: StoppingGameConfig, discount_factor: float, pi2: n
                 for s_prime in config.S:
                     for o in config.O:
                         r = config.R[0][a1][0][s]
-                        file_str = (
-                                file_str + f"R: {a1} : {s} : {s_prime} : {o} {r:.80f}\n"
-                        )
+                        file_str = file_str + f"R: {a1} : {s} : {s_prime} : {o} {r:.80f}\n"
         return file_str