fix: fix alignscore dependency resolution in Dockerfile (#982)

NVIDIA · Feb 8, 2025 · 1d9f278 · 1d9f278
1 parent 5ba7c57
commit 1d9f278
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/nemoguardrails/library/factchecking/align_score/Dockerfile b/nemoguardrails/library/factchecking/align_score/Dockerfile
@@ -9,9 +9,12 @@ WORKDIR /app
 
 # Clone and install the alignscore package
 RUN git clone https://github.com/yuh-zha/AlignScore.git
+
+COPY constraints.txt /app/AlignScore
+
 WORKDIR /app/AlignScore
-RUN pip install --no-cache-dir .
 
+RUN pip install --no-cache-dir -c constraints.txt .
 # Download the Spacy en_core_web_sm model
 RUN python -m spacy download en_core_web_sm
 

diff --git a/nemoguardrails/library/factchecking/align_score/constraints.txt b/nemoguardrails/library/factchecking/align_score/constraints.txt
@@ -0,0 +1,3 @@
+
+spacy==3.7.5
+# transformers>=4.20.1,<4.30 # resolve issue with alignscore pyproject.toml
diff --git a/nemoguardrails/library/factchecking/align_score/requirements.txt b/nemoguardrails/library/factchecking/align_score/requirements.txt
@@ -4,3 +4,5 @@ fastapi>=0.109.1
 starlette>=0.36.2
 typer>=0.7.0
 uvicorn>=0.23.2
+
+torch==2.0.1
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@

		spacy==3.7.5
		# transformers>=4.20.1,<4.30 # resolve issue with alignscore pyproject.toml