Set KERAS_BACKEND=theano before importing DDE

Because DDE only works with theano, this avoids any issues with having this environment variable improperly set
ReactionMechanismGenerator · Jan 18, 2019 · fb10a7b · fb10a7b
1 parent 867f57a
commit fb10a7b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/rmgpy/ml/estimator.py b/rmgpy/ml/estimator.py
@@ -31,6 +31,7 @@
 import os
 import numpy as np
 
+os.environ['KERAS_BACKEND'] = 'theano'
 from dde.predictor import Predictor
 
 from rmgpy.thermo import ThermoData