Как установить переменную среды TF_Keras = 1 для преобразования onnx?

Недавно обновился до tensorflow 2.0, и у меня возникли проблемы с переносом моих моделей .h5 в .onnx. Раньше это была очень простая процедура, но теперь у меня проблема. Когда я запускаю следующий код:

# onnx testing
import onnx    
import keras2onnx
import os
import tensorflow as tf
from tensorflow.keras.models import load_model

folder = r'\\rdnas'
os.chdir(folder)

#os.environ["TF_KERAS"]='1'
model_loc = folder+'\\model.h5'
model = tf.keras.models.load_model(model_loc)
model.summary()  

# Onnx covnersion  

onnx_model = keras2onnx.convert_keras(model)
temp_model_file = 'model.onnx'
onnx.save_model(onnx_model, temp_model_file)

Когда я запускаю код, я получаю следующую ошибку

Using TensorFlow backend.
Can't import tf2onnx module, so the conversion on a model with any custom/lambda layer will fail!
Model: "sequential_1"
_________________________________________________________________
Layer (type)                 Output Shape              Param #   
=================================================================
DVA_input (Dense)            (None, 512)               670720    
_________________________________________________________________
dropout_3 (Dropout)          (None, 512)               0         
_________________________________________________________________
dense_3 (Dense)              (None, 256)               131328    
_________________________________________________________________
dropout_4 (Dropout)          (None, 256)               0         
_________________________________________________________________
dense_4 (Dense)              (None, 128)               32896     
_________________________________________________________________
dropout_5 (Dropout)          (None, 128)               0         
_________________________________________________________________
dense_5 (Dense)              (None, 10)                1290      
_________________________________________________________________
Predicted_Volume (Dense)     (None, 1)                 11        
=================================================================
Total params: 836,245
Trainable params: 836,245
Non-trainable params: 0
_________________________________________________________________
Traceback (most recent call last):

  File "<ipython-input-1-f9d072fc6a73>", line 19, in <module>
    onnx_model = keras2onnx.convert_keras(model)

  File "C:\Users\JTBLONIGAN\AppData\Local\Continuum\anaconda3\envs\Keras-GPU\lib\site-packages\keras2onnx\main.py", line 67, in convert_keras
    " Please set environment variable TF_KERAS = 1.")

Exception: This is a tensorflow keras model, but keras standalone converter is used. Please set environment variable TF_KERAS = 1.

Я пробовал использовать закомментированную строку

os.environ["TF_KERAS"]='1' but nothing seemed to happen.

У кого-нибудь была эта проблема или кто-нибудь знает, как изменить эту переменную для преобразования модели? Ценю любую помощь.

keras2onnx==1.6.0
Keras==2.2.4
onnx==1.6.0
tensorflow==2.0.0