Commit 1ce585be authored by Yuxin Wu's avatar Yuxin Wu

try to fix build

parent e1a879d7
...@@ -318,7 +318,7 @@ class ScalarPrinter(TrainingMonitor): ...@@ -318,7 +318,7 @@ class ScalarPrinter(TrainingMonitor):
def compile_regex(rs): def compile_regex(rs):
if rs is None: if rs is None:
return None return None
rs = set([r if isinstance(r, re.RegexObject) else re.compile(r) for r in rs]) rs = set([re.compile(r) for r in rs])
return rs return rs
self._whitelist = compile_regex(whitelist) self._whitelist = compile_regex(whitelist)
......
from case_script import TestPythonScript from case_script import TestPythonScript
from tensorpack.tfutils.common import get_tf_version_number
class InfoGANTest(TestPythonScript): class InfoGANTest(TestPythonScript):
...@@ -8,4 +10,6 @@ class InfoGANTest(TestPythonScript): ...@@ -8,4 +10,6 @@ class InfoGANTest(TestPythonScript):
return '../examples/GAN/InfoGAN-mnist.py' return '../examples/GAN/InfoGAN-mnist.py'
def test(self): def test(self):
if get_tf_version_number() < 1.4:
return True # requires leaky_relu
self.assertSurvive(self.script, args=None) self.assertSurvive(self.script, args=None)
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment