From 85a44c230b1d072a93a5db67e2a3ffde4037f0c6 Mon Sep 17 00:00:00 2001 From: Aiden Grossman Date: Mon, 6 Jan 2025 06:08:30 +0000 Subject: [PATCH] Misc Created using spr 1.3.4 --- gematria/model/python/inference_test.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/gematria/model/python/inference_test.py b/gematria/model/python/inference_test.py index 02651a66..40962adc 100644 --- a/gematria/model/python/inference_test.py +++ b/gematria/model/python/inference_test.py @@ -213,9 +213,7 @@ def test_predict_multi_task(self): def check_predict_deltas(self, model): """Checks the prediction of the model when predicting also deltas.""" input_protos = copy.deepcopy(self.block_protos) - output_protos = tuple( - inference.predict_for_protos(model, input_protos) - ) + output_protos = tuple(inference.predict_for_protos(model, input_protos)) self.assertLen(output_protos, len(self.block_protos)) for index, (in_proto, out_proto) in enumerate(