From a42c4eb2bb37be3e988bb88936e9759223cd3269 Mon Sep 17 00:00:00 2001
From: Geoffrey Irving <irving@naml.us>
Date: Tue, 18 Sep 2018 11:35:43 -0700
Subject: [PATCH] Fix setup.py for tensorflow -rc versions

---
 setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index b89f777..5ec1fce 100644
--- a/setup.py
+++ b/setup.py
@@ -1,3 +1,4 @@
+import re
 from setuptools import setup, find_packages
 import sys
 
@@ -49,6 +50,6 @@ setup(name='baselines',
 try:
     from distutils.version import StrictVersion
     import tensorflow
-    assert StrictVersion(tensorflow.__version__) >= StrictVersion('1.4.0')
+    assert StrictVersion(re.sub(r'-rc\d+$', '', tensorflow.__version__)) >= StrictVersion('1.4.0')
 except ImportError:
     assert False, "TensorFlow needed, of version above 1.4"