Add einops

pyproject.toml

@@ -5,5 +5,6 @@ description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.11"
 dependencies = [
+    "einops>=0.8.1",
     "sentence-transformers>=3.4.1",
 ]

requirements.txt

@@ -4,6 +4,8 @@ certifi==2025.1.31
     # via requests
 charset-normalizer==3.4.1
     # via requests
+einops==0.8.1
+    # via embed-experiments (pyproject.toml)
 filelock==3.17.0
     # via
     #   huggingface-hub

uv.lock

@@ -71,16 +71,29 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/d1/d6/3965ed04c63042e047cb6a3e6ed1a63a35087b6a609aa3a15ed8ac56c221/colorama-0.4.6-py2.py3-none-any.whl", hash = "sha256:4f1d9991f5acc0ca119f9d443620b77f9d6b33703e51011c16baf57afb285fc6", size = 25335 },
 ]
 
+[[package]]
+name = "einops"
+version = "0.8.1"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/e5/81/df4fbe24dff8ba3934af99044188e20a98ed441ad17a274539b74e82e126/einops-0.8.1.tar.gz", hash = "sha256:de5d960a7a761225532e0f1959e5315ebeafc0cd43394732f103ca44b9837e84", size = 54805 }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/87/62/9773de14fe6c45c23649e98b83231fffd7b9892b6cf863251dc2afa73643/einops-0.8.1-py3-none-any.whl", hash = "sha256:919387eb55330f5757c6bea9165c5ff5cfe63a642682ea788a6d472576d81737", size = 64359 },
+]
+
 [[package]]
 name = "embed-experiments"
 version = "0.1.0"
 source = { virtual = "." }
 dependencies = [
+    { name = "einops" },
     { name = "sentence-transformers" },
 ]
 
 [package.metadata]
-requires-dist = [{ name = "sentence-transformers", specifier = ">=3.4.1" }]
+requires-dist = [
+    { name = "einops", specifier = ">=0.8.1" },
+    { name = "sentence-transformers", specifier = ">=3.4.1" },
+]
 
 [[package]]
 name = "filelock"