From a32f7acc9f54dba1c728cb1e596bd00bf3b4eb5f Mon Sep 17 00:00:00 2001
From: Pavol Rusnak <pavol@rusnak.io>
Date: Fri, 14 Apr 2023 15:37:11 +0200
Subject: [PATCH] py : cleanup dependencies (#962)

after #545 we do not need torch, tqdm and requests in the dependencies
---
 .devops/full.Dockerfile | 5 +++--
 flake.nix               | 1 -
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/.devops/full.Dockerfile b/.devops/full.Dockerfile
index a75bc976f..491d67676 100644
--- a/.devops/full.Dockerfile
+++ b/.devops/full.Dockerfile
@@ -5,9 +5,10 @@ FROM ubuntu:$UBUNTU_VERSION as build
 RUN apt-get update && \
     apt-get install -y build-essential python3 python3-pip
 
+COPY requirements.txt requirements.txt
+
 RUN pip install --upgrade pip setuptools wheel \
-    && pip install numpy requests sentencepiece tqdm \
-    && pip install torch --index-url https://download.pytorch.org/whl/cpu
+    && pip install -r requirements.txt
 
 WORKDIR /app
 
diff --git a/flake.nix b/flake.nix
index 91d2edd79..5363052b1 100644
--- a/flake.nix
+++ b/flake.nix
@@ -10,7 +10,6 @@
           inherit system;
         };
         llama-python = pkgs.python310.withPackages (ps: with ps; [
-          torch
           numpy
           sentencepiece
         ]);