summaryrefslogtreecommitdiff
path: root/examples/pybullet/tensorflow/humanoid_running.py
diff options
context:
space:
mode:
Diffstat (limited to 'examples/pybullet/tensorflow/humanoid_running.py')
-rw-r--r--examples/pybullet/tensorflow/humanoid_running.py6
1 files changed, 5 insertions, 1 deletions
diff --git a/examples/pybullet/tensorflow/humanoid_running.py b/examples/pybullet/tensorflow/humanoid_running.py
index ffd502fa3..aedbb1278 100644
--- a/examples/pybullet/tensorflow/humanoid_running.py
+++ b/examples/pybullet/tensorflow/humanoid_running.py
@@ -1,4 +1,8 @@
-import tf.compat.v1 as tf
+try:
+ import tf.compat.v1 as tf
+except Exception:
+ import tensorflow.compat.v1 as tf
+ tf.disable_eager_execution()
import sys
import numpy as np
import argparse