Change script interface to not take test_data_dir

There is no need to take this path since it exists in the tests
directory and the script will only exist in the tests directory.