[ML][Pipeline] Change CustomFilter interface and implementation