amend

pytorch · Jan 9, 2024 · 839483b · 839483b
1 parent 4f32045
commit 839483b
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/examples/cql/utils.py b/examples/cql/utils.py
@@ -152,7 +152,7 @@ def make_replay_buffer(
 
 def make_offline_replay_buffer(rb_cfg):
     data = D4RLExperienceReplay(
-        rb_cfg.dataset,
+        dataset_id=rb_cfg.dataset,
         split_trajs=False,
         batch_size=rb_cfg.batch_size,
         sampler=SamplerWithoutReplacement(drop_last=False),

diff --git a/examples/decision_transformer/utils.py b/examples/decision_transformer/utils.py
@@ -239,7 +239,7 @@ def make_offline_replay_buffer(rb_cfg, reward_scaling):
         exclude,
     )
     data = D4RLExperienceReplay(
-        rb_cfg.dataset,
+        dataset_id=rb_cfg.dataset,
         split_trajs=True,
         batch_size=rb_cfg.batch_size,
         sampler=RandomSampler(),  # SamplerWithoutReplacement(drop_last=False),

diff --git a/examples/iql/utils.py b/examples/iql/utils.py
@@ -133,7 +133,7 @@ def make_replay_buffer(
 
 def make_offline_replay_buffer(rb_cfg):
     data = D4RLExperienceReplay(
-        name=rb_cfg.dataset,
+        dataset_id=rb_cfg.dataset,
         split_trajs=False,
         batch_size=rb_cfg.batch_size,
         sampler=SamplerWithoutReplacement(drop_last=False),