Pass through args/kwargs in Python context mgr

mrb · Dec 23, 2011 · a50c810 · a50c810
1 parent 9cb1ede
commit a50c810
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/python/chashring.py b/lib/python/chashring.py
@@ -98,10 +98,10 @@ def free(self):
         hash_ring.hash_ring_free(self._hash_ring_ptr)
 
 @contextmanager
-def create_hash_ring(nodes):
+def create_hash_ring(*args, **kwargs):
     """ Use this to safely use a HashRing object without worrying about having to
         call 'free' afterwards.
     """
-    h = HashRing(nodes)
+    h = HashRing(*args, **kwargs)
     yield h
     h.free()