Use temp dir for logs in test of demo

Signed-off-by: Fabrice Normandin <[email protected]>

project/algorithms/jax_image_classifier_test.py

@@ -1,3 +1,5 @@
+from pathlib import Path
+
 import flax
 import flax.linen
 import pytest
@@ -29,8 +31,8 @@ class TestJaxImageClassifier(LightningModuleTests[JaxImageClassifier]):
 
 
 @pytest.mark.slow
-def test_demo():
+def test_demo(tmp_path: Path):
     """Test the demo at the bottom of the module."""
     from .jax_image_classifier import demo
 
-    demo(devices=1, overfit_batches=0.1, max_epochs=1)
+    demo(devices=1, overfit_batches=0.1, max_epochs=1, default_log_dir=tmp_path / "logs")