refactor: reorganize the code and fix failing tests

Author: BatyLeo

Project.toml

@@ -12,20 +12,18 @@ DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"
 Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
 InferOpt = "4846b161-c94e-4150-8dac-c7ae193c601f"
 MLUtils = "f1d291b0-491e-4a28-83b9-f70985020b54"
-Plots = "91a5bcdd-55d7-5caf-9e0b-520d859cae80"
 ProgressMeter = "92933f4c-e287-5a05-a399-4b506db050ca"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 UnicodePlots = "b8865327-cd53-5732-bb35-84acbb429228"
 ValueHistories = "98cad3c8-aec3-5f06-8e41-884608649ab7"
 
 [compat]
-DecisionFocusedLearningBenchmarks = "0.5.0, 0.6"
+DecisionFocusedLearningBenchmarks = "0.6.1"
 DocStringExtensions = "0.9.5"
 Flux = "0.16.9"
 InferOpt = "0.7.1"
 MLUtils = "0.4.8"
-Plots = "1.41.6"
 ProgressMeter = "1.11.0"
 Random = "1.11.0"
 Statistics = "1.11.1"

src/DecisionFocusedLearningAlgorithms.jl

@@ -25,7 +25,7 @@ include("algorithms/abstract_algorithm.jl")
 include("algorithms/supervised/fyl.jl")
 include("algorithms/supervised/anticipative_imitation.jl")
 include("algorithms/supervised/dagger.jl")
-include("algorithms/MirrorDescent/mirror_descent.jl")
+include("algorithms/mirror_descent/mirror_descent.jl")
 
 export TrainingContext