diff --git a/RepoCoder/make_window.py b/RepoCoder/make_window.py index 0da90b6..e991ad5 100644 --- a/RepoCoder/make_window.py +++ b/RepoCoder/make_window.py @@ -138,7 +138,7 @@ def build_window(self): } }) print(f'build {len(code_windows)} ground truth windows for {self.repo} with window size {self.window_size}') - output_path = FilePathBuilder.search_first_window_path(self.benchmark, CONSTANTS.rg, self.repo, self.window_size) + output_path = FilePathBuilder.search_first_window_path(self.benchmark, CONSTANTS.gt, self.repo, self.window_size) Tools.dump_pickle(code_windows, output_path) class PredictionWindowMaker: