|
100 | 100 | " )\n", |
101 | 101 | "\n", |
102 | 102 | " # DiffDock (relaxed-protein) results\n", |
103 | | - " globals()[\n", |
104 | | - " f\"diffdock_relaxed_protein_dockgen_bust_results_csv_filepath_{repeat_index}\"\n", |
105 | | - " ] = os.path.join(\n", |
106 | | - " globals()[\"diffdock_output_dir\"],\n", |
107 | | - " f\"diffdock_dockgen_output_{repeat_index}\",\n", |
108 | | - " \"bust_results.csv\",\n", |
| 103 | + " globals()[f\"diffdock_relaxed_protein_dockgen_bust_results_csv_filepath_{repeat_index}\"] = (\n", |
| 104 | + " os.path.join(\n", |
| 105 | + " globals()[\"diffdock_output_dir\"],\n", |
| 106 | + " f\"diffdock_dockgen_output_{repeat_index}\",\n", |
| 107 | + " \"bust_results.csv\",\n", |
| 108 | + " )\n", |
109 | 109 | " )\n", |
110 | 110 | " globals()[\n", |
111 | 111 | " f\"diffdock_relaxed_protein_dockgen_relaxed_bust_results_csv_filepath_{repeat_index}\"\n", |
|
121 | 121 | " f\"diffdock_dockgen_output_{repeat_index}\",\n", |
122 | 122 | " \"bust_results.csv\",\n", |
123 | 123 | " )\n", |
124 | | - " globals()[\n", |
125 | | - " f\"diffdockv1_dockgen_relaxed_bust_results_csv_filepath_{repeat_index}\"\n", |
126 | | - " ] = os.path.join(\n", |
127 | | - " globals()[\"diffdockv1_output_dir\"],\n", |
128 | | - " f\"diffdock_dockgen_output_{repeat_index}_relaxed\",\n", |
129 | | - " \"bust_results.csv\",\n", |
| 124 | + " globals()[f\"diffdockv1_dockgen_relaxed_bust_results_csv_filepath_{repeat_index}\"] = (\n", |
| 125 | + " os.path.join(\n", |
| 126 | + " globals()[\"diffdockv1_output_dir\"],\n", |
| 127 | + " f\"diffdock_dockgen_output_{repeat_index}_relaxed\",\n", |
| 128 | + " \"bust_results.csv\",\n", |
| 129 | + " )\n", |
130 | 130 | " )\n", |
131 | 131 | "\n", |
132 | 132 | " # FABind results\n", |
|
147 | 147 | " f\"dockgen_{repeat_index}\",\n", |
148 | 148 | " \"bust_results.csv\",\n", |
149 | 149 | " )\n", |
150 | | - " globals()[\n", |
151 | | - " f\"dynamicbind_dockgen_relaxed_bust_results_csv_filepath_{repeat_index}\"\n", |
152 | | - " ] = os.path.join(\n", |
153 | | - " globals()[\"dynamicbind_output_dir\"],\n", |
154 | | - " f\"dockgen_{repeat_index}_relaxed\",\n", |
155 | | - " \"bust_results.csv\",\n", |
| 150 | + " globals()[f\"dynamicbind_dockgen_relaxed_bust_results_csv_filepath_{repeat_index}\"] = (\n", |
| 151 | + " os.path.join(\n", |
| 152 | + " globals()[\"dynamicbind_output_dir\"],\n", |
| 153 | + " f\"dockgen_{repeat_index}_relaxed\",\n", |
| 154 | + " \"bust_results.csv\",\n", |
| 155 | + " )\n", |
156 | 156 | " )\n", |
157 | 157 | "\n", |
158 | 158 | " # NeuralPLexer results\n", |
|
161 | 161 | " f\"neuralplexer_dockgen_outputs_{repeat_index}\",\n", |
162 | 162 | " \"bust_results.csv\",\n", |
163 | 163 | " )\n", |
164 | | - " globals()[\n", |
165 | | - " f\"neuralplexer_dockgen_relaxed_bust_results_csv_filepath_{repeat_index}\"\n", |
166 | | - " ] = os.path.join(\n", |
167 | | - " globals()[\"neuralplexer_output_dir\"],\n", |
168 | | - " f\"neuralplexer_dockgen_outputs_{repeat_index}_relaxed\",\n", |
169 | | - " \"bust_results.csv\",\n", |
| 164 | + " globals()[f\"neuralplexer_dockgen_relaxed_bust_results_csv_filepath_{repeat_index}\"] = (\n", |
| 165 | + " os.path.join(\n", |
| 166 | + " globals()[\"neuralplexer_output_dir\"],\n", |
| 167 | + " f\"neuralplexer_dockgen_outputs_{repeat_index}_relaxed\",\n", |
| 168 | + " \"bust_results.csv\",\n", |
| 169 | + " )\n", |
170 | 170 | " )\n", |
171 | 171 | "\n", |
172 | 172 | " # RoseTTAFold-All-Atom results\n", |
|
199 | 199 | " f\"vina_diffdock_dockgen_outputs_{repeat_index}\",\n", |
200 | 200 | " \"bust_results.csv\",\n", |
201 | 201 | " )\n", |
202 | | - " globals()[\n", |
203 | | - " f\"vina_diffdock_dockgen_relaxed_bust_results_csv_filepath_{repeat_index}\"\n", |
204 | | - " ] = os.path.join(\n", |
205 | | - " globals()[\"vina_output_dir\"],\n", |
206 | | - " f\"vina_diffdock_dockgen_outputs_{repeat_index}_relaxed\",\n", |
207 | | - " \"bust_results.csv\",\n", |
| 202 | + " globals()[f\"vina_diffdock_dockgen_relaxed_bust_results_csv_filepath_{repeat_index}\"] = (\n", |
| 203 | + " os.path.join(\n", |
| 204 | + " globals()[\"vina_output_dir\"],\n", |
| 205 | + " f\"vina_diffdock_dockgen_outputs_{repeat_index}_relaxed\",\n", |
| 206 | + " \"bust_results.csv\",\n", |
| 207 | + " )\n", |
208 | 208 | " )\n", |
209 | 209 | "\n", |
210 | 210 | " # P2Rank-Vina results\n", |
|
213 | 213 | " f\"vina_p2rank_dockgen_outputs_{repeat_index}\",\n", |
214 | 214 | " \"bust_results.csv\",\n", |
215 | 215 | " )\n", |
216 | | - " globals()[\n", |
217 | | - " f\"vina_p2rank_dockgen_relaxed_bust_results_csv_filepath_{repeat_index}\"\n", |
218 | | - " ] = os.path.join(\n", |
219 | | - " globals()[\"vina_output_dir\"],\n", |
220 | | - " f\"vina_p2rank_dockgen_outputs_{repeat_index}_relaxed\",\n", |
221 | | - " \"bust_results.csv\",\n", |
| 216 | + " globals()[f\"vina_p2rank_dockgen_relaxed_bust_results_csv_filepath_{repeat_index}\"] = (\n", |
| 217 | + " os.path.join(\n", |
| 218 | + " globals()[\"vina_output_dir\"],\n", |
| 219 | + " f\"vina_p2rank_dockgen_outputs_{repeat_index}_relaxed\",\n", |
| 220 | + " \"bust_results.csv\",\n", |
| 221 | + " )\n", |
222 | 222 | " )\n", |
223 | 223 | "\n", |
224 | 224 | " # Consensus ensemble results\n", |
225 | | - " globals()[\n", |
226 | | - " f\"consensus_ensemble_dockgen_bust_results_csv_filepath_{repeat_index}\"\n", |
227 | | - " ] = os.path.join(\n", |
228 | | - " os.path.join(\"..\", \"data\", \"test_cases\", \"dockgen\"),\n", |
229 | | - " f\"top_consensus_ensemble_predictions_{repeat_index}\",\n", |
230 | | - " \"bust_results.csv\",\n", |
231 | | - " )\n", |
232 | | - " globals()[\n", |
233 | | - " f\"consensus_ensemble_dockgen_relaxed_bust_results_csv_filepath_{repeat_index}\"\n", |
234 | | - " ] = os.path.join(\n", |
235 | | - " os.path.join(\"..\", \"data\", \"test_cases\", \"dockgen\"),\n", |
236 | | - " f\"top_consensus_ensemble_predictions_{repeat_index}_relaxed\",\n", |
237 | | - " \"bust_results.csv\",\n", |
| 225 | + " globals()[f\"consensus_ensemble_dockgen_bust_results_csv_filepath_{repeat_index}\"] = (\n", |
| 226 | + " os.path.join(\n", |
| 227 | + " os.path.join(\"..\", \"data\", \"test_cases\", \"dockgen\"),\n", |
| 228 | + " f\"top_consensus_ensemble_predictions_{repeat_index}\",\n", |
| 229 | + " \"bust_results.csv\",\n", |
| 230 | + " )\n", |
| 231 | + " )\n", |
| 232 | + " globals()[f\"consensus_ensemble_dockgen_relaxed_bust_results_csv_filepath_{repeat_index}\"] = (\n", |
| 233 | + " os.path.join(\n", |
| 234 | + " os.path.join(\"..\", \"data\", \"test_cases\", \"dockgen\"),\n", |
| 235 | + " f\"top_consensus_ensemble_predictions_{repeat_index}_relaxed\",\n", |
| 236 | + " \"bust_results.csv\",\n", |
| 237 | + " )\n", |
238 | 238 | " )\n", |
239 | 239 | "\n", |
240 | 240 | "# Mappings\n", |
|
0 commit comments