move dataloader code to its own file

src/ISOKANN.jl

@@ -33,6 +33,7 @@ include("performance.jl") # performance metric loggers
 include("benchmarks.jl") # benchmark runs, deprecated by scripts/*
 
 include("cuda.jl")       # fixes for cuda
+include("dataloader.jl")
 
 
 include("reactionpath.jl")

src/data.jl

@@ -39,15 +39,7 @@ function datasubsample(model, data, nx)
     return xs, ys
 end
 
-# this allows to use the current isokann implementation with a DataLoader
-# where (according to MLUtils practice) the ys have shape (dim x nkoop x npoints)
-# we therefore permute the last dims to adhere to the ISOKANN.jl standard
-using MLUtils
-function datasubsample(model, data::DataLoader, nx)
-    x, y = first(data)
-    y = permutedims(y, (1, 3, 2))
-    return (x, y)
-end
+
 
 
 function subsample_inds(model, xs, n)

src/dataloader.jl

@@ -0,0 +1,9 @@
+# this allows to use the current isokann implementation with a DataLoader
+# where (according to MLUtils practice) the ys have shape (dim x nkoop x npoints)
+# we therefore permute the last dims to adhere to the ISOKANN.jl standard
+using MLUtils
+function datasubsample(model, data::DataLoader, nx)
+    x, y = first(data)
+    y = permutedims(y, (1, 3, 2))
+    return (x, y)
+end