address comments

apache · Sep 10, 2014 · cf1c55e · cf1c55e
1 parent 3133a60
commit cf1c55e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/pyspark/worker.py b/python/pyspark/worker.py
@@ -74,7 +74,7 @@ def main(infile, outfile):
                 _broadcastRegistry[bid] = Broadcast(bid, value)
             else:
                 bid = - bid - 1
-                _broadcastRegistry.pop(bid, None)
+                _broadcastRegistry.remove(bid)
 
         _accumulatorRegistry.clear()
         command = pickleSer._read_with_length(infile)