From 687e6c0288440f121fce76c54e5ce90dd2f896f8 Mon Sep 17 00:00:00 2001
From: loic-lb <loic.le-bescond@centralesupelec.fr>
Date: Thu, 22 Aug 2024 15:38:38 +0200
Subject: [PATCH] fix: Remove cuda from test_emb

---
 tests/test_emb.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_emb.py b/tests/test_emb.py
index fef9d00..88a3069 100644
--- a/tests/test_emb.py
+++ b/tests/test_emb.py
@@ -81,7 +81,7 @@ def setup_class(self) -> None:
     @pytest.mark.parametrize("coord_dir,arch_name,custom_model,pretrained_weights,transforms_dict,batch_size,num_workers,device", [
         (None,"clam", None, "IMAGENET1K_V2", {"totensor": {}, "normalize": {"mean": [0.485, 0.456, 0.406], "std": [0.229, 0.224, 0.225]}}, 1, 1, "cpu"),
         ("./coords", "phikon", None, None, None, 8, 2, "cpu"),
-        ("./coords", None, resnet18(pretrained=True), None, None, 4, 4, "cuda")])
+        ("./coords", None, resnet18(pretrained=True), None, None, 4, 4, "cpu")])
     def test_extract_model_based_embeddings(
         self,
         coord_dir,