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