Skip to content

Commit 9065cd3

Browse files
authored
fix: julia docs test (#1981)
Signed-off-by: Keming <kemingyang@tensorchord.ai>
1 parent 63a0971 commit 9065cd3

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

e2e/docs/testdata/julia_mnist/build.envd

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,3 +6,4 @@ def build():
66
install.julia()
77
install.julia_packages(name=["Flux", "MLDatasets"])
88
runtime.command(commands={"julia-mnist": "julia mlp_mnist.jl"})
9+
runtime.environ(env={"DATADEPS_ALWAYS_ACCEPT": "true"})

e2e/docs/testdata/julia_mnist/mlp_mnist.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -68,9 +68,9 @@ train_loader = simple_loader(train_data, batchsize = 256)
6868

6969
opt_state = Flux.setup(Adam(3e-4), model);
7070

71-
# Then train for 10 epochs, printing out details as we go:
71+
# Then train for 3 epochs, printing out details as we go:
7272

73-
for epoch in 1:10
73+
for epoch in 1:3
7474
loss = 0.0
7575
for (x, y) in train_loader
7676
# Compute the loss and the gradients:

0 commit comments

Comments
 (0)