From 2a230563bea94e42bebf1c8aa969f8db0570d12e Mon Sep 17 00:00:00 2001 From: Tobias Hangleiter <tobias.hangleiter@rwth-aachen.de> Date: Thu, 24 Apr 2025 20:28:15 +0200 Subject: [PATCH] isort --- doc/walkthroughs/walkthrough.py | 3 +-- src/python_spectrometer/_audio_manager.py | 2 +- src/python_spectrometer/_plot_manager.py | 6 +++--- src/python_spectrometer/core.py | 11 +++++------ src/python_spectrometer/daq/__init__.pyi | 15 +++++---------- src/python_spectrometer/daq/atsaverage.py | 10 +++++----- src/python_spectrometer/daq/atssimple.py | 4 ++-- src/python_spectrometer/daq/base.py | 2 +- src/python_spectrometer/daq/qcodes.py | 5 ++--- .../daq/swabian_instruments.py | 2 +- src/python_spectrometer/daq/zurich_instruments.py | 2 +- tests/test_live_view.py | 2 +- tests/test_serialization.py | 3 ++- 13 files changed, 30 insertions(+), 37 deletions(-) diff --git a/doc/walkthroughs/walkthrough.py b/doc/walkthroughs/walkthrough.py index 44215f1..914a278 100755 --- a/doc/walkthroughs/walkthrough.py +++ b/doc/walkthroughs/walkthrough.py @@ -26,8 +26,7 @@ import matplotlib.pyplot as plt import numpy as np import scipy as sc -from python_spectrometer import daq, Spectrometer - +from python_spectrometer import Spectrometer, daq # %% [markdown] # Matplotlib backend, choose `qt` for a separate window. diff --git a/src/python_spectrometer/_audio_manager.py b/src/python_spectrometer/_audio_manager.py index 144c92c..8d34ef3 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 f574136..2547bd6 100644 --- a/src/python_spectrometer/_plot_manager.py +++ b/src/python_spectrometer/_plot_manager.py @@ -4,9 +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 29033b7..cdb93aa 100644 --- a/src/python_spectrometer/core.py +++ b/src/python_spectrometer/core.py @@ -5,13 +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 @@ -24,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 83c9169..c408c10 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 4591168..8bcf9c0 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 3d96ccd..f095a47 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 9d38a42..1687d8b 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 238fa70..f248f8a 100644 --- a/src/python_spectrometer/daq/qcodes.py +++ b/src/python_spectrometer/daq/qcodes.py @@ -4,9 +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 @@ -35,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 1d1050a..8bfc328 100644 --- a/src/python_spectrometer/daq/swabian_instruments.py +++ b/src/python_spectrometer/daq/swabian_instruments.py @@ -4,8 +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 c351b17..90bf079 100644 --- a/src/python_spectrometer/daq/zurich_instruments.py +++ b/src/python_spectrometer/daq/zurich_instruments.py @@ -58,8 +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 diff --git a/tests/test_live_view.py b/tests/test_live_view.py index e5245e5..bfcca0e 100644 --- a/tests/test_live_view.py +++ b/tests/test_live_view.py @@ -11,7 +11,7 @@ from qutil.plotting import is_using_mpl_gui_backend from qutil.plotting.live_view import LiveViewBase, LiveViewT from python_spectrometer import Spectrometer, daq -from python_spectrometer.core import TimeLiveView, FrequencyLiveView +from python_spectrometer.core import FrequencyLiveView, TimeLiveView def start_animation(fig): diff --git a/tests/test_serialization.py b/tests/test_serialization.py index ae664a3..38d08a3 100644 --- a/tests/test_serialization.py +++ b/tests/test_serialization.py @@ -6,9 +6,10 @@ import time from tempfile import mkdtemp import pytest -from python_spectrometer import Spectrometer, daq from qutil.plotting import is_using_mpl_gui_backend +from python_spectrometer import Spectrometer, daq + def remove_file_if_exists(file): try: -- GitLab