|
353 | 353 | "source": [
|
354 | 354 | "import sys\n",
|
355 | 355 | "if not on_colab and sys.platform.startswith((\"linux\", \"darwin\")):\n",
|
356 |
| - " !conda install -q -y -c conda-forge openmmforcefields" |
| 356 | + " !mamba install -q -y -c conda-forge openmmforcefields\n", |
| 357 | + " # Notes:\n", |
| 358 | + " # - If you do not have mamba installed, install it or use conda instead\n", |
| 359 | + " # - Under MacOS with an M1 chip you may need to use\n", |
| 360 | + " # CONDA_SUBDIR=osx-64 in front of the above command" |
357 | 361 | ]
|
358 | 362 | },
|
359 | 363 | {
|
|
394 | 398 | "from rdkit.Chem import AllChem\n",
|
395 | 399 | "import mdtraj as md\n",
|
396 | 400 | "import pdbfixer\n",
|
397 |
| - "import simtk.openmm as mm\n", |
398 |
| - "import simtk.openmm.app as app\n", |
399 |
| - "from simtk.openmm import unit\n", |
| 401 | + "import openmm as mm\n", |
| 402 | + "import openmm.app as app\n", |
| 403 | + "from openmm import unit\n", |
400 | 404 | "from openff.toolkit.topology import Molecule, Topology\n",
|
401 | 405 | "from openmmforcefields.generators import GAFFTemplateGenerator"
|
402 | 406 | ]
|
|
718 | 722 | " mol_topology = off_mol_topology.to_openmm()\n",
|
719 | 723 | " mol_positions = off_mol.conformers[0]\n",
|
720 | 724 | "\n",
|
721 |
| - " # convert units from Ångström to Nanometers\n", |
722 |
| - " for atom in mol_positions:\n", |
723 |
| - " coords = atom / atom.unit\n", |
724 |
| - " atom = (coords / 10.0) * unit.nanometers # since openmm works in nm\n", |
| 725 | + " # convert units from Ångström to nanometers\n", |
| 726 | + " # since OpenMM works in nm\n", |
| 727 | + " mol_positions = mol_positions.to(\"nanometers\")\n", |
725 | 728 | "\n",
|
726 | 729 | " # combine topology and positions in modeller object\n",
|
727 | 730 | " omm_mol = app.Modeller(mol_topology, mol_positions)\n",
|
|
0 commit comments