tf probability can now be used in readthedocs

N3PDF · Sep 15, 2020 · 339efd5 · 339efd5
1 parent 9622aa3
commit 339efd5
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -4,10 +4,7 @@
 import re
 
 
-requirements = ['numpy', 'pyyaml']
-#if os.environ.get('READTHEDOCS') != 'True':
-requirements.append('tensorflow')
-requirements.append('tensorflow-probability')
+requirements = ['numpy', 'pyyaml', 'tensorflow', 'tensorflow-probability']
 
 PACKAGE = 'pdfflow'