Rasterize_graph errors when mask is passed

[Mohinta2892]

Hello!

I am getting the 2 errors when I run tests/cases/rasterize_points.py with a mask passed as an argument as below:

Test code snippet:

        pipeline = GraphTestSource3D() + RasterizeGraph(
            GraphKeys.TEST_GRAPH,
            ArrayKeys.RASTERIZED,
            ArraySpec(voxel_size=(40, 4, 4)),
            RasterizationSettings(radius=1, fg_value=0, bg_value=1, mask=ArrayKeys.GT_LABELS),
        )

        with build(pipeline):
            request = BatchRequest()
            roi = Roi((0, 0, 0), (200, 200, 200))

            request[GraphKeys.TEST_GRAPH] = GraphSpec(roi=roi)
            request[ArrayKeys.GT_LABELS] = ArraySpec(roi=roi)
            request[ArrayKeys.RASTERIZED] = ArraySpec(roi=roi)

            batch = pipeline.request_batch(request)

            rasterized = batch.arrays[ArrayKeys.RASTERIZED].data
            self.assertEqual(rasterized[0, 0, 0], 0)
            self.assertEqual(rasterized[2, 20, 20], 1)
            self.assertEqual(rasterized[4, 49, 49], 0)
            

Error 1 : At line 224 of gunpowder/nodes/rasterize_graph.py:

>           for i, point in graph.data.items():
E           AttributeError: 'Graph' object has no attribute 'data'

This is because the Graph class now has nodes and edges as attributes and we should probably replace graph.data.items() with graph.nodes.

Error 2 : At line 253 Array(data=mask_array.data == label, spec=mask_array.spec) , an AssertionError occurs because Array() tries to cast the boolean mask to the original dtype of the mask_array:

E           AssertionError: data dtype bool does not match spec dtype <class 'numpy.uint64'>

Best,
Samia

[pattonw]

The pull request has been merged and to solve this