decoding_order#

Utilities for decoding order generation.

prxteinmpnn.utils.decoding_order

prxteinmpnn.utils.decoding_order.random_decoding_order(prng_key, num_residues)[source]#

Return a random decoding order.

Return type:

tuple[Int[Array, 'num_residues'], Union[Key[Array, ''], UInt32[Array, '2']]]

Parameters:
  • prng_key (Key[Array, ''] | UInt32[Array, '2'])

  • num_residues (int)

prxteinmpnn.utils.decoding_order.single_decoding_order(prng_key, num_residues)[source]#

Return a single decoding order (identity).

Return type:

tuple[Int[Array, 'num_residues'], Union[Key[Array, ''], UInt32[Array, '2']]]

Parameters:
  • prng_key (Key[Array, ''] | UInt32[Array, '2'])

  • num_residues (int)