(data_name)
| 87 | |
| 88 | |
| 89 | def get_dataset_dir(data_name): |
| 90 | data_list = ['aitz_test', 'chinese_app_test', 'gui_odyssey_test', 'android_control_high_test', 'android_control_low_test'] |
| 91 | assert data_name in data_list, "Error, unkonw eval dataset." |
| 92 | data_split = None |
| 93 | data_dir = None |
| 94 | data_subset = None |
| 95 | |
| 96 | current_file_path = os.path.abspath(__file__) |
| 97 | data_dir = os.path.dirname(os.path.dirname(current_file_path)) |
| 98 | |
| 99 | match data_name: |
| 100 | case 'aitz_test': |
| 101 | data_dir = os.path.join(data_dir, "eval_data", "aitz_test") |
| 102 | data_split = "test" |
| 103 | data_subset = ["general", "install", "web_shopping", "google_apps"] |
| 104 | case 'chinese_app_test': |
| 105 | data_dir = os.path.join(data_dir, "eval_data", "chinese_app_test") |
| 106 | data_split = "test" |
| 107 | data_subset = ["domestic"] |
| 108 | case 'gui_odyssey_test': |
| 109 | data_dir = os.path.join(data_dir, "eval_data", "odyssey") |
| 110 | data_split = "test" |
| 111 | data_subset = ["odyssey"] |
| 112 | case 'android_control_high_test': |
| 113 | data_dir = os.path.join(data_dir, "eval_data", "android_control_high_test") |
| 114 | data_split = "test" |
| 115 | data_subset = ["android_control"] |
| 116 | case 'android_control_low_test': |
| 117 | data_dir = os.path.join(data_dir, "eval_data", "android_control_low_test") |
| 118 | data_split = "test" |
| 119 | data_subset = ["android_control"] |
| 120 | |
| 121 | return data_dir, data_split, data_subset |
no outgoing calls
no test coverage detected