From 6d9ec5781a0836067276498a13c5f44eb64593c0 Mon Sep 17 00:00:00 2001
From: Adam Fekete <adam@fekete.co.uk>
Date: Wed, 26 Mar 2025 04:00:22 +0100
Subject: [PATCH] pull checkpoints

---
 .gitlab-ci.yml |  2 ++
 Dockerfile     | 18 +++++++++---------
 2 files changed, 11 insertions(+), 9 deletions(-)

diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 8a66630..820bd08 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -28,6 +28,8 @@ stages:
     GIT_SUBMODULE_DEPTH: 1
     GIT_SUBMODULE_UPDATE_FLAGS: --jobs 4
   before_script:
+    - git lfs pull -I mattergen/checkpoints/chemical_system --exclude=""
+    - git lfs pull -I mattergen/checkpoints/chemical_system_energy_above_hull --exclude=""
     - echo "{\"auths\":{\"${CI_REGISTRY}\":{\"auth\":\"$(printf "%s:%s" "${CI_REGISTRY_USER}" "${CI_REGISTRY_PASSWORD}" | base64 | tr -d '\n')\"},\"$CI_DEPENDENCY_PROXY_SERVER\":{\"auth\":\"$(printf "%s:%s" ${CI_DEPENDENCY_PROXY_USER} "${CI_DEPENDENCY_PROXY_PASSWORD}" | base64 | tr -d '\n')\"}}}" > /kaniko/.docker/config.json
 
 build:
diff --git a/Dockerfile b/Dockerfile
index 007200e..c1b8d8a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -4,15 +4,15 @@ FROM quay.io/jupyter/minimal-notebook:lab-4.2.5
 # Fix: https://github.com/koalaman/shellcheck/wiki/SC3014
 SHELL ["/bin/bash", "-o", "pipefail", "-c"]
 
-USER root
-
-RUN apt-get update --yes \
- && apt-get install --yes --no-install-recommends \
-    git-lfs \
- && apt-get clean \
- && rm -rf /var/lib/apt/lists/*
-
-USER ${NB_UID}
+# USER root
+#
+# RUN apt-get update --yes \
+#  && apt-get install --yes --no-install-recommends \
+#     git-lfs \
+#  && apt-get clean \
+#  && rm -rf /var/lib/apt/lists/*
+# 
+# USER ${NB_UID}
 
 # RUN mamba install --yes -c pytorch -c nvidia \
 #     pytorch==2.2.1 \
-- 
GitLab