diff --git a/python/lammps/numpy_wrapper.py b/python/lammps/numpy_wrapper.py index 91042c43c8..4e33913a69 100644 --- a/python/lammps/numpy_wrapper.py +++ b/python/lammps/numpy_wrapper.py @@ -19,11 +19,15 @@ import warnings from ctypes import POINTER, c_void_p, c_char_p, c_double, c_int, c_int32, c_int64, cast +from .constants import LAMMPS_AUTODETECT, LAMMPS_INT, LAMMPS_INT_2D, \ + LAMMPS_DOUBLE, LAMMPS_DOUBLE_2D, LAMMPS_INT64, LAMMPS_INT64_2D, \ + LMP_STYLE_GLOBAL, LMP_STYLE_ATOM, LMP_STYLE_LOCAL, \ + LMP_TYPE_SCALAR, LMP_TYPE_VECTOR, LMP_TYPE_ARRAY, \ + LMP_SIZE_VECTOR, LMP_SIZE_ROWS, LMP_SIZE_COLS, \ + LMP_VAR_EQUAL, LMP_VAR_ATOM, LMP_VAR_VECTOR, LMP_VAR_STRING -from .constants import * # lgtm [py/polluting-import] from .data import NeighList - class numpy_wrapper: """lammps API NumPy Wrapper diff --git a/python/lammps/pylammps.py b/python/lammps/pylammps.py index 7dd3823bbf..4b740f35ff 100644 --- a/python/lammps/pylammps.py +++ b/python/lammps/pylammps.py @@ -28,7 +28,7 @@ import tempfile from collections import namedtuple from .core import lammps -from .constants import * # lgtm [py/polluting-import] +from .constants import LMP_VAR_EQUAL, LMP_VAR_ATOM, LMP_VAR_VECTOR, LMP_VAR_STRING # -------------------------------------------------------------------------