Fixed type of context manager

This commit is contained in:
Richard Bowman 2021-04-21 12:54:53 +01:00
parent fd32c22e30
commit 8c50f73b34

View file

@ -1,6 +1,6 @@
import logging
from contextlib import contextmanager
from typing import Tuple
from typing import Tuple, Generator
import labthings.fields as fields
from flask import abort
@ -88,7 +88,7 @@ class LSTExtension(BaseExtension):
@contextmanager
def find_picamera() -> Tuple[PiCamera, BaseCamera, Microscope]:
def find_picamera() -> Generator[Tuple[PiCamera, BaseCamera, Microscope], None, None]:
"""Locate the microscope and raise a sensible error if it's missing."""
microscope = find_component("org.openflexure.microscope")