Swim speed applies only to vertical movement when in lava

[ChiefArug]

Minecraft Version: 1.21.1

NeoForge Version: 21.1.92

Steps to Reproduce:

1. /attribute @s neoforge:swim_speed base set 100
2. Swim in water. Note you are very fast no matter what direction you go in
3. Swim in lava. Note you are very fast only when going up.

Description of issue:
Swim speed applies only to vertical movement in lava.

Code analysis:
This appears to be because in the LivingEntity patch the fluid is checked to be water before applying bonus sideways velocity:

NeoForge/patches/net/minecraft/world/entity/LivingEntity.java.patch

Lines 521 to 529 in 5e4bfbf

	+ if (this.isInWater() || (this.isInFluidType(fluidState) && !this.moveInFluid(fluidState, p_365480_, d1))) {
	float f = this.isSprinting() ? 0.9F : this.getWaterSlowDown();
	float f1 = 0.02F;
	float f2 = (float)this.getAttributeValue(Attributes.WATER_MOVEMENT_EFFICIENCY);
	@@ -2249,6 +_,7 @@
	f = 0.96F;
	}
	+ f1 *= (float)this.getAttributeValue(net.neoforged.neoforge.common.NeoForgeMod.SWIM_SPEED);

However the jumpInFluid method that is called for vertical movement does not check fluid type before applying bonus momentum:

NeoForge/src/main/java/net/neoforged/neoforge/common/extensions/ILivingEntityExtension.java

Lines 32 to 34 in 5e4bfbf

	default void jumpInFluid(FluidType type) {
	self().setDeltaMovement(self().getDeltaMovement().add(0.0D, (double) 0.04F * self().getAttributeValue(NeoForgeMod.SWIM_SPEED), 0.0D));
	}

Suggested Solution:
In the jumpInFluid method check the fluid type is water before applying bonus momentum.