Merge 24a78e1 into 9b2f58f

chainer · Aug 26, 2019 · 2f384ff · 2f384ff
2 parents 9b2f58f + 24a78e1
commit 2f384ff
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/chainerrl/agents/categorical_dqn.py b/chainerrl/agents/categorical_dqn.py
@@ -51,7 +51,8 @@ def _apply_categorical_projection(y, y_probs, z):
     assert u.shape == (batch_size, n_atoms)
 
     if cuda.available and xp is cuda.cupy:
-        scatter_add = xp.scatter_add
+        import cupyx
+        scatter_add = cupyx.scatter_add
     else:
         scatter_add = np.add.at