Skip to content

Commit 9497a45

Browse files
fix(ollama): make device request a list (#799)
1 parent 2f9139c commit 9497a45

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modules/ollama/testcontainers/ollama/__init__.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ def __init__(
101101
def _check_and_add_gpu_capabilities(self):
102102
info = self.get_docker_client().client.info()
103103
if "nvidia" in info["Runtimes"]:
104-
self._kwargs = {**self._kwargs, "device_requests": DeviceRequest(count=-1, capabilities=[["gpu"]])}
104+
self._kwargs = {**self._kwargs, "device_requests": [DeviceRequest(count=-1, capabilities=[["gpu"]])]}
105105

106106
def start(self) -> "OllamaContainer":
107107
"""

0 commit comments

Comments
 (0)