old_contents: from local_worker import worker_loop, wait_for_messages
  import sys
  import logging
  import model
  
  
  from pubsub_queue import PubsubQueue
  from sqs_queue import SQSQueue
  
  import argparse
  logging.basicConfig()
  
  
  def main(args=sys.argv):
      logger = logging.getLogger('studio-remote-worker')
      parser = argparse.ArgumentParser(
          description='Studio remote worker. \
                       Usage: studio-remote-worker \
                       ')
      parser.add_argument('--config', help='configuration file', default=None)
  
      parser.add_argument(
          '--guest',
          help='Guest mode (does not require db credentials)',
          action='store_true')
  
      parser.add_argument(
          '--single-run',
          help='quit after a single run (regardless of the state of the queue)',
          action='store_true')
  
      parser.add_argument('--queue', help='queue name', required=True)
      parser.add_argument(
          '--verbose', '-v',
          help='Verbosity level. Allowed vaules: ' +
               'debug, info, warn, error, crit ' +
               'or numerical value of logger levels.',
          default=None)
  
      parser.add_argument(
          '--timeout', '-t',
          help='Timeout after which remote worker stops listening (in seconds)',
          type=int,
          default=-1)
  
      parsed_args, script_args = parser.parse_known_args(args)
      verbose = model.parse_verbosity(parsed_args.verbose)
      logger.setLevel(verbose)
      if parsed_args.queue.startswith('ec2_') or \
         parsed_args.queue.startswith('sqs_'):
          queue = SQSQueue(parsed_args.queue, verbose=verbose)
      else:
          queue = PubsubQueue(parsed_args.queue, verbose=verbose)
      logger.info('Waiting for the work in the queue...')
  
      timeout_before = parsed_args.timeout
      timeout_after = timeout_before if timeout_before > 0 else 0
      wait_for_messages(queue, timeout_before, logger)
  
      logger.info('Starting working')
      worker_loop(queue, parsed_args,
                  setup_pyenv=True,
                  single_experiment=parsed_args.single_run,
                  fetch_artifacts=True,
                  timeout=timeout_after,
                  verbose=verbose)
  
  
  if __name__ == "__main__":
      main()
  
new_contents: from local_worker import worker_loop, wait_for_messages
  import sys
  import logging
  import model
  
  
  from pubsub_queue import PubsubQueue
  from sqs_queue import SQSQueue
  
  import argparse
  logging.basicConfig()
  
  
  def main(args=sys.argv):
      logger = logging.getLogger('studio-remote-worker')
      parser = argparse.ArgumentParser(
          description='Studio remote worker. \
                       Usage: studio-remote-worker \
                       ')
      parser.add_argument('--config', help='configuration file', default=None)
  
      parser.add_argument(
          '--guest',
          help='Guest mode (does not require db credentials)',
          action='store_true')
  
      parser.add_argument(
          '--single-run',
          help='quit after a single run (regardless of the state of the queue)',
          action='store_true')
  
      parser.add_argument('--queue', help='queue name', required=True)
      parser.add_argument(
          '--verbose', '-v',
          help='Verbosity level. Allowed vaules: ' +
               'debug, info, warn, error, crit ' +
               'or numerical value of logger levels.',
          default=None)
  
      parser.add_argument(
          '--timeout', '-t',
          help='Timeout after which remote worker stops listening (in seconds)',
          type=int,
          default=-1)
  
      parsed_args, script_args = parser.parse_known_args(args)
      verbose = model.parse_verbosity(parsed_args.verbose)
      logger.setLevel(verbose)
      if parsed_args.queue.startswith('ec2_') or \
         parsed_args.queue.startswith('sqs_'):
          queue = SQSQueue(parsed_args.queue, verbose=verbose)
      else:
          queue = PubsubQueue(parsed_args.queue, verbose=verbose)
      logger.info('Waiting for the work in the queue...')
  
      timeout_before = parsed_args.timeout
      timeout_after = timeout_before if timeout_before > 0 else 0
      # wait_for_messages(queue, timeout_before, logger)
  
      logger.info('Starting working')
      worker_loop(queue, parsed_args,
                  setup_pyenv=True,
                  single_experiment=parsed_args.single_run,
                  fetch_artifacts=True,
                  timeout=timeout_after,
                  verbose=verbose)
  
  
  if __name__ == "__main__":
      main()
  
subject: fix in remote_worker.py