From 3a29066f54cf09cd88618c883af06f560e257cc5 Mon Sep 17 00:00:00 2001 From: Sasha Abramowitz Date: Wed, 2 Mar 2022 16:25:18 +0200 Subject: [PATCH] docs: updated docs to better represent available options --- mava/systems/tf/value_decomposition/system.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mava/systems/tf/value_decomposition/system.py b/mava/systems/tf/value_decomposition/system.py index 8bf848748..45ba6ed8b 100644 --- a/mava/systems/tf/value_decomposition/system.py +++ b/mava/systems/tf/value_decomposition/system.py @@ -47,7 +47,7 @@ def __init__( self, environment_factory: Callable[[bool], dm_env.Environment], network_factory: Callable[[acme_specs.BoundedArray], Dict[str, snt.Module]], - mixer: snt.Module, + mixer: Union[snt.Module, str], exploration_scheduler_fn: Union[ EpsilonScheduler, Mapping[str, EpsilonScheduler], @@ -97,7 +97,7 @@ def __init__( environment_factory: function to instantiate an environment. network_factory: function to instantiate system networks. - mixer: mixing network + mixer: mixing network. Either a sonnet module or the strings "qmix"/"vdn" exploration_scheduler_fn: function to schedule exploration. e.g. epsilon greedy logger_factory: function to