Skip to content

Commit beb1b84

Browse files
fix precompilation error (#236)
1 parent 342b846 commit beb1b84

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

src/datasets/graphs/chickenpox.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ function chickenpox_datadir(dir = nothing)
1818
return dir
1919
end
2020

21-
function generate_task(data::AbstractArray, num_timesteps_in::Int, num_timesteps_out::Int)
21+
function generate_task_chickenpox(data::AbstractArray, num_timesteps_in::Int, num_timesteps_out::Int)
2222
features = []
2323
targets = []
2424
for i in 1:(size(data,3)-num_timesteps_in-num_timesteps_out)
@@ -46,7 +46,7 @@ function create_chickenpox_dataset( normalize::Bool, num_timesteps_in::Int, num_
4646
f = (f .- Statistics.mean(f, dims=(2))) ./ Statistics.std(f, dims=(2)) #Z-score normalization
4747
end
4848

49-
x, y = generate_task(f, num_timesteps_in, num_timesteps_out)
49+
x, y = generate_task_chickenpox(f, num_timesteps_in, num_timesteps_out)
5050

5151
g = Graph(; edge_index = (src, dst),
5252
node_data = (features = x, targets = y))

src/datasets/graphs/windmillenergy.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ function windmillenergy_datadir(size::String, dir = nothing)
2222
return dir
2323
end
2424

25-
function generate_task(data::AbstractArray, num_timesteps_in::Int, num_timesteps_out::Int)
25+
function generate_task_windmillenergy(data::AbstractArray, num_timesteps_in::Int, num_timesteps_out::Int)
2626
features = []
2727
targets = []
2828
for i in 1:(size(data,3)-num_timesteps_in-num_timesteps_out)
@@ -49,7 +49,7 @@ function create_windmillenergy_dataset(s::String, normalize::Bool, num_timesteps
4949
f = (f .- Statistics.mean(f, dims=(2))) ./ Statistics.std(f, dims=(2)) #Z-score normalization
5050
end
5151

52-
x, y = generate_task(f, num_timesteps_in, num_timesteps_out)
52+
x, y = generate_task_windmillenergy(f, num_timesteps_in, num_timesteps_out)
5353

5454
g = Graph(; edge_index = (src, dst),
5555
edge_data = weights,

0 commit comments

Comments
 (0)