Ajayyy00 commited on
Commit
583fea5
·
1 Parent(s): 1de5011

Initialize _step_reward_total in __init__ to fix AttributeError

Browse files
Files changed (1) hide show
  1. server/play_environment.py +1 -0
server/play_environment.py CHANGED
@@ -196,6 +196,7 @@ class CyberSOCEnvironment(Environment):
196
  self._disruption_cost: float = 0.0 # accumulates per clean host/subnet isolated
197
  self._discovered_iocs: set = set() # IOCs revealed via run_forensics or enrich_ioc
198
  self._quarantined_files: set[tuple[str, str]] = set()
 
199
 
200
  def _reset_rubric(self):
201
  """Initialize live containment requirements for dynamic grading in adaptive mode."""
 
196
  self._disruption_cost: float = 0.0 # accumulates per clean host/subnet isolated
197
  self._discovered_iocs: set = set() # IOCs revealed via run_forensics or enrich_ioc
198
  self._quarantined_files: set[tuple[str, str]] = set()
199
+ self._step_reward_total: float = 0.0
200
 
201
  def _reset_rubric(self):
202
  """Initialize live containment requirements for dynamic grading in adaptive mode."""