Repository: systemml
Updated Branches:
  refs/heads/master 2fc26b3dc -> 77c98d693


[MINOR] Fixed import error in Keras2DML


Project: http://git-wip-us.apache.org/repos/asf/systemml/repo
Commit: http://git-wip-us.apache.org/repos/asf/systemml/commit/77c98d69
Tree: http://git-wip-us.apache.org/repos/asf/systemml/tree/77c98d69
Diff: http://git-wip-us.apache.org/repos/asf/systemml/diff/77c98d69

Branch: refs/heads/master
Commit: 77c98d693c3b2d407094de50accac615a638183f
Parents: 2fc26b3
Author: Niketan Pansare <npan...@us.ibm.com>
Authored: Tue Sep 11 13:26:23 2018 -0700
Committer: Niketan Pansare <npan...@us.ibm.com>
Committed: Tue Sep 11 13:26:23 2018 -0700

----------------------------------------------------------------------
 src/main/python/systemml/mllearn/estimators.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/systemml/blob/77c98d69/src/main/python/systemml/mllearn/estimators.py
----------------------------------------------------------------------
diff --git a/src/main/python/systemml/mllearn/estimators.py 
b/src/main/python/systemml/mllearn/estimators.py
index 0b3de41..d231b08 100644
--- a/src/main/python/systemml/mllearn/estimators.py
+++ b/src/main/python/systemml/mllearn/estimators.py
@@ -1018,7 +1018,7 @@ class Keras2DML(Caffe2DML):
         regularization_type: regularization type (default: "L2")
         """
         from .keras2caffe import convertKerasToCaffeNetwork, 
convertKerasToCaffeSolver
-        import tempfile
+        import tempfile, keras
         if isinstance(keras_model, keras.models.Sequential):
             # Convert the sequential model to functional model
             if keras_model.model is None:

Reply via email to