From 3ec4ed0c98e54f1e90cdb120923756e3d6ad8036 Mon Sep 17 00:00:00 2001
From: David Ankin <daveankin@gmail.com>
Date: Tue, 1 Apr 2025 14:09:22 -0400
Subject: [PATCH] fix: fixes #797 by making device requests a list

---
 modules/ollama/testcontainers/ollama/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/ollama/testcontainers/ollama/__init__.py b/modules/ollama/testcontainers/ollama/__init__.py
index ea089f149..002b02d61 100644
--- a/modules/ollama/testcontainers/ollama/__init__.py
+++ b/modules/ollama/testcontainers/ollama/__init__.py
@@ -101,7 +101,7 @@ def __init__(
     def _check_and_add_gpu_capabilities(self):
         info = self.get_docker_client().client.info()
         if "nvidia" in info["Runtimes"]:
-            self._kwargs = {**self._kwargs, "device_requests": DeviceRequest(count=-1, capabilities=[["gpu"]])}
+            self._kwargs = {**self._kwargs, "device_requests": [DeviceRequest(count=-1, capabilities=[["gpu"]])]}
 
     def start(self) -> "OllamaContainer":
         """