<commit_before>import java.util.List;
import java.util.ArrayList;

public class FunctionBuilder extends InstructionBuilder {
	private String name;
	private DataType dataType;
	private List<Variable> variableList;
	private List<Instruction> instructionList;
	private int variableCount, argumentCount;

	public FunctionBuilder(String name, DataType dataType) {
		this.name = name;
		this.dataType = dataType;
		this.variableList = new ArrayList<Variable>();
		this.instructionList = new ArrayList<Instruction>();
		this.variableCount = 0;
		this.argumentCount = 0;
	}

	public String getInstructionName() {
		return "function";
	}

	public void addVariable(String name, DataType dataType) {
		variableList.add(new Variable(name, dataType, Variable.Kind.LOCAL_VARIABLE, variableCount++));
	}

	public void addArgument(String name, DataType dataType) {
		variableList.add(new Variable(name, dataType, Variable.Kind.ARGUMENT, argumentCount++));
	}

	public void addInstruction(Instruction inst) {
		instructionList.add(inst);
	}

	public Function toFunction() {
		return new Function(name, dataType, variableList, instructionList);
	}
}<commit_msg>have addVariable() and addArgument() in FunctionBuilder return Variable object created<commit_after>import java.util.List;
import java.util.ArrayList;

public class FunctionBuilder extends InstructionBuilder {
	private String name;
	private DataType dataType;
	private List<Variable> variableList;
	private List<Instruction> instructionList;
	private int variableCount, argumentCount;

	public FunctionBuilder(String name, DataType dataType) {
		this.name = name;
		this.dataType = dataType;
		this.variableList = new ArrayList<Variable>();
		this.instructionList = new ArrayList<Instruction>();
		this.variableCount = 0;
		this.argumentCount = 0;
	}

	public String getInstructionName() {
		return "function";
	}

	public Variable addVariable(String name, DataType dataType) {
		Variable newVariable = new Variable(name, dataType, Variable.Kind.LOCAL_VARIABLE, variableCount++);
		variableList.add(newVariable);
		return newVariable;
	}

	public Variable addArgument(String name, DataType dataType) {
		Variable newArgument = new Variable(name, dataType, Variable.Kind.ARGUMENT, argumentCount++);
		variableList.add(newArgument);
		return newArgument;
	}

	public void addInstruction(Instruction inst) {
		instructionList.add(inst);
	}

	public Function toFunction() {
		return new Function(name, dataType, variableList, instructionList);
	}
}
{'fraction_non_alphanumeric': 0.09489344591877764, 'fraction_numerical': 0.0016083634901487736, 'mean_word_length': 5.835164835164835, 'pattern_counts': {'":': 0, '<': 11, '<?xml version=': 0, '>': 11, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '5283603', 'n_tokens_mistral': 742, 'n_tokens_neox': 691, 'n_words': 189}