<commit_before>import nibabel as nib
import numpy as np
import os


def load_freesurfer_label(annot_input, label_name):
	"""
	Get source node list for freesurfer label.
	"""

	annot = nib.freesurfer.read_annot(annot_input)
	label_text = label_name
	label_ind = annot[2].index(label_text)
	labels = np.where(np.in1d(annot[0], label_ind))
	labels = np.where(np.in1d(cort, labels))[0]
	src = np.array(labels, dtype=np.int32)

	return src


def get_freesurfer_label(annot_input, label_name):
	"""
	Print freesurfer label names.
	"""
    annot = nib.freesurfer.read_annot(annot_input)
    print annot[2]
<commit_msg>fix load.py<commit_after>import nibabel as nib
import numpy as np


def load_freesurfer_label(annot_input, label_name):
	"""
	Get source node list for freesurfer label.
	"""

	annot = nib.freesurfer.read_annot(annot_input)
	label_text = label_name
	label_ind = annot[2].index(label_text)
	labels = np.where(np.in1d(annot[0], label_ind))
	labels = np.where(np.in1d(cort, labels))[0]
	src = np.array(labels, dtype=np.int32)

	return src


def get_freesurfer_label(annot_input, label_name):
	"""
	Print freesurfer label names.
	"""
	annot = nib.freesurfer.read_annot(annot_input)
	print annot[2]

{'fraction_non_alphanumeric': 0.11877076411960133, 'fraction_numerical': 0.013289036544850499, 'mean_word_length': 5.243523316062176, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 3, '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': '3693830', 'n_tokens_mistral': 488, 'n_tokens_neox': 464, 'n_words': 95}