Don't want to write this by hand? GodCraft sets minecraft:lava_movement for you with a switch.Get GodCraftOverview
minecraft:lava_movement is an entity component: it goes in the components object of a custom mob, or in one of its component groups. Sets how fast the entity moves while in lava, like a strider.
Example
Inside the components object of BP/entities/<name>.json (values are the defaults or placeholders):
BP/entities/<name>.json
{
"minecraft:lava_movement": {}
}