[tab] data/eval/unit_tests/add_import_easy.md (markdown)
# Assistant
```bash
python src/input_pipeline/validation_to_testcases.py --input_file data/temp/validation.jsonl --output_file data/temp/validation_testcases.jsonl
```

# User
<stdout>
Traceback (most recent call last):
  File "/fast/home/mihir.mahajan/Projects/tab-model-eval/src/input_pipeline/validation_to_testcases.py", line 57, in <module>
    convert_to_incremental_jsonl(line, args.output_file, task_name)
    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/fast/home/mihir.mahajan/Projects/tab-model-eval/src/input_pipeline/validation_to_testcases.py", line 14, in convert_to_incremental_jsonl
    data = json.loads(line)
           ^^^^
NameError: name 'json' is not defined. Did you forget to import 'json'?
</stdout>


# Assistant
```bash
cat -n src/input_pipeline/validation_to_testcases.py
```

# User
<stdout>
     1  import tyro
     2  import os
     3  from dataclasses import dataclass
     4
     5
     6  @dataclass
     7  class Args:
     8      input_file: str = "data/eval/val/validation.jsonl"
     9      output_file: str = "data/eval/val/validation_testcases.jsonl"
    10      task_name: str = "validation_set"
    11
    12
    13  def convert_to_incremental_jsonl(line, output_file, task_name):
    14      data = json.loads(line)
    15
    16      system_prompt = data.get("system_prompt", "")
    17      conversations = data.get("conversations", [])
    18
    19      output_lines = []
    20      context = []
    21
    22      for i, conv in enumerate(conversations):
    23          # Convert "Assistant"/"User" to "assistant"/"user"
    24          role = conv["from"].lower()
    25          content = conv["value"]
    26
    27          # Add current message to context
    28          context.append({"role": role, "content": content})
    29
    30          if i > 0 and role == "assistant":
    31              task_entry = {
    32                  "system_prompt": system_prompt,
    33                  "task_id": f"{task_name}/{i-1}",
    34                  "context": context[:-1].copy(),
    35                  "expected_final_response": content,
    36              }
    37              output_lines.append(task_entry)
    38
    39      with open(output_file, "a") as f:
    40          for line in output_lines:
    41              f.write(json.dumps(line) + "
")
    42
    43      print(f"Converted {len(output_lines)} entries to {output_file}")
    44
    45
    46  if __name__ == "__main__":
    47      args = tyro.cli(Args)
    48
    49      if os.path.exists(args.output_file):
    50          os.remove(args.output_file)
    51
    52      # iterate over jsonl file and convert to incremental jsonl
    53      conversation_id = 0
    54      with open(args.input_file, "r") as f:
    55          for line in f:
    56              task_name = f"conversation_{conversation_id}/{args.task_name}"
    57              convert_to_incremental_jsonl(line, args.output_file, task_name)
    58              conversation_id += 1
</stdout>

# Assistant
```bash
sed -i '1i\import json' src/input_pipeline/validation_to_testcases.py
```

# User
<stdout>
</stdout>

# Assistant
```bash
python src/input_pipeline/validation_to_testcases.py --input_file data/temp/validation.jsonl --output_file data/temp/validation_testcases.jsonl
```

# User
<stdout>
Converted 7 entries to data/temp/validation_testcases.jsonl
Converted 3 entries to data/temp/validation_testcases.jsonl
Converted 4 entries to data/temp/validation_testcases.jsonl
Converted 7 entries to data/temp/validation_testcases.jsonl
Converted 4 entries to data/temp/validation_testcases.jsonl
Converted 8 entries to data/temp/validation_testcases.jsonl
Converted 8 entries to data/temp/validation_testcases.jsonl
Converted 3 entries to data/temp/validation_testcases.jsonl
Converted 3 entries to data/temp/validation_testcases.jsonl
Converted 3 entries to data/temp/validation_testcases.jsonl
Converted 3 entries to data/temp/validation_testcases.jsonl
Converted 5 entries to data/temp/validation_testcases.jsonl
Converted 3 entries to data/temp/validation_testcases.jsonl
Converted 2 entries to data/temp/validation_testcases.jsonl
Converted 4 entries to data/temp/validation_testcases.jsonl
Converted 3 entries to data/temp/validation_testcases.jsonl
Converted 3 entries to data/temp/validation_testcases.jsonl
Converted 5 entries to data/temp/validation_testcases.jsonl
Converted 6 entries to data/temp/validation_testcases.jsonl
Converted 3 entries to data/temp/validation_testcases.jsonl
Converted 3 entries to data/temp/validation_testcases.jsonl
</stdout>