Skip to content

Commit 112e843

Browse files
prbzrgprbzrg
committed
fix broken
1 parent d3445cd commit 112e843

File tree

1 file changed

+10
-70
lines changed

1 file changed

+10
-70
lines changed

‎test/smoke_tests.jl‎

Lines changed: 10 additions & 70 deletions
Original file line numberDiff line numberDiff line change
@@ -231,52 +231,22 @@ Test.@testset "Smoke Tests" begin
231231

232232
Test.@test !isnothing(MLJBase.fit!(mach)) broken =
233233
GROUP != "All" &&
234-
compute_mode.adback isa ADTypes.AutoEnzyme{<:Enzyme.ForwardMode} &&
235-
(
236-
omode isa ContinuousNormalizingFlows.TrainMode || (
237-
omode isa ContinuousNormalizingFlows.TestMode &&
238-
compute_mode isa ContinuousNormalizingFlows.VectorMode
239-
)
240-
)
234+
compute_mode.adback isa ADTypes.AutoEnzyme{<:Enzyme.ForwardMode}
241235
Test.@test !isnothing(MLJBase.transform(mach, (df, df2))) broken =
242236
GROUP != "All" &&
243-
compute_mode.adback isa ADTypes.AutoEnzyme{<:Enzyme.ForwardMode} &&
244-
(
245-
omode isa ContinuousNormalizingFlows.TrainMode || (
246-
omode isa ContinuousNormalizingFlows.TestMode &&
247-
compute_mode isa ContinuousNormalizingFlows.VectorMode
248-
)
249-
)
237+
compute_mode.adback isa ADTypes.AutoEnzyme{<:Enzyme.ForwardMode}
250238
Test.@test !isnothing(MLJBase.fitted_params(mach)) broken =
251239
GROUP != "All" &&
252-
compute_mode.adback isa ADTypes.AutoEnzyme{<:Enzyme.ForwardMode} &&
253-
(
254-
omode isa ContinuousNormalizingFlows.TrainMode || (
255-
omode isa ContinuousNormalizingFlows.TestMode &&
256-
compute_mode isa ContinuousNormalizingFlows.VectorMode
257-
)
258-
)
240+
compute_mode.adback isa ADTypes.AutoEnzyme{<:Enzyme.ForwardMode}
259241
Test.@test !isnothing(MLJBase.serializable(mach)) broken =
260242
GROUP != "All" &&
261-
compute_mode.adback isa ADTypes.AutoEnzyme{<:Enzyme.ForwardMode} &&
262-
(
263-
omode isa ContinuousNormalizingFlows.TrainMode || (
264-
omode isa ContinuousNormalizingFlows.TestMode &&
265-
compute_mode isa ContinuousNormalizingFlows.VectorMode
266-
)
267-
)
243+
compute_mode.adback isa ADTypes.AutoEnzyme{<:Enzyme.ForwardMode}
268244

269245
Test.@test !isnothing(
270246
ContinuousNormalizingFlows.CondICNFDist(mach, omode, r2),
271247
) broken =
272248
GROUP != "All" &&
273-
compute_mode.adback isa ADTypes.AutoEnzyme{<:Enzyme.ForwardMode} &&
274-
(
275-
omode isa ContinuousNormalizingFlows.TrainMode || (
276-
omode isa ContinuousNormalizingFlows.TestMode &&
277-
compute_mode isa ContinuousNormalizingFlows.VectorMode
278-
)
279-
)
249+
compute_mode.adback isa ADTypes.AutoEnzyme{<:Enzyme.ForwardMode}
280250
else
281251
model = ContinuousNormalizingFlows.ICNFModel(
282252
icnf;
@@ -288,50 +258,20 @@ Test.@testset "Smoke Tests" begin
288258

289259
Test.@test !isnothing(MLJBase.fit!(mach)) broken =
290260
GROUP != "All" &&
291-
compute_mode.adback isa ADTypes.AutoEnzyme{<:Enzyme.ForwardMode} &&
292-
(
293-
omode isa ContinuousNormalizingFlows.TrainMode || (
294-
omode isa ContinuousNormalizingFlows.TestMode &&
295-
compute_mode isa ContinuousNormalizingFlows.VectorMode
296-
)
297-
)
261+
compute_mode.adback isa ADTypes.AutoEnzyme{<:Enzyme.ForwardMode}
298262
Test.@test !isnothing(MLJBase.transform(mach, df)) broken =
299263
GROUP != "All" &&
300-
compute_mode.adback isa ADTypes.AutoEnzyme{<:Enzyme.ForwardMode} &&
301-
(
302-
omode isa ContinuousNormalizingFlows.TrainMode || (
303-
omode isa ContinuousNormalizingFlows.TestMode &&
304-
compute_mode isa ContinuousNormalizingFlows.VectorMode
305-
)
306-
)
264+
compute_mode.adback isa ADTypes.AutoEnzyme{<:Enzyme.ForwardMode}
307265
Test.@test !isnothing(MLJBase.fitted_params(mach)) broken =
308266
GROUP != "All" &&
309-
compute_mode.adback isa ADTypes.AutoEnzyme{<:Enzyme.ForwardMode} &&
310-
(
311-
omode isa ContinuousNormalizingFlows.TrainMode || (
312-
omode isa ContinuousNormalizingFlows.TestMode &&
313-
compute_mode isa ContinuousNormalizingFlows.VectorMode
314-
)
315-
)
267+
compute_mode.adback isa ADTypes.AutoEnzyme{<:Enzyme.ForwardMode}
316268
Test.@test !isnothing(MLJBase.serializable(mach)) broken =
317269
GROUP != "All" &&
318-
compute_mode.adback isa ADTypes.AutoEnzyme{<:Enzyme.ForwardMode} &&
319-
(
320-
omode isa ContinuousNormalizingFlows.TrainMode || (
321-
omode isa ContinuousNormalizingFlows.TestMode &&
322-
compute_mode isa ContinuousNormalizingFlows.VectorMode
323-
)
324-
)
270+
compute_mode.adback isa ADTypes.AutoEnzyme{<:Enzyme.ForwardMode}
325271

326272
Test.@test !isnothing(ContinuousNormalizingFlows.ICNFDist(mach, omode)) broken =
327273
GROUP != "All" &&
328-
compute_mode.adback isa ADTypes.AutoEnzyme{<:Enzyme.ForwardMode} &&
329-
(
330-
omode isa ContinuousNormalizingFlows.TrainMode || (
331-
omode isa ContinuousNormalizingFlows.TestMode &&
332-
compute_mode isa ContinuousNormalizingFlows.VectorMode
333-
)
334-
)
274+
compute_mode.adback isa ADTypes.AutoEnzyme{<:Enzyme.ForwardMode}
335275
end
336276
end
337277
end

0 commit comments

Comments
 (0)