developing main experiment loop, functions required to implement in bc and other imitation methods

2021-07-20 05:58:24 -07:00
parent c0755f5bb5
commit 2fb5d5e5b1
4 changed files with 143 additions and 1 deletions
--- a/src/init.py
+++ b/src/init.py
@@ -1 +1,2 @@
-from src.expert_data import generate_expert_data, load_expert_data
+from src.expert_data import generate_expert_data, load_expert_data
+from src.data_utils import InteractionDatasetSingleAgent