Correct Array datatype

bluesky · Nov 4, 2024 · 291544c · 291544c
1 parent d53578a
commit 291544c
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/src/ophyd_async/epics/pmac/_pmac_io.py b/src/ophyd_async/epics/pmac/_pmac_io.py
@@ -1,7 +1,6 @@
 import numpy as np
-import numpy.typing as npt
 
-from ophyd_async.core import DeviceVector, StandardReadable
+from ophyd_async.core import Array1D, DeviceVector, StandardReadable
 from ophyd_async.epics import motor
 
 from ..signal import epics_signal_r, epics_signal_rw
@@ -12,14 +11,14 @@ class Pmac(StandardReadable):
 
     def __init__(self, prefix: str, name: str = "") -> None:
         self.time_array = epics_signal_rw(
-            npt.NDArray[np.float64], prefix + ":ProfileTimeArray"
+            Array1D[np.float64], prefix + ":ProfileTimeArray"
         )
         cs_letters = "ABCUVWXYZ"
         # 1 indexed CS axes so we can index into them from the compound motor input link
         self.positions = DeviceVector(
             {
                 i + 1: epics_signal_rw(
-                    npt.NDArray[np.float64], f"{prefix}:{letter}:Positions"
+                    Array1D[np.float64], f"{prefix}:{letter}:Positions"
                 )
                 for i, letter in enumerate(cs_letters)
             }
@@ -33,7 +32,7 @@ def __init__(self, prefix: str, name: str = "") -> None:
         self.velocities = DeviceVector(
             {
                 i + 1: epics_signal_rw(
-                    npt.NDArray[np.float64], f"{prefix}:{letter}:Velocities"
+                    Array1D[np.float64], f"{prefix}:{letter}:Velocities"
                 )
                 for i, letter in enumerate(cs_letters)
             }