Merge pull request #158 from chauhankaranraj/update-tf

Set tensorflow less than 2.5 to fix pipfile lock error with tf 2.5
AICoE · Jul 19, 2021 · c062e16 · c062e16
2 parents 675f129 + 294adee
commit c062e16
Show file tree

Hide file tree

Showing 4 changed files with 433 additions and 130 deletions.
diff --git a/Pipfile b/Pipfile
@@ -15,7 +15,7 @@ dateparser = "*"
 mlflow = "*"
 numpy = "*"
 scikit-learn = "*"
-tensorflow = "*"
+tensorflow = ">=2.0, <2.5"
 
 [requires]
 python_version = "3.8"