diff --git a/poetry.lock b/poetry.lock
index 3eac4e1..9436e87 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -5242,13 +5242,13 @@ files = [
 
 [[package]]
 name = "pyinfra"
-version = "3.3.3"
+version = "3.3.4"
 description = ""
 optional = false
 python-versions = ">=3.10,<3.11"
 files = [
-    {file = "pyinfra-3.3.3-py3-none-any.whl", hash = "sha256:141795ac270def9c02a44c511ea7e95dd8ea5540d5665c9aa4701c4db263bbd6"},
-    {file = "pyinfra-3.3.3.tar.gz", hash = "sha256:4297fae895984a54510eb7cd0e34287b5a333579fe552101f703a775bb8f04f6"},
+    {file = "pyinfra-3.3.4-py3-none-any.whl", hash = "sha256:db3a417d619a3f26b74877d75a7665506109818782fabe297fcc362f0a60e773"},
+    {file = "pyinfra-3.3.4.tar.gz", hash = "sha256:e681123eea2fe0ef2b704f73e0db0c67cf06e349840a202acaa4e8643627a445"},
 ]
 
 [package.dependencies]
@@ -6961,4 +6961,4 @@ type = ["pytest-mypy"]
 [metadata]
 lock-version = "2.0"
 python-versions = ">=3.10,<3.11"
-content-hash = "e6ea1807d6625ce9bda49d74bd960425f53f42e650b714cc0bbb95d1d9ed1125"
+content-hash = "b5bb9e940b50638a51a9836744f876dc280e48146dfc39a410fa8c048bf4a222"
diff --git a/pyproject.toml b/pyproject.toml
index f9114c0..82499a3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -10,7 +10,7 @@ packages = [{ include = "image_prediction", from = "src" }]
 python = ">=3.10,<3.11"
 # FIXME: This should be recent pyinfra, but the recent protobuf packages are not compatible with tensorflow 2.9.0, also
 #  see RED-9948.
-pyinfra = { version = "3.3.3", source = "gitlab-research" }
+pyinfra = { version = "3.3.4", source = "gitlab-research" }
 kn-utils = { version = ">=0.3.2,<0.4.0", source = "gitlab-research" }
 dvc = "^2.34.0"
 dvc-ssh = "^2.20.0"