[pymva] Fix again retrieval of tf/keras version

lmoneta · Aug 24, 2023 · 7c5b771 · 7c5b771
1 parent d611bfd
commit 7c5b771
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tmva/pymva/src/RModelParser_Keras.cxx b/tmva/pymva/src/RModelParser_Keras.cxx
@@ -837,6 +837,7 @@ RModel Parse(std::string filename){
    // For each layer: type,name,activation,dtype,input tensor's name,
    // output tensor's name, kernel's name, bias's name
    // None object is returned for if property doesn't belong to layer
+   PyRunString("import tensorflow",fGlobalNS,fLocalNS);
    PyRunString("import tensorflow.keras as keras",fGlobalNS,fLocalNS);
    PyRunString("from tensorflow.keras.models import load_model",fGlobalNS,fLocalNS);
    PyRunString("print('TF/Keras Version: '+ tensorflow.__version__)",fGlobalNS,fLocalNS);