From 424a5545670422d859abc706a0b78e5b10573d4d Mon Sep 17 00:00:00 2001 From: "Raffin, Antonin" Date: Thu, 5 Dec 2019 14:50:11 +0100 Subject: [PATCH] Update docstring --- torchy_baselines/common/base_class.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/torchy_baselines/common/base_class.py b/torchy_baselines/common/base_class.py index 40c3061..c70c2a2 100644 --- a/torchy_baselines/common/base_class.py +++ b/torchy_baselines/common/base_class.py @@ -547,7 +547,9 @@ class BaseRLModel(object): @staticmethod def excluded_save_params(): """ - returns the names of the parameters that should be excluded from save + Returns the names of the parameters that should be excluded by default + when saving the model. + :return: ([str]) List of parameters that should be excluded from save """ return ["env", "eval_env", "replay_buffer", "rollout_buffer"]