Merge pull request #117 from Adnan-Ali-Ahmad/vaytet_osyris

Fix pint import error

src/osyris/core/array.py

@@ -1,7 +1,7 @@
 # SPDX-License-Identifier: BSD-3-Clause
 # Copyright (c) 2022 Osyris contributors (https://github.com/osyris-project/osyris)
 import numpy as np
-from pint.quantity import Quantity
+from pint import Quantity
 from pint.errors import DimensionalityError
 from .base import Base
 from .tools import value_to_string

src/osyris/core/vector.py

@@ -1,7 +1,7 @@
 # SPDX-License-Identifier: BSD-3-Clause
 # Copyright (c) 2022 Osyris contributors (https://github.com/osyris-project/osyris)
 import numpy as np
-from pint.quantity import Quantity
+from pint import Quantity
 from .base import Base
 from .array import Array
 from .tools import value_to_string

src/osyris/plot/histogram2d.py

@@ -2,7 +2,7 @@
 # Copyright (c) 2022 Osyris contributors (https://github.com/osyris-project/osyris)
 
 import numpy as np
-from pint.quantity import Quantity
+from pint import Quantity
 from typing import Union
 from ..core import Array, Plot
 from .render import render

src/osyris/plot/map.py

@@ -3,7 +3,7 @@
 
 import numpy as np
 import numpy.ma as ma
-from pint.quantity import Quantity
+from pint import Quantity
 from typing import Union
 from .direction import get_direction
 from .render import render

src/osyris/plot/scatter.py

@@ -1,7 +1,7 @@
 # SPDX-License-Identifier: BSD-3-Clause
 # Copyright (c) 2022 Osyris contributors (https://github.com/osyris-project/osyris)
 
-from pint.quantity import Quantity
+from pint import Quantity
 from ..core import Plot, Array
 from .. import units
 from .render import render

src/osyris/plot/wrappers.py

@@ -8,7 +8,7 @@
 from matplotlib.cm import ScalarMappable
 from matplotlib.collections import PatchCollection
 import numpy as np
-from pint.quantity import Quantity
+from pint import Quantity
 
 
 def _add_colorbar(obj, ax, cax=None, label=None):
@@ -41,7 +41,7 @@ def quiver(ax,
     default_args = {"angles": "xy", "pivot": "mid", "zorder": zorder}
     default_args.update(kwargs)
 
-    skips = np.around(np.array(z.shape) * 4.0 / 128.0 / density).astype(np.int)
+    skips = np.around(np.array(z.shape) * 4.0 / 128.0 / density).astype(int)
     skip = (slice(None, None, skips[0]), slice(None, None, skips[1]))
 
     args = [x[skip[0]], y[skip[1]], z[..., 0][skip], z[..., 1][skip]]