fix test

Lightning-AI · Oct 2, 2024 · c67b447 · c67b447
1 parent 92e19e5
commit c67b447
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_docker_builder.py b/tests/test_docker_builder.py
@@ -41,7 +41,7 @@ def test_color():
 """
 
 
-EXPECTED_GPU_DOCKERFILE = """# Change CUDA and cuDNN version here
+EXPECTED_GPU_DOCKERFILE = f"""# Change CUDA and cuDNN version here
 FROM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu20.04
 
 ENV DEBIAN_FRONTEND=noninteractive
@@ -73,7 +73,7 @@ def test_color():
 COPY . /app
 
 # Install litserve and requirements
-RUN pip install --no-cache-dir litserve==0.2.2 -r requirements.txt
+RUN pip install --no-cache-dir litserve=={ls.__version__} -r requirements.txt
 EXPOSE 8000
 CMD ["python", "/app/app.py"]
 """