Fix default buffer creation at fit_online

takuseno · May 6, 2024 · 2cfa571 · 2cfa571
1 parent c1560bf
commit 2cfa571
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/d3rlpy/algos/qlearning/base.py b/d3rlpy/algos/qlearning/base.py
@@ -654,7 +654,7 @@ def fit_online(
 
         # create default replay buffer
         if buffer is None:
-            buffer = create_fifo_replay_buffer(1000000)
+            buffer = create_fifo_replay_buffer(1000000, env=env)
 
         # check action-space
         assert_action_space_with_env(self, env)