fix mypy errors and avoid copy

ap-- · Sep 5, 2024 · c393d58 · c393d58
1 parent 5c660f5
commit c393d58
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/seabreeze/pyseabreeze/features/spectrometer.py b/src/seabreeze/pyseabreeze/features/spectrometer.py
@@ -745,12 +745,12 @@ class SeaBreezeSpectrometerFeatureHR4(SeaBreezeSpectrometerFeatureOBP2):
 
 
 class SeaBreezeSpectrometerFeatureFX(SeaBreezeSpectrometerFeatureOBP):
-    def _get_spectrum_raw(self):
+    def _get_spectrum_raw(self) -> NDArray[np.uint8]:
         timeout = int(
             self._integration_time_max * 1e-3
             + self.protocol.transport.default_timeout_ms
         )
         # the message type is different than the default defined in the protocol,
         # requires addition of a new message type in protocol to work
         datastring = self.protocol.query(0x00101000, timeout_ms=timeout)
-        return numpy.fromstring(datastring, dtype=numpy.uint8)
+        return numpy.frombuffer(datastring, dtype=numpy.uint8)