From 5d16cc283f1fc59162bce67275e6374bc6758262 Mon Sep 17 00:00:00 2001
From: Jacky Lee <39754370+jla524@users.noreply.github.com>
Date: Sun, 25 Jun 2023 10:38:58 -0700
Subject: [PATCH] Docker fix (#1039)

* Docker test

* Remove extra installs

* Don't run full test

* No need for testing dependencies
---
 .github/workflows/test.yml |  1 -
 test/Dockerfile            | 14 +++++++++++---
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 0cd896fcb1..70571ec4ef 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -217,7 +217,6 @@ jobs:
   testdocker:
     name: Docker Test
     runs-on: ubuntu-latest
-    if: ${{ false }}
 
     steps:
     - name: Checkout Code
diff --git a/test/Dockerfile b/test/Dockerfile
index 7bf10fc6a1..22be7b4a0c 100644
--- a/test/Dockerfile
+++ b/test/Dockerfile
@@ -1,4 +1,12 @@
 FROM ubuntu:20.04
-RUN apt-get update 
-RUN apt-get install -y python3-pip git
-RUN pip3 install git+https://github.com/geohot/tinygrad.git
+
+# Install python3.8, and pip3
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    python3.8 \
+    python3-pip \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install python dependencies
+COPY . ./tinygrad
+WORKDIR tinygrad
+RUN pip install -e .