From 1f2c9d2df175c0b357bbdde8c25030df689f0a5d Mon Sep 17 00:00:00 2001
From: Karamaz0V1 <florent.guiotte@gmail.com>
Date: Fri, 17 Sep 2021 19:48:50 +0200
Subject: [PATCH] Read cache dir from environment

---
 minigrida/protocols/jurse3.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/minigrida/protocols/jurse3.py b/minigrida/protocols/jurse3.py
index ca68511..385eee4 100644
--- a/minigrida/protocols/jurse3.py
+++ b/minigrida/protocols/jurse3.py
@@ -4,6 +4,7 @@
 # version 0.0
 # date 17 sept. 2021
 
+import os
 import importlib
 
 from joblib import Memory
@@ -11,11 +12,16 @@ from pathlib import Path
 
 from . import Jurse2
 
-CACHE = './cache'
+ENV_KEY = 'MINIGRIDA_CACHE'
+DEFAULT_CACHE = './cache'
+
+CACHE = os.environ[ENV_KEY] \
+            if ENV_KEY in os.environ \
+            else DEFAULT_CACHE
 
 
 class Jurse3(Jurse2):
-    """Jurse2 protocl with cache
+    """Jurse2 protocol with cache
 
     Same as Jurse2 but enable caching results to speed up
     hyperparameters tunning.