Skip to content

Commit d0a425b

Browse files
committed
check for buffer kwargs type
1 parent a3cce9f commit d0a425b

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

mighty/mighty_agents/base_agent.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -323,9 +323,10 @@ def initialize_agent(self) -> None:
323323
if isinstance(self.buffer_class, type) and issubclass(
324324
self.buffer_class, PrioritizedReplay
325325
):
326-
self.buffer_kwargs = OmegaConf.to_container(
327-
self.buffer_kwargs, resolve=True
328-
)
326+
if isinstance(self.buffer_kwargs, DictConfig):
327+
self.buffer_kwargs = OmegaConf.to_container(
328+
self.buffer_kwargs, resolve=True
329+
)
329330
# 1) Get observation-space shape
330331
try:
331332
obs_space = self.env.single_observation_space

0 commit comments

Comments
 (0)