From a756f40223d59f7662f2dc702ff9a0a76b484996 Mon Sep 17 00:00:00 2001 From: "Dormann, Noah" Date: Thu, 5 Dec 2019 08:10:25 +0100 Subject: [PATCH] comment refactoring Co-Authored-By: Raffin, Antonin --- 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 2554e89..c25a6bb 100644 --- a/torchy_baselines/common/base_class.py +++ b/torchy_baselines/common/base_class.py @@ -506,7 +506,7 @@ class BaseRLModel(object): Save all the attributes of the object and the model parameters in a zip-file for continuous learning :param path: (str) path to the file where the data should be saved - :param exclude: (list) name of parameters that should be excluded, use standard exclude params if None + :param exclude: ([str]) name of parameters that should be excluded in addition to the default one :param include: (list) name of parameters that might be excluded but should be included anyway :return: """