Fixed an attribute error with data_dir.  Should be ready to tested with adjacent parts of the pipeline