Add new attributes to Game repr

PySymGym · Jan 17, 2024 · 47a2f5d · 47a2f5d
1 parent e1c397c
commit 47a2f5d
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/VSharp.ML.AIAgent/common/game.py b/VSharp.ML.AIAgent/common/game.py
@@ -8,18 +8,20 @@
 class StateHistoryElem:
  GraphVertexId: int
  NumOfVisits: int
+ StepWhenVisitedLastTime: int
 
 
 @dataclass_json
 @dataclass(slots=True)
 class State:
  Id: int
  Position: int
- PredictedUsefulness: float
  PathConditionSize: int
  VisitedAgainVertices: int
  VisitedNotCoveredVerticesInZone: int
  VisitedNotCoveredVerticesOutOfZone: int
+ StepWhenMovedLastTime: int
+ InstructionsVisitedInCurrentBlock: int
  History: list[StateHistoryElem]
  Children: list[int]
 
@@ -30,13 +32,14 @@ def __hash__(self) -> int:
 @dataclass_json
 @dataclass(slots=True)
 class GameMapVertex:
- Uid: int
  Id: int
  InCoverageZone: bool
  BasicBlockSize: int
  CoveredByTest: bool
  VisitedByState: bool
  TouchedByState: bool
+ ContainsCall: bool
+ ContainsThrow: bool
  States: list[int]
 
 
@@ -69,7 +72,6 @@ class GameMap:
  MaxSteps: int
  CoverageToStart: int
  AssemblyFullName: str
- CoverageZone: bool
  NameOfObjectToCover: str
  MapName: str