Cannot have abstract method SourceSet.getAllJava()

Question

On updating android gradle plugin to 4.0.1 from 3.6.4 and gradle to 6.1.1 from 5.6.4 this piece of code breaks