bump and pin versions (#112)

databricks · May 21, 2024 · f1a83bd · f1a83bd
1 parent 39d3e3b
commit f1a83bd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -30,13 +30,13 @@
 
 install_requires=[
     "triton>=2.1.0",
-    "stanford-stk>=0.0.6",
+    "stanford-stk==0.7.0",
 ]
 
 extra_deps = {}
 
 extra_deps["gg"] = [
-    "grouped_gemm",
+    "grouped_gemm==0.1.1",
 ]
 
 extra_deps["dev"] = [