From 7c0d9254906ce968a63d8e845faf6665b3086365 Mon Sep 17 00:00:00 2001
From: "A. Karimov" <ainureg@users.noreply.github.com>
Date: Thu, 12 Dec 2024 14:50:55 +0300
Subject: [PATCH] increase random tensor size (#243)

Co-authored-by: Ainur Karimov <ai.karimov@innopolis.ru>
---
 tests/unit/losses/providers/torch/test_segmentation.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/unit/losses/providers/torch/test_segmentation.py b/tests/unit/losses/providers/torch/test_segmentation.py
index f996a1b0..31e194e8 100644
--- a/tests/unit/losses/providers/torch/test_segmentation.py
+++ b/tests/unit/losses/providers/torch/test_segmentation.py
@@ -66,7 +66,7 @@ def test_iou_batch():
 
 def test__threshold():
 
-    x = torch.rand(2, 3)
+    x = torch.rand(10, 10)
     out = _threshold(x, threshold=0.5)
     assert torch.max(out) == 1
     assert torch.min(out) == 0