Additional pybind11 cleanup

lsst · Apr 6, 2017 · 72bc62e · 72bc62e
1 parent 0803f08
commit 72bc62e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/python/lsst/meas/extensions/photometryKron/__init__.py b/python/lsst/meas/extensions/photometryKron/__init__.py
@@ -20,7 +20,7 @@
 # see <http://www.lsstcorp.org/LegalNotices/>.
 #
 
-from __future__ import absolute_import
+from __future__ import absolute_import, division, print_function
 
 from lsst.meas.base import BasePlugin, wrapSimpleAlgorithm
 from .photometryKron import KronFluxAlgorithm, KronFluxControl, KronAperture

diff --git a/python/lsst/meas/extensions/photometryKron/photometryKron.cc b/python/lsst/meas/extensions/photometryKron/photometryKron.cc
@@ -130,6 +130,11 @@ void declareKronAperture(py::module &mod) {
 }  // <anonymous>
 
 PYBIND11_PLUGIN(photometryKron) {
+    py::module::import("lsst.afw.geom");
+    py::module::import("lsst.afw.image");
+    py::module::import("lsst.afw.table");
+    py::module::import("lsst.daf.base");
+
     py::module mod("photometryKron");
 
     declareKronFluxControl(mod);