jonathan-laurent · Whojo · Jun 29, 2022 · Jun 29, 2022 · Jun 29, 2022 · Jun 29, 2022
diff --git a/.gitignore b/.gitignore
@@ -9,6 +9,7 @@
 /docs/build/
 /docs/site/
 
+private
 sessions
 archive
 

diff --git a/redesign/Project.toml b/redesign/Project.toml
@@ -7,13 +7,17 @@ version = "0.1.0"
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
 Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"
+EllipsisNotation = "da5c29d0-fa7d-589e-88eb-ea29b0a81949"
 Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
 JET = "c3a54625-cd67-489e-a8e7-0a5a0ff4e31b"
+ParameterSchedulers = "d7d3b36b-41b8-4d0d-a2bf-768c6151755e"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
 ReinforcementLearningBase = "e575027e-6cd6-5018-9292-cdc6200d2b44"
 ReinforcementLearningEnvironments = "25e41dd2-4622-11e9-1641-f1adca772921"
 Setfield = "efcf1570-3423-57d1-acb7-fd33fddbac46"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
 Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
+StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
+Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
diff --git a/redesign/src/BatchedEnvs.jl b/redesign/src/BatchedEnvs.jl
@@ -3,7 +3,7 @@ Interface for batchable environements that can be run on the GPU.
 """
 module BatchedEnvs
 
-export num_actions, valid_action, act, terminated
+export num_actions, valid_action, act, terminated, vectorize_state
 
 function num_actions end
 
@@ -13,4 +13,6 @@ function act end
 
 function terminated end
 
+function vectorize_state end
+
 end
-Original file line number
+Diff line change
@@ Expand Up / @@ -9,6 +9,7 @@ @@
     /docs/build/
     /docs/site/
+    private
     sessions
     archive
@@ Expand Down @@