File tree Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Original file line number Diff line number Diff line change @@ -132,11 +132,11 @@ end
132
132
fx, f_pushforward = res
133
133
df (Δx, Δp) = f_pushforward (NamedTuple (), Δx, Δp)
134
134
135
- df_dx = df (One (), Zero ())
136
- df_dp = df (Zero (), One ())
135
+ df_dx:: Thunk = df (One (), Zero ())
136
+ df_dp:: Thunk = df (Zero (), One ())
137
137
@test fx == f (x, p) # Check we still get the normal value, right
138
- @test extern ( df_dx) isa expected_type_df_dx
139
- @test extern ( df_dp) isa expected_type_df_dp
138
+ @test df_dx ( ) isa expected_type_df_dx
139
+ @test df_dp ( ) isa expected_type_df_dp
140
140
141
141
142
142
res = rrule (f, x, p)
145
145
dself, df_dx, df_dp = f_pullback (One ())
146
146
@test fx == f (x, p) # Check we still get the normal value, right
147
147
@test dself == NO_FIELDS
148
- @test extern ( df_dx) isa expected_type_df_dx
149
- @test extern ( df_dp) isa expected_type_df_dp
148
+ @test df_dx ( ) isa expected_type_df_dx
149
+ @test df_dp ( ) isa expected_type_df_dp
150
150
end
151
151
end
You can’t perform that action at this time.
0 commit comments