<commit_before>import os
import pip
from shutil import copy2


class PrivilegeException(Exception):
    def __init__(self, value):
        self.paramater = value
    def __str__(self):
        return str(self.paramater)

def install(package):
    pip.main(['install', package])


def main():
    if os.geteuid() is not 0:
        raise PrivilegeException("You must have root privileges to run this install script.")

    print("Checking dependencies...")
    try:
        import colorama
    except (ImportError, ModuleNotFoundError):
        print("Required module [colorama] was not found. Installing via pip.")
        install('colorama')

    print("Copying files...")
    path = '/opt/wr'
    os.mkdir(path)
    copy2('./main.py', path)
    copy2('./joblib.py', path)
    copy2('./auxiliary.py', path)

    print("Creating symlink...")
    os.symlink('/opt/wr/main.py', '/usr/local/bin/wr')
    print("Successfully installed!")<commit_msg>added forgotten ifmain statement<commit_after>import os
import pip
from shutil import copy2


class PrivilegeException(Exception):
    def __init__(self, value):
        self.paramater = value
    def __str__(self):
        return str(self.paramater)


def install(package):
    pip.main(['install', package])


def main():
    if os.geteuid() is not 0:
        raise PrivilegeException("You must have root privileges to run this install script.")

    print("Checking dependencies...")
    try:
        import colorama
    except (ImportError, ModuleNotFoundError):
        print("Required module [colorama] was not found. Installing via pip.")
        install('colorama')

    print("Copying files...")
    path = '/opt/wr'
    os.mkdir(path)
    copy2('./main.py', path)
    copy2('./joblib.py', path)
    copy2('./auxiliary.py', path)

    print("Creating symlink...")
    os.symlink('/opt/wr/main.py', '/usr/local/bin/wr')
    print("Successfully installed!")


if __name__ == '__main__':
    main()
{'fraction_non_alphanumeric': 0.1391170431211499, 'fraction_numerical': 0.00513347022587269, 'mean_word_length': 3.0268595041322315, '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': '439847', 'n_tokens_mistral': 597, 'n_tokens_neox': 577, 'n_words': 158}