Run tests in CI, except training

Co-authored-by: Asuka Minato <[email protected]>
open-spaced-repetition · Aug 27, 2023 · c45aab4 · c45aab4
1 parent a955aeb
commit c45aab4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/.github/workflows/check.sh b/.github/workflows/check.sh
@@ -9,3 +9,9 @@ cargo fmt --check || (
 )
 
 cargo clippy -- -Dwarnings
+
+install -d tests/data/
+pushd tests/data/
+wget https://github.com/open-spaced-repetition/fsrs-optimizer-burn/files/12394182/collection.anki21.zip
+unzip *.zip
+SKIP_TRAINING=1 cargo test
diff --git a/src/training.rs b/src/training.rs
@@ -181,6 +181,10 @@ mod tests {
 
     #[test]
     fn training() {
+        if std::env::var("SKIP_TRAINING").is_ok() {
+            println!("Skipping test in CI");
+            return;
+        }
         use burn_ndarray::NdArrayBackend;
         use burn_ndarray::NdArrayDevice;
         type Backend = NdArrayBackend<f32>;