def setUp(self): self._domain = data.make_protein_domain(length=12) self._model = _get_model(self._domain) self._embed_fn = embed.get_embed_fn(model=self._model, domain=self._domain, reduce_fn=embed.masked_reduce_fn) super().setUp()
def test_encode_string_sequences(self, domain_length, length, expected_output_length): seqs = ['ABCD', 'EFG'] domain = data.make_protein_domain( length=domain_length) if domain_length else None output_batch = embed._encode_string_sequences( seqs, domain=domain, length=length) self.assertEqual(output_batch.shape, (2, expected_output_length))
def setUp(self): self._domain = data.make_protein_domain(length=12) self._model = _get_model(self._domain) super().setUp()