Create test_unit.sh (#162)

Co-authored-by: Ben Frederickson <[email protected]> Co-authored-by: Marc Romeyn <[email protected]>
NVIDIA-Merlin · Feb 2, 2022 · 5bda79e · 5bda79e
1 parent 1ebae3d
commit 5bda79e
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/ci/test_unit.sh b/ci/test_unit.sh
@@ -0,0 +1,18 @@
+#!/bin/bash
+
+# Get latest models version
+cd /models/
+git pull origin main
+
+container=$1
+
+## Tensorflow container
+if [ "$container" == "merlin-tensorflow-training" ]; then
+    make tests-tf
+# Pytorch container
+elif [ "$container" == "merlin-pytorch-training" ]; then
+    make tests-torch
+# Inference container
+elif [ "$container" == "merlin-inference" ]; then
+    make tests-tf make tests-torch
+fi