def row_norms(X, squared=False): if isinstance(X, np.ndarray): return skm.row_norms(X, squared=squared) return X.map_blocks(skm.row_norms, chunks=(X.chunks[0],), drop_axis=1, squared=squared)