From be807df92be14edb3347e493da6250aa6fde407c Mon Sep 17 00:00:00 2001
From: mcencini <matteo.cencini@gmail.com>
Date: Thu, 5 Dec 2024 11:14:34 +0100
Subject: [PATCH] add download from ixi and caching dataset.

---
 .github/workflows/test-ci.yml | 14 ++++++++++++++
 examples/02-synth-data.py     | 19 +++++++++++++------
 examples/03-fitting.py        | 17 ++++++++++++-----
 3 files changed, 39 insertions(+), 11 deletions(-)

diff --git a/.github/workflows/test-ci.yml b/.github/workflows/test-ci.yml
index ed7cafd..0795392 100644
--- a/.github/workflows/test-ci.yml
+++ b/.github/workflows/test-ci.yml
@@ -147,6 +147,13 @@ jobs:
         uses: actions/setup-python@v4
         with:
           python-version: "3.10"
+          
+      - name: Restore cached ixi directory
+        uses: actions/cache/restore@v4
+        id: cache-restore
+        with:
+          path: ~/.ixi
+          key: ${{ runner.os }}-ixi
         
       - name: Install dependencies
         shell: bash -l {0}
@@ -162,6 +169,13 @@ jobs:
       - name: Display data
         run: ls -R
         working-directory: docs_build/_static
+        
+      - name: Cache ixi directory
+        uses: actions/cache/save@v4
+        if: ${{ steps.cache-restore.outputs.cache-hit != 'true' }}
+        with:
+          path: ~/.ixi
+          key: ${{ runner.os }}-ixi
       
       - name: Upload artifact
         id: artifact-upload-step
diff --git a/examples/02-synth-data.py b/examples/02-synth-data.py
index 5346ead..0049b2b 100644
--- a/examples/02-synth-data.py
+++ b/examples/02-synth-data.py
@@ -30,12 +30,19 @@
 
 import torchio as tio
 
-ixi_dataset = tio.datasets.IXI(
-    "/home/mcencini//ixi/",
-    modalities=("PD", "T1", "T2"),
-    download=False,
-)
-
+try:
+    ixi_dataset = tio.datasets.IXI(
+        "/home/mcencini/ixi/",
+        modalities=("PD", "T1", "T2"),
+        download=False,
+    )
+except:
+    ixi_dataset = tio.datasets.IXI(
+        "$HOME/.ixi/",
+        modalities=("PD", "T1", "T2"),
+        download=True,
+    )
+    
 # get subject 0
 sample_subject = ixi_dataset[0]
 
diff --git a/examples/03-fitting.py b/examples/03-fitting.py
index b30144f..3ca8557 100644
--- a/examples/03-fitting.py
+++ b/examples/03-fitting.py
@@ -30,11 +30,18 @@
 import torchsim
 
 
-ixi_dataset = tio.datasets.IXI(
-    "/home/mcencini//ixi/",
-    modalities=("PD", "T1", "T2"),
-    download=False,
-)
+try:
+    ixi_dataset = tio.datasets.IXI(
+        "/home/mcencini/ixi/",
+        modalities=("PD", "T1", "T2"),
+        download=False,
+    )
+except:
+    ixi_dataset = tio.datasets.IXI(
+        "$HOME/.ixi/",
+        modalities=("PD", "T1", "T2"),
+        download=True,
+    )
 
 # get subject 0
 sample_subject = ixi_dataset[0]