fix

impICNF · Sep 19, 2024 · 5dbf3f8 · 5dbf3f8
1 parent 47ec78a
commit 5dbf3f8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/instability_tests.jl b/test/instability_tests.jl
@@ -37,5 +37,6 @@ Test.@testset "Instability" begin
  JET.test_opt(
  ContinuousNormalizingFlows.loss,
  Base.typesof(icnf, ContinuousNormalizingFlows.TrainMode(), r, ps, st);
+ target_modules = [ContinuousNormalizingFlows],
  )
 end