added impor_array call

smdogroup · Mar 7, 2024 · c818b17 · c818b17
1 parent 429933a
commit c818b17
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 14 deletions.
diff --git a/tmr/TMR.pxd b/tmr/TMR.pxd
@@ -18,13 +18,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-# For MPI
-cimport mpi4py.MPI as MPI
-
-# Import numpy
-cimport numpy as np
-import numpy as np
-
 # Import TACS
 from tacs import TACS
 from tacs.TACS import Vec, VecInterp, Assembler, Mg, Element, Function, Pc

diff --git a/tmr/TMR.pyx b/tmr/TMR.pyx
@@ -25,6 +25,9 @@ cimport mpi4py.MPI as MPI
 cimport numpy as np
 import numpy as np
 
+# Ensure that numpy is initialized
+np.import_array()
+
 # Import the string library
 from libcpp.string cimport string
 from libc.string cimport strcpy
@@ -49,9 +52,6 @@ cdef tmr_init():
     if not TMRIsInitialized():
         TMRInitialize()
 
-# Ensure that numpy is initialized
-np.import_array()
-
 # Initialize the MPI libraries in TMR (if not already done)
 tmr_init()
 

diff --git a/tmr/cpp_headers/TMR.pxd b/tmr/cpp_headers/TMR.pxd
@@ -26,10 +26,6 @@ cimport mpi4py.MPI as MPI
 from libc.string cimport const_char
 from libc.stdint cimport int32_t, int16_t
 
-# Import numpy
-cimport numpy as np
-import numpy as np
-
 # Import the definitions from the headers
 from paropt.cpp_headers.ParOpt cimport ParOptScalar, ParOptVec, ParOptProblem
 from tacs.cpp_headers.TACS cimport (