From f600591468d2226fb8ad700294c18cf215c5809d Mon Sep 17 00:00:00 2001 From: Vlad Firoiu Date: Mon, 15 Apr 2019 03:51:37 +0100 Subject: [PATCH] Cast MultiCategorical num_outputs to int. (#4629) --- python/ray/rllib/models/catalog.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/ray/rllib/models/catalog.py b/python/ray/rllib/models/catalog.py index 776773552..ce91742c3 100644 --- a/python/ray/rllib/models/catalog.py +++ b/python/ray/rllib/models/catalog.py @@ -147,7 +147,7 @@ class ModelCatalog(object): elif isinstance(action_space, gym.spaces.multi_discrete.MultiDiscrete): if torch: raise NotImplementedError - return MultiCategorical, sum(action_space.nvec) + return MultiCategorical, int(sum(action_space.nvec)) raise NotImplementedError("Unsupported args: {} {}".format( action_space, dist_type))