diff --git a/src/python_spectrometer/_audio_manager.py b/src/python_spectrometer/_audio_manager.py index 144c92cc9f6562b9153556b0ebce89ee37df93a9..8d34ef328e95acfe05330a8b4767b058f2ed5dfa 100644 --- a/src/python_spectrometer/_audio_manager.py +++ b/src/python_spectrometer/_audio_manager.py @@ -3,7 +3,7 @@ import os import queue import threading import time -from typing import Union, Literal +from typing import Literal, Union import numpy as np from scipy import signal diff --git a/src/python_spectrometer/_plot_manager.py b/src/python_spectrometer/_plot_manager.py index 4811473f0b48a553577e2594662973afffde42b4..5c5dfb46920d22c82954565e2995ba650c662eda 100644 --- a/src/python_spectrometer/_plot_manager.py +++ b/src/python_spectrometer/_plot_manager.py @@ -4,10 +4,9 @@ import os import sys import warnings import weakref -from typing import (Dict, Any, Optional, Tuple, ContextManager, Union, List, - Literal, TypeVar, Callable) - -from collections.abc import Mapping, Iterable +from collections.abc import Iterable, Mapping +from typing import (Any, Callable, ContextManager, Dict, List, Literal, Optional, Tuple, TypeVar, + Union) import matplotlib.pyplot as plt import numpy as np diff --git a/src/python_spectrometer/core.py b/src/python_spectrometer/core.py index 125c41933059d4b13a549a2a1f03ae05c4e62331..3ae2208897e7301977fde90cfcaa3c3386db76e8 100644 --- a/src/python_spectrometer/core.py +++ b/src/python_spectrometer/core.py @@ -5,14 +5,13 @@ import platform import shelve import sys import warnings +from collections.abc import Generator, Iterator, Mapping, Sequence from datetime import datetime from pathlib import Path from pprint import pprint -from queue import Queue, Empty, LifoQueue -from threading import Thread, Event -from typing import (Any, Callable, Dict, List, Literal, Optional, Tuple, Union, cast) - -from collections.abc import Generator, Iterator, Mapping, Sequence +from queue import Empty, LifoQueue, Queue +from threading import Event, Thread +from typing import Any, Callable, Dict, List, Literal, Optional, Tuple, Union, cast from unittest import mock import dill @@ -25,8 +24,7 @@ from qutil import io, misc from qutil.functools import cached_property, chain, partial from qutil.io import AsyncDatasaver from qutil.itertools import count -from qutil.plotting import is_using_mpl_gui_backend -from qutil.plotting import live_view +from qutil.plotting import is_using_mpl_gui_backend, live_view from qutil.signal_processing.real_space import Id, welch from qutil.typecheck import check_literals from qutil.ui import progressbar diff --git a/src/python_spectrometer/daq/__init__.pyi b/src/python_spectrometer/daq/__init__.pyi index 83c91696a8ddb3effe78cc899bb549e0ec6aad4a..c408c10c0369d5a32992f704387236dab18a0b22 100644 --- a/src/python_spectrometer/daq/__init__.pyi +++ b/src/python_spectrometer/daq/__init__.pyi @@ -5,16 +5,11 @@ __all__ = ['atsaverage', 'atssimple', 'qcodes', 'simulator', 'swabian_instrument from . import atsaverage, atssimple, qcodes, simulator, swabian_instruments, zurich_instruments from .atsaverage import AlazarATS9xx0 -from .qcodes import (Keysight344xxA, NationalInstrumentsUSB6003, - # Backwards "compatibility" +from .qcodes import (Keysight344xxA, NationalInstrumentsUSB6003, # Backwards "compatibility" keysight_344xxA, national_instruments_daq) from .settings import DAQSettings -from .simulator import (QoptColoredNoise, - # Backwards "compatibility" - qopt_colored_noise) -from .swabian_instruments import (SwabianInstrumentsTimeTagger, - # Backwards "compatibility" +from .simulator import QoptColoredNoise, qopt_colored_noise # Backwards "compatibility" +from .swabian_instruments import (SwabianInstrumentsTimeTagger, # Backwards "compatibility" timetagger) -from .zurich_instruments import (ZurichInstrumentsMFLIDAQ, ZurichInstrumentsMFLIScope, - # Backwards "compatibility" - MFLI_daq, MFLI_scope) +from .zurich_instruments import (MFLI_daq, MFLI_scope, # Backwards "compatibility" + ZurichInstrumentsMFLIDAQ, ZurichInstrumentsMFLIScope) diff --git a/src/python_spectrometer/daq/atsaverage.py b/src/python_spectrometer/daq/atsaverage.py index 4591168367cf532491b0ce7337829002ed5bbc18..8bcf9c04699959c431b898fd4e672bade6e422f6 100644 --- a/src/python_spectrometer/daq/atsaverage.py +++ b/src/python_spectrometer/daq/atsaverage.py @@ -20,7 +20,7 @@ from __future__ import annotations import dataclasses import string import time -from typing import Callable, Type, Literal +from typing import Callable, Literal, Type from qutil.domains import ReciprocalDiscreteInterval from qutil.functools import cached_property @@ -34,13 +34,13 @@ except ImportError: from numpy import array as NDArray try: + from atsaverage.alazar import InputRangeID + from atsaverage.config2 import (BoardConfiguration, CaptureClockConfiguration, + CaptureClockType, Channel, EngineTriggerConfiguration, + InputConfiguration, SampleRateID, create_scanline_definition) from atsaverage.core import AlazarCard from atsaverage.masks import PeriodicMask from atsaverage.operations import Downsample - from atsaverage.alazar import InputRangeID - from atsaverage.config2 import (BoardConfiguration, EngineTriggerConfiguration, - CaptureClockConfiguration, CaptureClockType, SampleRateID, - InputConfiguration, Channel, create_scanline_definition) except ImportError as e: raise RuntimeError('This DAQ requires the atsaverage library. Clone it from ' 'https://git.rwth-aachen.de/qutech/cpp-atsaverage/') from e diff --git a/src/python_spectrometer/daq/atssimple.py b/src/python_spectrometer/daq/atssimple.py index 3d96ccd1b1169565080f0886b7a5f39b4289f728..f095a47b3293ddca5abb4984001603cec7475461 100644 --- a/src/python_spectrometer/daq/atssimple.py +++ b/src/python_spectrometer/daq/atssimple.py @@ -38,7 +38,7 @@ from __future__ import annotations import dataclasses import time -from typing import Type, Literal, Callable +from typing import Callable, Literal, Type import numpy as np from qutil.domains import ReciprocalDiscreteInterval @@ -53,7 +53,7 @@ except ImportError: from numpy import array as NDArray try: - from atssimple import atsapi, acquire_downsample_windows, ATSSimpleCard + from atssimple import ATSSimpleCard, acquire_downsample_windows, atsapi except ImportError as e: raise RuntimeError('This DAQ requires the atssimple library. Clone it from ' 'https://git-ce.rwth-aachen.de/qutech/lab_software/atssimple/') from e diff --git a/src/python_spectrometer/daq/base.py b/src/python_spectrometer/daq/base.py index 9d38a4230308311583b33b8b95162fe220884bd3..1687d8b6958c2e386aeb134d2a929c667cead12c 100644 --- a/src/python_spectrometer/daq/base.py +++ b/src/python_spectrometer/daq/base.py @@ -5,7 +5,7 @@ import sys import warnings from abc import ABC from collections.abc import Generator -from typing import Any, Dict, Type, Optional +from typing import Any, Dict, Optional, Type import numpy as np from qutil.functools import cached_property diff --git a/src/python_spectrometer/daq/qcodes.py b/src/python_spectrometer/daq/qcodes.py index 573e19b76336e43b1176ca46c5d12fca2d47c168..f248f8abba79c3f4f0a70c2a3e3d55b165805bc1 100644 --- a/src/python_spectrometer/daq/qcodes.py +++ b/src/python_spectrometer/daq/qcodes.py @@ -4,10 +4,9 @@ import dataclasses import logging import sys import warnings +from collections.abc import Sequence from types import ModuleType from typing import Any, Dict, Type - -from collections.abc import Sequence from unittest.mock import Mock import numpy as np @@ -36,9 +35,8 @@ else: from qcodes.instrument_drivers.Keysight import Keysight344xxA as _Keysight344xxA_qcodes except ImportError: # Maybe qcodes < 0.46 and name not available in top-level Keysight module - from qcodes.instrument_drivers.Keysight.private.Keysight_344xxA_submodules import ( + from qcodes.instrument_drivers.Keysight.private.Keysight_344xxA_submodules import \ _Keysight_344xxA as _Keysight344xxA_qcodes - ) try: from qcodes_contrib_drivers.drivers.Andor import AndorIDus4xx as _AndorIDus4xx_qcodes except ImportError: diff --git a/src/python_spectrometer/daq/swabian_instruments.py b/src/python_spectrometer/daq/swabian_instruments.py index b983525f42e8e59c89f0ce52dda2447a92ef4baf..8bfc328bd2ebb8f087baa32cf3be600df932ee89 100644 --- a/src/python_spectrometer/daq/swabian_instruments.py +++ b/src/python_spectrometer/daq/swabian_instruments.py @@ -4,9 +4,8 @@ from __future__ import annotations import dataclasses import sys import warnings -from typing import Any, Dict - from collections.abc import Iterable, Sequence +from typing import Any, Dict from .base import DAQ, AcquisitionGenerator diff --git a/src/python_spectrometer/daq/zurich_instruments.py b/src/python_spectrometer/daq/zurich_instruments.py index f302fbd97534444d70b803b569e3f025858c78e8..90bf079c0570d710453da73b5e1a37e4a11c913d 100644 --- a/src/python_spectrometer/daq/zurich_instruments.py +++ b/src/python_spectrometer/daq/zurich_instruments.py @@ -58,9 +58,8 @@ import sys import time import warnings from abc import ABC -from typing import Any, Dict, Optional, Type - from collections.abc import Mapping +from typing import Any, Dict, Optional, Type import numpy as np from packaging import version