diff --git a/piker/ui/_graphics/_curve.py b/piker/ui/_graphics/_curve.py index 9e6d08e6..49b93910 100644 --- a/piker/ui/_graphics/_curve.py +++ b/piker/ui/_graphics/_curve.py @@ -20,7 +20,7 @@ Fast, smooth, sexy curves. from typing import Tuple import pyqtgraph as pg -from PyQt5 import QtCore, QtGui +from PyQt5 import QtCore, QtGui, QtWidgets from ..._profile import pg_profile_enabled @@ -139,7 +139,12 @@ class FastAppendCurve(pg.PlotCurveItem): # print(f'bounding rect: {br}') return br - def paint(self, p, opt, widget): + def paint( + self, + p: QtGui.QPainter, + opt: QtWidgets.QStyleOptionGraphicsItem, + w: QtWidgets.QWidget + ) -> None: profiler = pg.debug.Profiler(disabled=not pg_profile_enabled()) # p.setRenderHint(p.Antialiasing, True) diff --git a/piker/ui/_graphics/_ohlc.py b/piker/ui/_graphics/_ohlc.py index e96aae90..0be7853f 100644 --- a/piker/ui/_graphics/_ohlc.py +++ b/piker/ui/_graphics/_ohlc.py @@ -22,7 +22,7 @@ from typing import List, Optional, Tuple import numpy as np import pyqtgraph as pg from numba import jit, float64, int64 # , optional -from PyQt5 import QtCore, QtGui +from PyQt5 import QtCore, QtGui, QtWidgets from PyQt5.QtCore import QLineF, QPointF # from numba import types as ntypes # from ..data._source import numba_ohlc_dtype @@ -339,9 +339,14 @@ class BarItems(pg.GraphicsObject): if flip_cache: self.setCacheMode(QtGui.QGraphicsItem.DeviceCoordinateCache) - def paint(self, p, opt, widget): + def paint( + self, + p: QtGui.QPainter, + opt: QtWidgets.QStyleOptionGraphicsItem, + w: QtWidgets.QWidget + ) -> None: - profiler = pg.debug.Profiler(disabled=not pg_profile_enabled()) # , delayed=False) + profiler = pg.debug.Profiler(disabled=not pg_profile_enabled()) # p.setCompositionMode(0) p.setPen(self.bars_pen)