diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index b7f280f..eb0d89f 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -6,7 +6,7 @@ repos: - id: end-of-file-fixer - id: trailing-whitespace - repo: https://github.com/psf/black - rev: 23.12.1 + rev: 24.1.1 hooks: - id: black files: ^src/ diff --git a/src/probeinterface/generator.py b/src/probeinterface/generator.py index d363beb..95b3b1c 100644 --- a/src/probeinterface/generator.py +++ b/src/probeinterface/generator.py @@ -2,6 +2,7 @@ This module contains useful helper functions for generating probes. """ + from __future__ import annotations import numpy as np diff --git a/src/probeinterface/io.py b/src/probeinterface/io.py index fa170a5..7aa2c5b 100644 --- a/src/probeinterface/io.py +++ b/src/probeinterface/io.py @@ -9,6 +9,7 @@ * Neurodata Without Borders (.nwb) """ + from __future__ import annotations from pathlib import Path from typing import Union, Optional diff --git a/src/probeinterface/library.py b/src/probeinterface/library.py index 2374c3c..04e3442 100644 --- a/src/probeinterface/library.py +++ b/src/probeinterface/library.py @@ -8,6 +8,7 @@ The gin platform enables contributions from users. """ + from __future__ import annotations import os from pathlib import Path diff --git a/src/probeinterface/plotting.py b/src/probeinterface/plotting.py index 999a3fa..fea326c 100644 --- a/src/probeinterface/plotting.py +++ b/src/probeinterface/plotting.py @@ -4,6 +4,7 @@ Depending on Probe.ndim, the plotting is done in 2D or 3D """ + from __future__ import annotations import numpy as np from matplotlib import path as mpl_path diff --git a/src/probeinterface/utils.py b/src/probeinterface/utils.py index 35b1411..a9be5ba 100644 --- a/src/probeinterface/utils.py +++ b/src/probeinterface/utils.py @@ -1,6 +1,7 @@ """ Some utility functions """ + from __future__ import annotations from importlib import import_module from types import ModuleType diff --git a/src/probeinterface/wiring.py b/src/probeinterface/wiring.py index cdd51b7..1d0a691 100644 --- a/src/probeinterface/wiring.py +++ b/src/probeinterface/wiring.py @@ -1,6 +1,7 @@ """ Automatically set the `Probe.device_channel_indices` field. """ + from __future__ import annotations import numpy as np