From 22d7a5ffc762bb5a25a9c5db5ac34b6fa0b23a1e Mon Sep 17 00:00:00 2001 From: Emmanuel Bengio Date: Wed, 8 May 2024 15:27:39 -0600 Subject: [PATCH] tox --- src/gflownet/models/seq_transformer.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/gflownet/models/seq_transformer.py b/src/gflownet/models/seq_transformer.py index 84e604d1..54557922 100644 --- a/src/gflownet/models/seq_transformer.py +++ b/src/gflownet/models/seq_transformer.py @@ -65,7 +65,6 @@ def logZ(self, cond_info: Optional[torch.Tensor]): return self._logZ(torch.ones((1, 1), device=self._logZ.weight.device)) return self._logZ(cond_info) - def forward(self, xs: SeqBatch, cond, batched=False): """Returns a GraphActionCategorical and a tensor of state predictions.