Update ProteinChainsExt.jl

Author: AntonOresten

ext/ProteinChainsExt.jl

@@ -4,12 +4,15 @@ using PyBoltz
 using ProteinChains: ProteinStructure
 using ProteinChains.BioStructures: MolecularStructure
 
+to_structure(x) = ProteinStructure(x)
+to_structure(::Missing) = missing
+
 function PyBoltz.predict(input, ::Type{ProteinStructure}; options...)
-    return PyBoltz.predict(input, MolecularStructure; options...) .|> ProteinStructure
+    return PyBoltz.predict(input, MolecularStructure; options...) .|> to_structure
 end
 
 function PyBoltz.predict(input::PyBoltz.Schema.MolecularInput, ::Type{ProteinStructure}; options...)
-    return PyBoltz.predict(input, MolecularStructure; options...) |> ProteinStructure
+    return PyBoltz.predict(input, MolecularStructure; options...) |> to_structure
 end
 
 end