Clean cache before building

replicate · Oct 15, 2024 · 4dfe9a4 · 4dfe9a4
1 parent 48ec328
commit 4dfe9a4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/src/build.py b/src/build.py
@@ -83,6 +83,9 @@ def build_generation(args: argparse.Namespace, mg: MonoGen) -> None:
 def build(args: argparse.Namespace) -> None:
     os.makedirs(args.cache, exist_ok=True)
 
+    if args.clean_uv_cache:
+        clean_uv_cache()
+
     gens = []
     for mg in sorted(MONOGENS[args.environment], reverse=True):
         if mg.id < args.min_gen_id or mg.id > args.max_gen_id:
@@ -96,8 +99,6 @@ def build(args: argparse.Namespace) -> None:
         prune_cuda(args)
     if args.prune_uv_cache:
         prune_uv_cache()
-    if args.clean_uv_cache:
-        clean_uv_cache()
 
     logger.info(f'Calculating disk usage in {args.prefix}...')
     cmd = ['du', '-ch', '-d', '1', args.prefix]

diff --git a/src/entrypoint.sh b/src/entrypoint.sh
@@ -15,12 +15,6 @@ log() {
 }
 
 builder() {
-    if [ -n "${MONOBASE_CLEAN:-}" ]; then
-        log "Cleaning up $MONOBASE_PREFIX..."
-        rm -rf "${MONOBASE_PREFIX:?}"/{bin,cuda,monobase,uv}
-        rm -f "$DONE_FILE"
-    fi
-
     mkdir -p "$MONOBASE_PREFIX/bin"
 
     # Always install latest uv and pget first