Skip to content

Commit 3acd930

Browse files
authored
Fix step count (#90)
1 parent 5f0796a commit 3acd930

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

VSharp.ML.GameServer.Runner/Main.fs

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -145,10 +145,12 @@ let ws port outputDirectory (webSocket : WebSocket) (context: HttpContext) =
145145
| ServerStop -> loop <- false
146146
| Start gameMap ->
147147
printfn $"Start map {gameMap.MapName}, port {port}"
148+
let stepsToStart= gameMap.StepsToStart
149+
let stepsToPlay = gameMap.StepsToPlay
148150
let aiTrainingOptions =
149151
{
150-
stepsToSwitchToAI = gameMap.StepsToStart
151-
stepsToPlay = gameMap.StepsToPlay
152+
stepsToSwitchToAI = stepsToStart
153+
stepsToPlay = stepsToPlay
152154
defaultSearchStrategy =
153155
match gameMap.DefaultSearcher with
154156
| searcher.BFS -> BFSMode
@@ -158,7 +160,7 @@ let ws port outputDirectory (webSocket : WebSocket) (context: HttpContext) =
158160
mapName = gameMap.MapName
159161
oracle = Some oracle
160162
}
161-
let options = VSharpOptions(timeout = 15 * 60, outputDirectory = outputDirectory, searchStrategy = SearchStrategy.AI, aiAgentTrainingOptions = aiTrainingOptions, solverTimeout=2)
163+
let options = VSharpOptions(timeout = 15 * 60, outputDirectory = outputDirectory, searchStrategy = SearchStrategy.AI, aiAgentTrainingOptions = aiTrainingOptions, stepsLimit = uint(stepsToPlay + stepsToStart), solverTimeout=2)
162164
let explorationResult = explore gameMap options
163165

164166
Application.reset()

0 commit comments

Comments
 (0)