Add two customer scenarios case study (#230)

py-why · Mar 7, 2020 · 4eca7bd · 4eca7bd
1 parent bc044ed
commit 4eca7bd
Show file tree

Hide file tree

Showing 4 changed files with 1,641 additions and 3 deletions.
diff --git a/econml/tests/test_notebooks.py b/econml/tests/test_notebooks.py
@@ -9,9 +9,10 @@
 import traitlets
 
 _nbdir = os.path.join(os.path.dirname(__file__), '..', '..', 'notebooks')
-_notebooks = [path
-              for path in os.listdir(_nbdir)
-              if path.endswith('.ipynb')]
+_nbsubdirs = ['.', 'CustomerScenarios']  # TODO: add AutoML notebooks
+_notebooks = [
+    os.path.join(subdir, path) for subdir
+    in _nbsubdirs for path in os.listdir(os.path.join(_nbdir, subdir)) if path.endswith('.ipynb')]
 
 
 @pytest.mark.parametrize("file", _notebooks)