Update transform.cc wrapper for modern style

lsst · Mar 8, 2017 · 0967862 · 0967862
1 parent 0d09cb8
commit 0967862
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 8 deletions.
diff --git a/python/lsst/afw/geom/__init__.py b/python/lsst/afw/geom/__init__.py
@@ -41,3 +41,4 @@
 from .transformMap import *
 from .utils import *
 from .endpoint import *
+from .transform import *
diff --git a/python/lsst/afw/geom/transform.cc b/python/lsst/afw/geom/transform.cc
@@ -38,7 +38,6 @@ using namespace py::literals;
 namespace lsst {
 namespace afw {
 namespace geom {
-
 namespace {
 
 template<typename Class>
@@ -84,14 +83,10 @@ void declareTransform(py::module& mod, std::string const & fromName, std::string
     });
 }
 
-}  // <anonymous>
-
-PYBIND11_PLUGIN(_transform) {
-    // TODO uncomment these import(s) as part of the afw pybind11 cleanup pass,
-    // once we can make library names not have the leading underscore
-    // py::module::import("lsst.afw.geom.endpoint");
+PYBIND11_PLUGIN(transform) {
+    py::module mod("transform");
 
-    py::module mod("_transform");
+    py::module::import("lsst.afw.geom.endpoint");
 
     // Need to import numpy for ndarray and eigen conversions
     if (_import_array() < 0) {
@@ -119,6 +114,7 @@ PYBIND11_PLUGIN(_transform) {
     return mod.ptr();
 }
 
+}  // <anonymous>
 }  // geom
 }  // afw
 }  // lsst