Backported DataMapProvider#gather overload passing the registry provider

src/main/java/net/neoforged/neoforge/common/data/DataMapProvider.java

@@ -25,7 +25,6 @@
 import net.minecraft.data.CachedOutput;
 import net.minecraft.data.DataProvider;
 import net.minecraft.data.PackOutput;
-import net.minecraft.resources.RegistryOps;
 import net.minecraft.resources.ResourceKey;
 import net.minecraft.resources.ResourceLocation;
 import net.minecraft.tags.TagKey;
@@ -60,10 +59,10 @@ protected DataMapProvider(PackOutput packOutput, CompletableFuture<HolderLookup.
 
     @Override
     public CompletableFuture<?> run(CachedOutput cache) {
-        gather();
-
         return lookupProvider.thenCompose(provider -> {
-            final DynamicOps<JsonElement> dynamicOps = RegistryOps.create(JsonOps.INSTANCE, provider);
+            gather(provider);
+
+            final DynamicOps<JsonElement> dynamicOps = provider.createSerializationContext(JsonOps.INSTANCE);
 
             return CompletableFuture.allOf(this.builders.entrySet().stream().map(entry -> {
                 DataMapType<?, ?> type = entry.getKey();
@@ -85,6 +84,13 @@ private <T, R> CompletableFuture<?> generate(Path out, CachedOutput cache, Build
      */
     protected abstract void gather();
 
+    /**
+     * Generate data map entries.
+     */
+    protected void gather(HolderLookup.Provider provider) {
+        gather();
+    }
+
     @SuppressWarnings("unchecked")
     public <T, R> Builder<T, R> builder(DataMapType<R, T> type) {
         // Avoid any weird classcastexceptions at runtime if a builder was previously created with this method