SYMBOL INDEX (53 symbols across 8 files)

FILE: demo/fastapicpu.py
  function add_code_generation_args (line 17) | def add_code_generation_args(parser):
  function device (line 129) | def device():
  function create_item (line 153) | async def create_item(request: Request):

FILE: demo/gpus.py
  function auto_configure_device_map (line 8) | def auto_configure_device_map(num_gpus: int) -> Dict[str, int]:
  function load_model_on_gpus (line 45) | def load_model_on_gpus(checkpoint_path: Union[str, os.PathLike], num_gpu...

FILE: demo/run_demo.py
  function get_model (line 35) | def get_model(args):
  function add_code_generation_args (line 81) | def add_code_generation_args(parser):
  function set_random_seed (line 221) | def set_random_seed(seed):
  function main (line 228) | def main():

FILE: evaluation/evaluation.py
  function postprocess_generation (line 29) | def postprocess_generation(sample, generation_mode="completion"):
  function process_test (line 43) | def process_test(sample, problems, dataset_type, language_type, generati...
  function evaluate_functional_correctness (line 94) | def evaluate_functional_correctness(
  function main (line 214) | def main():

FILE: evaluation/execution.py
  function dicts_to_jsonl (line 16) | def dicts_to_jsonl(data_list: list, filename: str, compress: bool = True...
  function check_correctness (line 45) | def check_correctness(
  function time_limit (line 409) | def time_limit(seconds: float):
  function swallow_io (line 422) | def swallow_io():
  function create_tempdir (line 431) | def create_tempdir():
  class TimeoutException (line 437) | class TimeoutException(Exception):
  class WriteOnlyStringIO (line 441) | class WriteOnlyStringIO(io.StringIO):
    method read (line 444) | def read(self, *args, **kwargs):
    method readline (line 447) | def readline(self, *args, **kwargs):
    method readlines (line 450) | def readlines(self, *args, **kwargs):
    method readable (line 453) | def readable(self, *args, **kwargs):
  class redirect_stdin (line 458) | class redirect_stdin(contextlib._RedirectStream):  # type: ignore
  function chdir (line 463) | def chdir(root):
  function reliability_guard (line 477) | def reliability_guard(maximum_memory_bytes: Optional[int] = None):

FILE: evaluation/generation.py
  function add_code_generation_specific_args (line 17) | def add_code_generation_specific_args(parser):
  class CodeStoppingCriteria (line 145) | class CodeStoppingCriteria(StoppingCriteria):
    method __init__ (line 150) | def __init__(
    method __call__ (line 166) | def __call__(self, input_ids: torch.LongTensor, scores: torch.FloatTen...
  function run_generation_distributed (line 183) | def run_generation_distributed(args, model, tokenizer):
  function main (line 283) | def main(args, node_rank: int, local_rank: int, master_port: int, num_de...
  function server (line 315) | def server(args):

FILE: evaluation/inspect_jsonl.py
  function main (line 9) | def main(

FILE: evaluation/utils.py
  function set_random_seed (line 172) | def set_random_seed(seed):
  function stream_jsonl (line 179) | def stream_jsonl(filename: str) -> Iterable[Dict]:
  function stream_jsonl_all (line 196) | def stream_jsonl_all(filename: str) -> Iterable[Dict]:
  function read_dataset (line 210) | def read_dataset(
  function read_translation_dataset (line 257) | def read_translation_dataset(
  function process_extra_prompt (line 290) | def process_extra_prompt(
  function is_code_generation_finished (line 315) | def is_code_generation_finished(
  function first_block (line 368) | def first_block(string, stop_words):
  function cleanup_code (line 373) | def cleanup_code(
  function estimate_pass_at_k (line 435) | def estimate_pass_at_k(
  class Logger (line 461) | class Logger:
    method __init__ (line 462) | def __init__(self, name, log_level=logging.INFO, log_file=None, log_mo...
    method add_file_handler (line 483) | def add_file_handler(self, file_name):
    method debug (line 488) | def debug(self, message):
    method info (line 491) | def info(self, message):
    method warning (line 494) | def warning(self, message):
    method error (line 497) | def error(self, message):
    method critical (line 500) | def critical(self, message):