diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml index 166687b066a909878a222c779efb2aaa4513812e..db54f5f834aadb1dae8535147ee5d3659a0735ea 100644 --- a/.gitlab-ci.yml +++ b/.gitlab-ci.yml @@ -16,7 +16,7 @@ unittests_py2: stage: test tags: - docker - image: tensorflow/tensorflow:latest + image: tensorflow/tensorflow:1.13.1 script: - python -m unittest test @@ -24,6 +24,6 @@ unittests_py3: stage: test tags: - docker - image: tensorflow/tensorflow:latest-py3 + image: tensorflow/tensorflow:1.13.1-py3 script: - python -m unittest test diff --git a/README.md b/README.md index b58a8c3bea4730e6aae0d16f781954a58f6abf32..a73509d2b70376484e819cb6369024aab673a152 100644 --- a/README.md +++ b/README.md @@ -29,7 +29,9 @@ Via [pip](https://pypi.python.org/pypi/lbn): pip install lbn ``` -NumPy and TensorFlow are the only dependencies. +NumPy and TensorFlow (1.X) are the only dependencies. + +TensorFlow 2.0 is not *yet* supported. ### Testing diff --git a/requirements.txt b/requirements.txt index 0c8c3d2e7969d38f8b76ec8f79e2067fea30cdaa..6f22da80ecd93b0f4e030c25b3fdd3801747b09f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,2 @@ numpy -tensorflow +tensorflow<2