From 2d21b034593ecd1b6303080f128856afb88aaa58 Mon Sep 17 00:00:00 2001
From: Jan Zavadil <79144013+hzavadil98@users.noreply.github.com>
Date: Wed, 5 Feb 2025 10:45:36 +0100
Subject: [PATCH] Removed the test veryfying softmaxity of output

The model should output raw logits for nn.CrossEntropyLoss() #37
---
 tests/test_models.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/tests/test_models.py b/tests/test_models.py
index 15a7504..be6bd3f 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -17,6 +17,4 @@ def test_christian_model(image_shape, num_classes):
     y = model(x)
 
     assert y.shape == (n, num_classes), f"Shape: {y.shape}"
-    assert y.sum(dim=1).allclose(torch.ones(n), atol=1e-5), (
-        f"Softmax output should sum to 1, but got: {y.sum()}"
-    )
+