minecraft-assets/data/minecraft/worldgen/placed_feature/lake_lava_surface.json

{
  "feature": "minecraft:lake_lava",
  "placement": [
    {
      "type": "minecraft:rarity_filter",
      "chance": 200
    },
    {
      "type": "minecraft:in_square"
    },
    {
      "type": "minecraft:heightmap",
      "heightmap": "WORLD_SURFACE_WG"
    },
    {
      "type": "minecraft:biome"
    }
  ]
}