update config func

lucas-yyy000 · Dec 22, 2020 · 1624647 · 1624647
1 parent 6c67ad9
commit 1624647
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/easyrl/configs/__init__.py b/easyrl/configs/__init__.py
@@ -14,7 +14,7 @@ class CFG:
 cfg = CFG()
 
 
-def set_config(alg, config_func=None):
+def set_config(alg=None, config_func=None):
     global cfg
     if config_func is not None:
         cfg.alg = config_func()