diff --git a/pyproject.toml b/pyproject.toml index fb693ad..68a1df9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -35,7 +35,11 @@ classifiers = [ "Typing :: Typed", ] # add your package dependencies here -dependencies = [] +dependencies = [ + "torch", + "numpy", + "einops", +] # https://peps.python.org/pep-0621/#dependencies-optional-dependencies # "extras" (e.g. for `pip install .[test]`)