fix

AntonOresten · AntonOresten · commit 27f8be0a62d4 · 2025-06-03T00:05:00.000+02:00
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -57,7 +57,7 @@ const accelerator = get(ENV, "PyBoltz_TEST_ACCELERATOR", "cpu")
                 ),
             ]
             predicted_structures = predict(inputs, MolecularStructure; seed=0, accelerator)
-            @test predicted_structures isa Vector{MolecularStructure}
+            @test predicted_structures isa Vector{Union{MolecularStructure,Union}}
             @test length(predicted_structures) == 4
             @testset "Order preservation" begin
                 @test countresidues.(predicted_structures) == [10, 30, 20, 40]

Original file line number	Diff line number	Diff line change
`@@ -57,7 +57,7 @@ const accelerator = get(ENV, "PyBoltz_TEST_ACCELERATOR", "cpu")`
`57`	`57`	`),`
`58`	`58`	`]`
`59`	`59`	`predicted_structures = predict(inputs, MolecularStructure; seed=0, accelerator)`
`60`		`- @test predicted_structures isa Vector{MolecularStructure}`
	`60`	`+ @test predicted_structures isa Vector{Union{MolecularStructure,Union}}`
`61`	`61`	`@test length(predicted_structures) == 4`
`62`	`62`	`@testset "Order preservation" begin`
`63`	`63`	`@test countresidues.(predicted_structures) == [10, 30, 20, 40]`