Make EdgePartition not Serializable