summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorerwincoumans <erwin.coumans@gmail.com>2022-03-05 13:50:07 -0800
committerGitHub <noreply@github.com>2022-03-05 13:50:07 -0800
commit103db133794e5f573359f626c581da793fd4b129 (patch)
treeaa08a3e0c2b8f48701b3f6fb45656bfa62abf7cb
parent0ec6091efb12adfa8c62431889de6b08ac2f1a92 (diff)
parent713392b78924e8b903ad11219cf63b78412fd1fe (diff)
downloadbullet3-103db133794e5f573359f626c581da793fd4b129.tar.gz
Merge pull request #4175 from cmower/cmower-fix-import-error
Fix import error in humanoid running examples.
-rw-r--r--examples/pybullet/tensorflow/humanoid_running.py6
-rw-r--r--examples/pybullet/tensorflow/humanoid_running_vr_follow.py6
2 files changed, 10 insertions, 2 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
diff --git a/examples/pybullet/tensorflow/humanoid_running_vr_follow.py b/examples/pybullet/tensorflow/humanoid_running_vr_follow.py
index 2df85cd88..98564c091 100644
--- a/examples/pybullet/tensorflow/humanoid_running_vr_follow.py
+++ b/examples/pybullet/tensorflow/humanoid_running_vr_follow.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