finish tutorial 14

deepchem · Mar 9, 2021 · bf441f5 · bf441f5
1 parent 698519d
commit bf441f5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 325 deletions.
diff --git a/deepchem/feat/complex_featurizers/complex_atomic_coordinates.py b/deepchem/feat/complex_featurizers/complex_atomic_coordinates.py
@@ -19,7 +19,10 @@ def compute_neighbor_list(coords, neighbor_cutoff, max_num_neighbors,
                           periodic_box_size):
   """Computes a neighbor list from atom coordinates."""
   N = coords.shape[0]
-  import mdtraj
+  try:
+    import mdtraj
+  except ModuleNotFoundError:
+    raise ImportError("This function requires mdtraj to be installed.")
   traj = mdtraj.Trajectory(coords.reshape((1, N, 3)), None)
   box_size = None
   if periodic_box_size is not None: