From d603ce921b211918ce0e3d27742032f5e7ece674 Mon Sep 17 00:00:00 2001 From: init-22 Date: Fri, 15 Nov 2024 19:11:38 +0530 Subject: [PATCH] fix: using the main repo and branch for testing --- docker/Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 88fc55243..ee9136cbf 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -59,9 +59,9 @@ RUN pip install --upgrade pip # Install Algorithmic efficiency repo RUN echo "Setting up algorithmic_efficiency repo" -ARG branch="python311" +ARG branch="main" ARG framework="both" -ARG git_url=https://github.com/init-22/algorithmic-efficiency.git +ARG git_url=https://github.com/mlcommons/algorithmic-efficiency.git RUN git clone $git_url && cd /algorithmic-efficiency RUN cd /algorithmic-efficiency && git checkout $branch