How do I prevent custom features (like trees and vegetation) to stop spawning over water bodies

Im working on a terrain generation and world-generation datapack (actually re-creating one of my old ones) And I am confused on how to stop custom features like trees to stop from spawning over rivers and lakes. Is there some proper predicate or tag in placed_feature or someting which I might be missing out on?