FIX: sklearn cs_graph

This fixes an issue with the recent sklearn version. See diff.