Fix build breakage due to soft torch import (#7790)

This commit is contained in:
Eric Liang
2020-03-28 19:08:31 -07:00
committed by GitHub
parent e4bd5db4d8
commit d6255c3395
2 changed files with 17 additions and 3 deletions
+2 -2
View File
@@ -150,7 +150,7 @@ matrix:
- if [ $RAY_CI_LINUX_WHEELS_AFFECTED != "1" ]; then exit; fi
# Explicitly sleep 60 seconds for logs to go through
- ./ci/travis/test-wheels.sh || cat /tmp/ray/session_latest/logs/* || sleep 60 || false
- ./ci/travis/test-wheels.sh || cat /tmp/ray/session_latest/logs/* || (sleep 60 && false)
cache: false
# Build MacOS wheels.
@@ -168,7 +168,7 @@ matrix:
- if [ $RAY_CI_MACOS_WHEELS_AFFECTED != "1" ]; then exit; fi
# Explicitly sleep 60 seconds for logs to go through
- ./ci/travis/test-wheels.sh || cat /tmp/ray/session_latest/logs/* || sleep 60 || false
- ./ci/travis/test-wheels.sh || cat /tmp/ray/session_latest/logs/* || (sleep 60 && false)
# RLlib: Learning tests (from rllib/tuned_examples/regression_tests/*.yaml).
- os: linux
+15 -1
View File
@@ -99,6 +99,17 @@ def try_import_tfp(error=False):
return None
# Fake module for torch.nn.
class NNStub:
pass
# Fake class for torch.nn.Module to allow it to be inherited from.
class ModuleStub:
def __init__(self, *a, **kw):
raise ImportError("Could not import `torch`.")
def try_import_torch(error=False):
"""
Args:
@@ -118,7 +129,10 @@ def try_import_torch(error=False):
except ImportError as e:
if error:
raise e
return None, None
nn = NNStub()
nn.Module = ModuleStub
return None, nn
def get_variable(value, framework="tf", tf_name="unnamed-variable"):