Fix typing error in SQIL implementation.

HumanCompatibleAI · Jul 18, 2023 · 18a2f11 · 18a2f11
1 parent 9c5b91c
commit 18a2f11
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/imitation/algorithms/sqil.py b/src/imitation/algorithms/sqil.py
@@ -181,7 +181,7 @@ def set_demonstrations(
                 next_obs=transition["next_obs"],
                 action=transition["acts"],
                 done=transition["dones"],
-                reward=1,
+                reward=np.array(1),
                 infos=[{}],
             )