code: def generate(module_name, code):
      
      try:
          ast_tree = ast.parse(code)
      except Exception:
          raise RuntimeError('Bad Python code')
  
      visitor = SearchSpaceGenerator(module_name)
      try:
          visitor.visit(ast_tree)
      except AssertionError as exc:
          raise RuntimeError('%d: %s' % (visitor.last_line, exc.args[0]))
      return visitor.search_space, astor.to_source(ast_tree)
docs: Generate search space.
      Return a serializable search space object.
      module_name: name of the module (str)
      code: user code (str)