From b14207cd4342821732ee95ba08c6e87c07d96f45 Mon Sep 17 00:00:00 2001
From: Travis Addair <tgaddair@gmail.com>
Date: Tue, 9 Jan 2024 10:47:33 -0800
Subject: [PATCH] fix: Only install stanford-stk on linux (#169)

---
 server/poetry.lock      | 2 +-
 server/pyproject.toml   | 2 +-
 server/requirements.txt | 4 ++--
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/server/poetry.lock b/server/poetry.lock
index dd1c1af93..ef051c4f4 100644
--- a/server/poetry.lock
+++ b/server/poetry.lock
@@ -2964,4 +2964,4 @@ torch = ["torch"]
 [metadata]
 lock-version = "2.0"
 python-versions = "^3.9"
-content-hash = "9ebdc761192d071363f54433cc7ec0a82dd05daef2ead21058b6aa3592c506df"
+content-hash = "1f96ce0474945371a5794dff060dde25f6350b4ad85923aa641be3a826d89566"
diff --git a/server/pyproject.toml b/server/pyproject.toml
index 4fb0e1c91..4e75acdf5 100644
--- a/server/pyproject.toml
+++ b/server/pyproject.toml
@@ -36,7 +36,7 @@ torch = {version = "2.1.1", optional = true }
 peft = {version = "0.4.0", optional = true }
 boto3 = "^1.28.34"
 urllib3 = "<=1.26.18"
-stanford-stk = "^0.0.6"
+stanford-stk = { version = "^0.0.6", markers = "sys_platform == 'linux'" }
 
 [tool.poetry.extras]
 torch = ["torch"]
diff --git a/server/requirements.txt b/server/requirements.txt
index da0fdd3e1..b7f32ba13 100644
--- a/server/requirements.txt
+++ b/server/requirements.txt
@@ -43,12 +43,12 @@ safetensors==0.3.1 ; python_version >= "3.9" and python_version < "4.0"
 sentencepiece==0.1.99 ; python_version >= "3.9" and python_version < "4.0"
 setuptools==69.0.2 ; python_version >= "3.9" and python_version < "4.0"
 six==1.16.0 ; python_version >= "3.9" and python_version < "4.0"
-stanford-stk==0.0.6 ; python_version >= "3.9" and python_version < "4.0"
+stanford-stk==0.0.6 ; python_version >= "3.9" and python_version < "4.0" and sys_platform == "linux"
 tiktoken==0.5.2 ; python_version >= "3.9" and python_version < "4.0"
 tokenizers==0.15.0 ; python_version >= "3.9" and python_version < "4.0"
 tqdm==4.66.1 ; python_version >= "3.9" and python_version < "4.0"
 transformers==4.36.0 ; python_version >= "3.9" and python_version < "4.0"
-triton==2.1.0 ; python_version >= "3.9" and python_version < "4.0"
+triton==2.1.0 ; python_version >= "3.9" and python_version < "4.0" and sys_platform == "linux"
 typer==0.6.1 ; python_version >= "3.9" and python_version < "4.0"
 typing-extensions==4.9.0 ; python_version >= "3.9" and python_version < "4.0"
 urllib3==1.26.18 ; python_version >= "3.9" and python_version < "4.0"