diff --git a/poetry.lock b/poetry.lock
index 9699f2954..acc1a7279 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:08b3f95fd602e7a475fd9686d4a05f83afbed4dbb1c06ecb8a0093715b7bad00
-size 560596
+oid sha256:8ef173727f702862b44ac62223a85676a64bb6c97fafea1ee2e59384d881c516
+size 559730
diff --git a/pyproject.toml b/pyproject.toml
index cbb5a983f..30aa1c174 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -126,6 +126,7 @@ configargparse = "^1.5.3"
 cupy-cuda113 = "^10.6.0"
 datadog = "^0.44.0"
 dotmap = "^1.3.30"
+einops = "^0.5.0"
 elasticsearch = "^8.3.1"
 Flask-Cors = "^3.0.10"
 Flask-SocketIO = "^5.2.0"