From 85d37432249df7f45b73de02026a51c30c0f1d6b Mon Sep 17 00:00:00 2001
From: "Dormann, Noah" <Noah.Dormann@dlr.de>
Date: Thu, 5 Dec 2019 08:09:26 +0100
Subject: [PATCH] added standart exclude parameters

Co-Authored-By: Raffin, Antonin <Antonin.Raffin@dlr.de>
---
 torchy_baselines/common/base_class.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torchy_baselines/common/base_class.py b/torchy_baselines/common/base_class.py
index 1df0bac..199467c 100644
--- a/torchy_baselines/common/base_class.py
+++ b/torchy_baselines/common/base_class.py
@@ -499,7 +499,7 @@ class BaseRLModel(object):
         returns the names of the parameters that should be excluded from save
         :return: (list) List of parameters that should be excluded from save
         """
-        return ["replay_buffer"]
+        return ["env", "eval_env", "replay_buffer", "rollout_buffer"]
 
     def save(self, path, exclude=None, include=None):
         """