From a97f49ebe3c8e28d2a9ba76711555a3378b62341 Mon Sep 17 00:00:00 2001
From: J08nY
Date: Wed, 4 Mar 2020 00:04:53 +0100
Subject: Fix some type issues.

---
 pyecsca/sca/scope/picoscope_sdk.py | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'pyecsca/sca/scope')

diff --git a/pyecsca/sca/scope/picoscope_sdk.py b/pyecsca/sca/scope/picoscope_sdk.py
index b21df09..a198367 100644
--- a/pyecsca/sca/scope/picoscope_sdk.py
+++ b/pyecsca/sca/scope/picoscope_sdk.py
@@ -159,6 +159,8 @@ class PicoScopeSdk(Scope):  # pragma: no cover
         return True
 
     def retrieve(self, channel: str) -> Optional[np.ndarray]:
+        if self.samples is None:
+            raise ValueError
         actual_samples = ctypes.c_int32(self.samples)
         overflow = ctypes.c_int16()
         assert_pico_ok(
@@ -179,6 +181,7 @@ class PicoScopeSdk(Scope):  # pragma: no cover
             raise ValueError
         return method(*args, **kwargs)
 
+
 @public
 class PS5000Scope(PicoScopeSdk):  # pragma: no cover
     MODULE = ps5000
-- 
cgit v1.3.1