def __enter__(self):
self.orig_fetch_max_wait_time = self.consumer.fetch_max_wait_time<EOL>self.orig_fetch_min_bytes = self.consumer.fetch_min_bytes<EOL>if self.block:<EOL><INDENT>self.consumer.fetch_max_wait_time = self.timeout<EOL>self.consumer.fetch_min_bytes = <NUM_LIT:1><EOL><DEDENT>else:<EOL><INDENT>self.consumer.fetch_min_bytes = <NUM_LIT:0><EOL><DEDENT>
Set fetch values based on blocking status