Fix logic in updater.py with respect to Counter class

Previously, it worked for Counter.py but not for collections.

Change-Id: I0fff144b0c261b47ce40a03766281f9df642f277
Signed-off-by: Vratko Polak <vrpolak@cisco.com>

diff --git a/tools/pcep_updater/updater.py b/tools/pcep_updater/updater.py
index aa711eac2b1f7d25640fdd48a0f50cc95f3dde15..95c2fb87da430da1edded201c632417e7e1f22d6 100644 (file)
--- a/tools/pcep_updater/updater.py
+++ b/tools/pcep_updater/updater.py
@@ -39,7 +39,7 @@ 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
 
 
@@ -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):