Run tests in CI, except training

Co-authored-by: Asuka Minato <i@asukaminato.eu.org>
open-spaced-repetition · Aug 27, 2023 · f2f9647 · f2f9647
1 parent b40a4c2
commit f2f9647
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
@@ -170,6 +170,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>;