Skip to content

Commit be7d4f8

Browse files
authored
Make notebook transformations identical to CLI (#61)
* Obvious additions after run-through * Make notebook transformations identical to CLI
1 parent 87894d5 commit be7d4f8

File tree

1 file changed

+35
-35
lines changed

1 file changed

+35
-35
lines changed

rbfe_tutorial/python_tutorial.ipynb

Lines changed: 35 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -44,9 +44,7 @@
4444
"source": [
4545
"from rdkit import Chem\n",
4646
"supp = Chem.SDMolSupplier(\"tyk2_ligands.sdf\", removeHs=False)\n",
47-
"ligands = [openfe.SmallMoleculeComponent.from_rdkit(mol) for mol in supp]\n",
48-
"\n",
49-
"name_to_ligand = {ligand.name: ligand for ligand in ligands}"
47+
"ligands = [openfe.SmallMoleculeComponent.from_rdkit(mol) for mol in supp]"
5048
]
5149
},
5250
{
@@ -173,7 +171,7 @@
173171
},
174172
{
175173
"cell_type": "code",
176-
"execution_count": 8,
174+
"execution_count": 7,
177175
"id": "2263838f",
178176
"metadata": {},
179177
"outputs": [],
@@ -212,7 +210,7 @@
212210
},
213211
{
214212
"cell_type": "code",
215-
"execution_count": 9,
213+
"execution_count": 8,
216214
"id": "9d2fbc22",
217215
"metadata": {},
218216
"outputs": [],
@@ -223,7 +221,7 @@
223221
},
224222
{
225223
"cell_type": "code",
226-
"execution_count": 10,
224+
"execution_count": 9,
227225
"id": "3f1706ee",
228226
"metadata": {},
229227
"outputs": [],
@@ -233,7 +231,7 @@
233231
},
234232
{
235233
"cell_type": "code",
236-
"execution_count": 11,
234+
"execution_count": 10,
237235
"id": "710285ca",
238236
"metadata": {},
239237
"outputs": [],
@@ -262,7 +260,7 @@
262260
},
263261
{
264262
"cell_type": "code",
265-
"execution_count": 12,
263+
"execution_count": 11,
266264
"id": "3f394a0d",
267265
"metadata": {
268266
"scrolled": true
@@ -282,7 +280,7 @@
282280
},
283281
{
284282
"cell_type": "code",
285-
"execution_count": 13,
283+
"execution_count": 12,
286284
"id": "fb839094",
287285
"metadata": {},
288286
"outputs": [
@@ -298,7 +296,7 @@
298296
"298.15 <Unit('kelvin')>"
299297
]
300298
},
301-
"execution_count": 13,
299+
"execution_count": 12,
302300
"metadata": {},
303301
"output_type": "execute_result"
304302
}
@@ -310,7 +308,7 @@
310308
},
311309
{
312310
"cell_type": "code",
313-
"execution_count": 14,
311+
"execution_count": 13,
314312
"id": "e83630f0",
315313
"metadata": {},
316314
"outputs": [],
@@ -331,7 +329,7 @@
331329
},
332330
{
333331
"cell_type": "code",
334-
"execution_count": 15,
332+
"execution_count": 14,
335333
"id": "7adf42d6",
336334
"metadata": {},
337335
"outputs": [],
@@ -352,7 +350,7 @@
352350
},
353351
{
354352
"cell_type": "code",
355-
"execution_count": 16,
353+
"execution_count": 15,
356354
"id": "44ba94ca",
357355
"metadata": {},
358356
"outputs": [],
@@ -388,7 +386,7 @@
388386
},
389387
{
390388
"cell_type": "code",
391-
"execution_count": 17,
389+
"execution_count": 16,
392390
"id": "66666a80",
393391
"metadata": {},
394392
"outputs": [],
@@ -410,12 +408,14 @@
410408
" sysA = openfe.ChemicalSystem(sysA_dict, name=f\"{mapping.componentA.name}_{leg}\")\n",
411409
" sysB = openfe.ChemicalSystem(sysB_dict, name=f\"{mapping.componentB.name}_{leg}\")\n",
412410
" \n",
411+
" prefix = \"easy_rbfe_\" # prefix is only to exactly reproduce CLI\n",
412+
" \n",
413413
" transformation = openfe.Transformation(\n",
414414
" stateA=sysA,\n",
415415
" stateB=sysB,\n",
416416
" mapping={'ligand': mapping},\n",
417417
" protocol=protocol, # use protocol created above\n",
418-
" name=f\"{sysA.name}_{sysB.name}\"\n",
418+
" name=f\"{prefix}{sysA.name}_{sysB.name}\"\n",
419419
" )\n",
420420
" transformations.append(transformation)\n",
421421
"\n",
@@ -434,7 +434,7 @@
434434
},
435435
{
436436
"cell_type": "code",
437-
"execution_count": 18,
437+
"execution_count": 17,
438438
"id": "d6cebd9a",
439439
"metadata": {},
440440
"outputs": [],
@@ -451,32 +451,32 @@
451451
},
452452
{
453453
"cell_type": "code",
454-
"execution_count": 19,
454+
"execution_count": 18,
455455
"id": "b96b57a9",
456456
"metadata": {},
457457
"outputs": [
458458
{
459459
"name": "stdout",
460460
"output_type": "stream",
461461
"text": [
462-
"lig_ejm_31_complex_lig_ejm_42_complex.json\r\n",
463-
"lig_ejm_31_complex_lig_ejm_46_complex.json\r\n",
464-
"lig_ejm_31_complex_lig_ejm_47_complex.json\r\n",
465-
"lig_ejm_31_complex_lig_ejm_48_complex.json\r\n",
466-
"lig_ejm_31_complex_lig_ejm_50_complex.json\r\n",
467-
"lig_ejm_31_solvent_lig_ejm_42_solvent.json\r\n",
468-
"lig_ejm_31_solvent_lig_ejm_46_solvent.json\r\n",
469-
"lig_ejm_31_solvent_lig_ejm_47_solvent.json\r\n",
470-
"lig_ejm_31_solvent_lig_ejm_48_solvent.json\r\n",
471-
"lig_ejm_31_solvent_lig_ejm_50_solvent.json\r\n",
472-
"lig_ejm_42_complex_lig_ejm_43_complex.json\r\n",
473-
"lig_ejm_42_solvent_lig_ejm_43_solvent.json\r\n",
474-
"lig_ejm_46_complex_lig_jmc_23_complex.json\r\n",
475-
"lig_ejm_46_complex_lig_jmc_27_complex.json\r\n",
476-
"lig_ejm_46_complex_lig_jmc_28_complex.json\r\n",
477-
"lig_ejm_46_solvent_lig_jmc_23_solvent.json\r\n",
478-
"lig_ejm_46_solvent_lig_jmc_27_solvent.json\r\n",
479-
"lig_ejm_46_solvent_lig_jmc_28_solvent.json\r\n"
462+
"easy_rbfe_lig_ejm_31_complex_lig_ejm_42_complex.json\r\n",
463+
"easy_rbfe_lig_ejm_31_complex_lig_ejm_46_complex.json\r\n",
464+
"easy_rbfe_lig_ejm_31_complex_lig_ejm_47_complex.json\r\n",
465+
"easy_rbfe_lig_ejm_31_complex_lig_ejm_48_complex.json\r\n",
466+
"easy_rbfe_lig_ejm_31_complex_lig_ejm_50_complex.json\r\n",
467+
"easy_rbfe_lig_ejm_31_solvent_lig_ejm_42_solvent.json\r\n",
468+
"easy_rbfe_lig_ejm_31_solvent_lig_ejm_46_solvent.json\r\n",
469+
"easy_rbfe_lig_ejm_31_solvent_lig_ejm_47_solvent.json\r\n",
470+
"easy_rbfe_lig_ejm_31_solvent_lig_ejm_48_solvent.json\r\n",
471+
"easy_rbfe_lig_ejm_31_solvent_lig_ejm_50_solvent.json\r\n",
472+
"easy_rbfe_lig_ejm_42_complex_lig_ejm_43_complex.json\r\n",
473+
"easy_rbfe_lig_ejm_42_solvent_lig_ejm_43_solvent.json\r\n",
474+
"easy_rbfe_lig_ejm_46_complex_lig_jmc_23_complex.json\r\n",
475+
"easy_rbfe_lig_ejm_46_complex_lig_jmc_27_complex.json\r\n",
476+
"easy_rbfe_lig_ejm_46_complex_lig_jmc_28_complex.json\r\n",
477+
"easy_rbfe_lig_ejm_46_solvent_lig_jmc_23_solvent.json\r\n",
478+
"easy_rbfe_lig_ejm_46_solvent_lig_jmc_27_solvent.json\r\n",
479+
"easy_rbfe_lig_ejm_46_solvent_lig_jmc_28_solvent.json\r\n"
480480
]
481481
}
482482
],

0 commit comments

Comments
 (0)