diff --git a/nata/containers/core.py b/nata/containers/core.py index 053d173..3d606ba 100644 --- a/nata/containers/core.py +++ b/nata/containers/core.py @@ -20,8 +20,7 @@ class BackendType(Protocol): name: str @staticmethod - def is_valid_backend(location: Path) -> bool: - ... # pragma: no cover + def is_valid_backend(location: Path) -> bool: ... # pragma: no cover class HasBackends: diff --git a/nata/containers/grid.py b/nata/containers/grid.py index 5cae239..37b654e 100644 --- a/nata/containers/grid.py +++ b/nata/containers/grid.py @@ -72,12 +72,10 @@ class GridBackend(Protocol): name: str location: Path - def __init__(self, location: FileLocation) -> None: - ... + def __init__(self, location: FileLocation) -> None: ... @staticmethod - def is_valid_backend(location: FileLocation) -> bool: - ... + def is_valid_backend(location: FileLocation) -> bool: ... dataset_name: str dataset_label: str @@ -100,8 +98,7 @@ def is_valid_backend(location: FileLocation) -> bool: @runtime_checkable class GridDataReader(GridBackend, Protocol): - def get_data(self, indexing: Optional[BasicIndexing] = None) -> np.ndarray: - ... + def get_data(self, indexing: Optional[BasicIndexing] = None) -> np.ndarray: ... class GridArray( diff --git a/nata/containers/particle.py b/nata/containers/particle.py index 4ee4d10..5ef183e 100644 --- a/nata/containers/particle.py +++ b/nata/containers/particle.py @@ -83,12 +83,10 @@ class ParticleBackend(Protocol): name: str location: Path - def __init__(self, location: FileLocation) -> None: - ... + def __init__(self, location: FileLocation) -> None: ... @staticmethod - def is_valid_backend(location: FileLocation) -> bool: - ... + def is_valid_backend(location: FileLocation) -> bool: ... dataset_name: str dataset_label: str @@ -111,8 +109,7 @@ class ParticleDataReader(ParticleBackend, Protocol): def get_data( self, indexing: Optional[BasicIndexing] = None, - ) -> np.ndarray: - ... + ) -> np.ndarray: ... class Quantity( @@ -800,9 +797,7 @@ def _decay_to_ParticleDataset( time=self.time[index[0]], ) - def __getitem__( - self, key: Any - ) -> Union[ + def __getitem__(self, key: Any) -> Union[ "ParticleDataset", "ParticleArray", "Particle", diff --git a/nata/plots/figure.py b/nata/plots/figure.py index 1b7802a..c976273 100644 --- a/nata/plots/figure.py +++ b/nata/plots/figure.py @@ -101,15 +101,17 @@ def scatter( s=size, c=color if has_colors else None, color=color if has_single_color else None, - norm=mpl_norm_from_scale( - colorscale, - ( - colorrange[0] if colorrange else np.min(color), - colorrange[1] if colorrange else np.max(color), - ), - ) - if has_colors - else None, + norm=( + mpl_norm_from_scale( + colorscale, + ( + colorrange[0] if colorrange else np.min(color), + colorrange[1] if colorrange else np.max(color), + ), + ) + if has_colors + else None + ), cmap=colormap, alpha=alpha, ) diff --git a/nata/types.py b/nata/types.py index 7e332d2..6ee140c 100644 --- a/nata/types.py +++ b/nata/types.py @@ -86,8 +86,7 @@ class BackendType(Protocol): #: point to data, either to open a file or to retrieve it. location: Optional[FileLocation] - def __init__(self, location: FileLocation) -> None: - ... + def __init__(self, location: FileLocation) -> None: ... @staticmethod def is_valid_backend(location: FileLocation) -> bool: diff --git a/nata/utils/zdf.py b/nata/utils/zdf.py index 3757690..b30ad6e 100644 --- a/nata/utils/zdf.py +++ b/nata/utils/zdf.py @@ -738,9 +738,9 @@ def read_track_data(self, trackInfo): for i in range(itermap.shape[0]): trackID = itermap[i, 0] - 1 npoints = itermap[i, 1] - (trackData[trackID])[ - trackNp[trackID] : trackNp[trackID] + npoints, : - ] = data[idx : idx + npoints, :] + (trackData[trackID])[trackNp[trackID] : trackNp[trackID] + npoints, :] = ( + data[idx : idx + npoints, :] + ) trackNp[trackID] += npoints idx += npoints diff --git a/tests/containers/core_test.py b/tests/containers/core_test.py index 5790a76..88ca319 100644 --- a/tests/containers/core_test.py +++ b/tests/containers/core_test.py @@ -23,11 +23,9 @@ class ExtendedProtocol(Protocol): some_other_prop: Tuple[int, ...] @staticmethod - def is_valid_backend(location: Path) -> bool: - ... + def is_valid_backend(location: Path) -> bool: ... - def some_other_method(self, foo: int) -> float: - ... + def some_other_method(self, foo: int) -> float: ... return ExtendedProtocol diff --git a/tests/containers/grid_test.py b/tests/containers/grid_test.py index 4d5bb3c..4f6e7fb 100644 --- a/tests/containers/grid_test.py +++ b/tests/containers/grid_test.py @@ -33,8 +33,7 @@ def __init__(self, location: Union[str, Path]) -> None: raise NotImplementedError @staticmethod - def is_valid_backend(location: Union[str, Path]) -> bool: - ... + def is_valid_backend(location: Union[str, Path]) -> bool: ... dataset_name = str() dataset_label = str()