diff --git a/gunpowder/nodes/__init__.py b/gunpowder/nodes/__init__.py index 1f10c25c..42cbf46f 100644 --- a/gunpowder/nodes/__init__.py +++ b/gunpowder/nodes/__init__.py @@ -46,4 +46,4 @@ from .zarr_source import ZarrSource from .zarr_write import ZarrWrite from .gp_array_source import ArraySource as GPArraySource -from .gp_graph_source import GraphSource as GPGraphSource \ No newline at end of file +from .gp_graph_source import GraphSource as GPGraphSource diff --git a/gunpowder/nodes/gp_array_source.py b/gunpowder/nodes/gp_array_source.py index d73572e8..0277abd3 100644 --- a/gunpowder/nodes/gp_array_source.py +++ b/gunpowder/nodes/gp_array_source.py @@ -22,4 +22,3 @@ def provide(self, request): else: outputs[self.key] = copy.deepcopy(self.array.crop(request[self.key].roi)) return outputs - diff --git a/gunpowder/nodes/random_location.py b/gunpowder/nodes/random_location.py index e93949ac..192d7722 100644 --- a/gunpowder/nodes/random_location.py +++ b/gunpowder/nodes/random_location.py @@ -459,13 +459,11 @@ def __select_random_location_with_points( # count all points inside the shifted ROI points = self.__get_points_in_roi(request_points_roi.shift(random_shift)) - assert point in points, ( - "Requested batch to contain point %s, but got points " - "%s" - % ( - point, - points, - ) + assert ( + point in points + ), "Requested batch to contain point %s, but got points " "%s" % ( + point, + points, ) return random_shift diff --git a/tests/cases/random_location.py b/tests/cases/random_location.py index e60d1ec5..52c4ccce 100644 --- a/tests/cases/random_location.py +++ b/tests/cases/random_location.py @@ -96,7 +96,7 @@ def test_output(): ), b: ArraySpec( roi=Roi(batch[random_shift_key].data, (20, 20, 20)) - ) + ), } ) )