diff --git a/modules/base/pymod/__init__.py b/modules/base/pymod/__init__.py index 08f9f9ca50b6e334878ebf3ae9466b38cd54c3ac..3d2137f7e76f6a66a8cf2041319d801742fafedf 100644 --- a/modules/base/pymod/__init__.py +++ b/modules/base/pymod/__init__.py @@ -18,6 +18,11 @@ #------------------------------------------------------------------------------ from _base import * +from ost import geom +from ost import io +from ost import mol +from ost import seq + try: from ost import gfx scene = gfx.Scene() @@ -29,3 +34,4 @@ try: from ost import img except ImportError: pass + diff --git a/modules/mol/base/pymod/export_residue_view.cc b/modules/mol/base/pymod/export_residue_view.cc index 54780c0d3b123787ffaeb8e053c9921d40dfaa23..29943f54a99392eb06afce6d84d91ffe9e0d4da6 100644 --- a/modules/mol/base/pymod/export_residue_view.cc +++ b/modules/mol/base/pymod/export_residue_view.cc @@ -57,6 +57,7 @@ void export_ResidueView() .def("GetChain",&ResidueView::GetChain) .def("GetAtomList", &ResidueView::GetAtomList, return_value_policy<copy_const_reference>()) + .def("GetAtomCount", &ResidueView::GetAtomCount) .add_property("atoms", make_function(&ResidueView::GetAtomList, return_value_policy<copy_const_reference>()))