diff --git a/setup.py b/setup.py index a658463..831a646 100644 --- a/setup.py +++ b/setup.py @@ -8,7 +8,7 @@ "torch>=1.3", "numpy", "scipy", - "lie_learn", + "lie_learn @ git+https://github.com/kalekundert/lie_learn@8f89f11bb0152f4ca694a72256afcedd468bade7", "joblib", "pymanopt", "autograd",