Use suite teardown resource cleanup

[integration/test.git] / tools / pcep_updater / updater.py

diff --git a/tools/pcep_updater/updater.py b/tools/pcep_updater/updater.py
index a126ffe7c460452b26dc38bd25f6166c5e0674ed..95c2fb87da430da1edded201c632417e7e1f22d6 100644 (file)
--- a/tools/pcep_updater/updater.py
+++ b/tools/pcep_updater/updater.py
@@ -31,12 +31,6 @@ But that may not be true for more mature implementation.
 # terms of the Eclipse Public License v1.0 which accompanies this distribution,
 # and is available at http://www.eclipse.org/legal/epl-v10.html
 
-__author__ = "Vratko Polak"
-__copyright__ = "Copyright(c) 2015, Cisco Systems, Inc."
-__license__ = "Eclipse Public License v1.0"
-__email__ = "vrpolak@cisco.com"
-
-
 import argparse
 import collections  # For deque and Counter.
 import ipaddr
@@ -45,10 +39,16 @@ import time
 try:
     from collections import Counter
 except ImportError:  # Python 2.6 does not have Counter in collections.
-    import Counter  # Assumes that user copies Counter.py around.
+    from Counter import Counter  # Assumes that user copies Counter.py around.
 import AuthStandalone
 
 
+__author__ = "Vratko Polak"
+__copyright__ = "Copyright(c) 2015, Cisco Systems, Inc."
+__license__ = "Eclipse Public License v1.0"
+__email__ = "vrpolak@cisco.com"
+
+
 def str2bool(text):
     """Utility converter, based on http://stackoverflow.com/a/19227287"""
     return text.lower() in ("yes", "true", "y", "t", "1")
@@ -89,7 +89,7 @@ class CounterDown(object):
     """Counter which also knows how many items are left to be added."""
 
     def __init__(self, tasks):
-        self.counter = Counter.Counter()
+        self.counter = Counter()
         self.opened = tasks
 
     def add(self, result):