Fix warnings reported by gen_doc.sh
[csit.git] / resources / libraries / python / IPv4NodeAddress.py
index 25179a3..27d55b7 100644 (file)
@@ -1,4 +1,4 @@
-# Copyright (c) 2016 Cisco and/or its affiliates.
+# Copyright (c) 2018 Cisco and/or its affiliates.
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at:
@@ -42,8 +42,8 @@ class IPv4NetworkGenerator(object):
 
     def next_network(self):
         """
-        :return: Next network in form (IPv4Network, subnet).
-        :raises: StopIteration if there are no more elements.
+        :returns: Next network in form (IPv4Network, subnet).
+        :raises StopIteration: If there are no more elements.
         """
         if len(self._networks):
             return self._networks.pop()
@@ -79,7 +79,8 @@ def get_variables(nodes, networks=IPV4_NETWORKS[:]):
         port_idx = 0
         ports = {}
         for node in nodes.values():
-            if_name = topo.get_interface_by_link_name(node, link)
+            if_key = topo.get_interface_by_link_name(node, link)
+            if_name = topo.get_interface_name(node, if_key)
             if if_name is not None:
                 port = {'addr': str(next(net_hosts)),
                         'node': node['host'],