Coverage for instanovo/types.py: 100%

1from __future__ import annotations 

2 

3import numpy as np 

4import torch 

5from jaxtyping import Bool, Float, Integer 

6 

7SequenceLogProbabilities = Float[torch.Tensor, ""] 

8TokenLogProbabilities = Float[torch.Tensor, "token"] 

9ResidueLogits = Float[torch.Tensor, " residue"] 

10ResidueLogProbabilities = Float[torch.Tensor, " residue"] 

11PrecursorFeatures = Float[torch.Tensor, "3"] 

12 

13Peptide = Integer[torch.Tensor, "token"] 

14PeptideMask = Bool[torch.Tensor, "token"] 

15PeptideEmbedding = Float[torch.Tensor, "token peptide_embedding"] 

16 

17SpectrumEmbedding = Float[torch.Tensor, "peak embedding"] 

18Spectrum = Float[torch.Tensor, "peak 2"] 

19SpectrumMask = Bool[torch.Tensor, " peak"] 

20 

21KnapsackChart = Bool[np.ndarray, "mass residue"] 

22 

23Mass = Float[torch.Tensor, ""] 

24DiscretizedMass = Integer[torch.Tensor, ""] 

25MassArray = Integer[np.ndarray, "mass_item"] 

26 

27TimeStep = Integer[torch.Tensor, ""] 

28TimeEmbedding = Float[torch.Tensor, "embedding"]