Changing shape of the searchlight

[AntoineBouyeure]

Hello,

I would like to change the shape of the searchlight to a cube (which is the default shape of classbrainiak.searchlight.searchlight.Searchlight that is used in the tutorials) to a sphere/ball. I can import a searchlight ball as classbrainiak.searchlight.searchlight.Ball, but I am not sure of what to do next. Specifiying shape in the form of sl = Searchlight(shape=Ball) or sl = Searchlight and then Searchlight.shape=Ball doesnt work. I can't also run directly searchlight with the brainiak.searchlight.searchlight.Ball object.

What should I do to change the shape of the searchlight? I'm sorry if the answer is kind of obvious...

[lcnature]

Hi,
Try this?

from brainiak.searchlight.searchlight import Ball, Searchlight
sl = Searchlight(shape=Ball)

[AntoineBouyeure]

Hi, thanks for your answer! Unfortunately when doing something like sl = Searchlight(sl_rad=5, shape=Ball), the resulting searchlight images are empty. And trying sl = Searchlight(shape=Ball(rad=5)) produces TypeError: 'Ball' object is not callable