Skip to content

Commit 96062ff

Browse files
committed
Only import MCF writer if networkx is installed
1 parent 01ffa8e commit 96062ff

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

mbuild/compound.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -28,9 +28,8 @@
2828
from mbuild.formats.lammpsdata import write_lammpsdata
2929
from mbuild.formats.gsdwriter import write_gsd
3030
from mbuild.formats.par_writer import write_par
31-
from mbuild.formats.cassandramcf import write_mcf
3231
from mbuild.periodic_kdtree import PeriodicCKDTree
33-
from mbuild.utils.io import run_from_ipython, import_
32+
from mbuild.utils.io import run_from_ipython, import_, has_networkx
3433
from mbuild.utils.jsutils import overwrite_nglview_default
3534
from mbuild.coordinate_transform import _translate, _rotate
3635

@@ -1912,8 +1911,10 @@ def save(self, filename, show_ports=False, forcefield_name=None,
19121911
'.gsd': write_gsd,
19131912
'.lammps': write_lammpsdata,
19141913
'.lmp': write_lammpsdata,
1915-
'.par': write_par,
1916-
'.mcf': write_mcf}
1914+
'.par': write_par,}
1915+
if has_networkx:
1916+
from mbuild.formats.cassandramcf import write_mcf
1917+
savers.update({'.mcf': write_mcf})
19171918

19181919
try:
19191920
saver = savers[extension]

0 commit comments

Comments
 (0)