CI: Need to conserve runner disk space

Author: cookpa

.github/workflows/test.yml

@@ -9,7 +9,7 @@ on:
 
 env:
   USE_ANTSPY_SOURCE_BUILD: false
-
+  ANTSXNET_CACHE_DIR: ${{ runner.temp }}/ANTsXNet
 jobs:
 
   ###########################################
@@ -40,7 +40,7 @@ jobs:
         uses: actions/cache/restore@v4
         with:
           key: ${{ steps.compute-key.outputs.cache-key }}
-          path: ${{ runner.temp }}/ANTsXNet
+          path: ${{ env.ANTSXNET_CACHE_DIR }}
 
       - name: Cache hit — skip download
         if: steps.restore-cache.outputs.cache-hit == 'true'
@@ -69,17 +69,14 @@ jobs:
         if: steps.restore-cache.outputs.cache-hit != 'true'
         run: |
           pip install -e .
-          python download_all_data.py --strict
-          # Move to cache save location
-          mkdir -p ${{ runner.temp }}/ANTsXNet
-          cp -r ~/.keras/ANTsXNet/* ${{ runner.temp }}/ANTsXNet/
+          python download_all_data.py --strict --cache-dir ${{ env.ANTSXNET_CACHE_DIR }}
 
       - name: Save newly downloaded ANTsXNet data to cache
         if: steps.restore-cache.outputs.cache-hit != 'true'
         uses: actions/cache/save@v4
         with:
           key: ${{ steps.compute-key.outputs.cache-key }}
-          path: ${{ runner.temp }}/ANTsXNet
+          path: ${{ env.ANTSXNET_CACHE_DIR }}
 
 
   ###########################################
@@ -125,12 +122,7 @@ jobs:
         uses: actions/cache/restore@v4
         with:
           key: ${{ needs.prepare-data.outputs.cache-key }}
-          path: ${{ runner.temp }}/ANTsXNet
-
-      - name: Move cached data into ~/.keras
-        run: |
-          mkdir -p ~/.keras/ANTsXNet
-          cp -r ${{ runner.temp }}/ANTsXNet/* ~/.keras/ANTsXNet/
+          path: ~/.keras/ANTsXNet
 
       - name: Install ANTsPyNet
         run: |

download_all_data.py

@@ -2,8 +2,11 @@
 import argparse
 import sys
 
-def download_all_data(strict=False):
+def download_all_data(strict=False, cache_dir=None):
     print("Downloading data files from get_antsxnet_data...")
+    if cache_dir is not None:
+        antspynet.set_antsxnet_cache_directory(cache_dir)
+        print(f"Using custom cache directory: {cache_dir}")
     try:
         data_keys = antspynet.get_antsxnet_data("show")
         for key in data_keys:
@@ -44,10 +47,11 @@ def download_all_data(strict=False):
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument("--strict", action="store_true", help="Exit on first failed download.")
+    parser.add_argument("--cache-dir", type=str, help="Custom cache directory for downloads.", default=None)
     args = parser.parse_args()
 
     try:
-        download_all_data(strict=args.strict)
+        download_all_data(strict=args.strict, cache_dir=args.cache_dir)
     except Exception as e:
         print(f"\nAborted due to error: {e}")
         sys.exit(1)