[MRG] BUG: Support array interface

This should fix #3390.