From be461a5ef6ba80ed4a62fb1ee9c866b51a9ca26a Mon Sep 17 00:00:00 2001
From: Eitan Turok <150733043+eitanturok@users.noreply.github.com>
Date: Mon, 26 Aug 2024 14:10:29 -0400
Subject: [PATCH] install from git, not pypi

---
 setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 7596bd8..714f7ae 100644
--- a/setup.py
+++ b/setup.py
@@ -64,13 +64,13 @@
     'packaging>=21.3.0,<24.2',
     'torch>=2.4.0,<2.4.1',
     'triton>=2.1.0',
-    'stanford-stk>=0.7.1',
+    'stanford-stk @ git+https://git@github.com/stanford-futuredata/stk.git@v0.7.1',
 ]
 
 extra_deps = {}
 
 extra_deps['gg'] = [
-    'grouped_gemm>=0.1.6',
+    'grouped_gemm @ git+https://git@github.com/tgale96/grouped_gemm.git@v0.1.6',
 ]
 
 extra_deps['dev'] = [