From b2cf7ea34dd06a2249543d6279e26c82a897db31 Mon Sep 17 00:00:00 2001
From: Takuya Akiba <t.akiba.65536@gmail.com>
Date: Tue, 15 Aug 2017 16:19:44 +0900
Subject: [PATCH] Add None to the type specification

---
 chainermn/link.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/chainermn/link.py b/chainermn/link.py
index 9a53d011..f3cb55ba 100644
--- a/chainermn/link.py
+++ b/chainermn/link.py
@@ -115,10 +115,10 @@ def add_link(self, link, rank_in=None, rank_out=None):
 
         Args:
             link (chainer.Link): The link object to be registered.
-            rank_in (int or list):
+            rank_in (int, list, or None):
                 Ranks from which it receives data. If None is specified,
                 the model does not receive from any machines.
-            rank_out (int or list):
+            rank_out (int, list, or None):
                 Ranks to which it sends data. If None is specified,
                 the model will not send to any machine.
         """