From ff5c3f1f0bf483af987d4746cc53277240d20f75 Mon Sep 17 00:00:00 2001
From: Oleksandr Chychkan <chychkan@gmail.com>
Date: Sat, 31 Jul 2021 23:16:00 -0700
Subject: [PATCH] Add support for Python 3.8

---
 requirements_3.8.txt | 10 ++++++++++
 scripts/0_setup.sh   | 10 +++++++++-
 2 files changed, 19 insertions(+), 1 deletion(-)
 create mode 100644 requirements_3.8.txt

diff --git a/requirements_3.8.txt b/requirements_3.8.txt
new file mode 100644
index 0000000..3650b5c
--- /dev/null
+++ b/requirements_3.8.txt
@@ -0,0 +1,10 @@
+numpy==1.19.5
+opencv-python==4.1.2.30
+scipy==1.4.1
+tensorflow==2.5.0
+colorama==0.4.3
+tqdm==4.47.0
+ffmpeg-python==0.2.0
+Pillow==7.2.0
+scikit-image==0.17.2
+h5py==3.1.0
diff --git a/scripts/0_setup.sh b/scripts/0_setup.sh
index 4100904..6191c9a 100755
--- a/scripts/0_setup.sh
+++ b/scripts/0_setup.sh
@@ -15,6 +15,14 @@ fi
 
 source .dfl/env/bin/activate
 
-pip install -r requirements.txt
+version=$(python -V | cut -f 2 -d ' ' | cut -f 1,2 -d .)
+reqs_file='requirements.txt'
+
+if [[ ! -z "$version" && -f "requirements_$version.txt" ]]; then
+  reqs_file="requirements_$version.txt"
+fi
+
+echo "Using $reqs_file"
+pip install -r $reqs_file
 
 echo "Done."
-- 
2.52.0