MAINT: use new LoggedPolicy HOF

enthought · Jan 31, 2016 · 2246ad0 · 2246ad0
1 parent 03776a6
commit 2246ad0
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/simplesat/sat/policy/priority_queue_policy.py b/simplesat/sat/policy/priority_queue_policy.py
@@ -8,7 +8,7 @@
 from simplesat.utils import DefaultOrderedDict, toposort, transitive_neighbors
 from simplesat.priority_queue import PriorityQueue, GroupPrioritizer
 from .policy import IPolicy
-from .policy_logger import PolicyLogger
+from .policy_logger import LoggedPolicy
 
 
 class PriorityQueuePolicy(IPolicy):
@@ -218,7 +218,4 @@ def _update_cache_from_assignments(self, assignments):
         assert ours == theirs, msg.format(ours, theirs, has_new_keys)
 
 
-def LoggedPriorityQueuePolicty(pool, installed_repository, *args, **kwargs):
-    policy = PriorityQueuePolicy(pool, installed_repository, *args, **kwargs)
-    logger = PolicyLogger(policy, extra_args=args, extra_kwargs=kwargs)
-    return logger
+LoggedPriorityQueuePolicty = LoggedPolicy(PriorityQueuePolicy)