Fix serialization

PySymGym · Dec 27, 2024 · 946a8d9 · 946a8d9
1 parent 615a7c1
commit 946a8d9
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/VSharp.Explorer/Explorer.fs b/VSharp.Explorer/Explorer.fs
@@ -47,13 +47,8 @@ type private SVMExplorer(explorationOptions: ExplorationOptions, statistics: SVM
 
     let folderToStoreSerializationResult =
         match options.aiOptions with
-        | Some ((Training trainingOptions)) ->
-            let mapName =
-                match trainingOptions with
-                | SendModel aiAgentTrainingModelOptions ->
-                    aiAgentTrainingModelOptions.aiAgentTrainingOptions.aiBaseOptions.mapName
-                | SendEachStep aiOptions -> aiOptions.aiAgentTrainingOptions.aiBaseOptions.mapName
-
+        | Some (DatasetGenerator aiOptions) ->
+            let mapName = aiOptions.mapName
             getFolderToStoreSerializationResult
                 (Path.GetDirectoryName explorationOptions.outputDirectory.FullName)
                 mapName