From 5771887089a2c4cd33381563b7bd2726b0f30fe2 Mon Sep 17 00:00:00 2001
From: Iago Bonnici <iago.bonnici@umontpellier.fr>
Date: Tue, 13 Feb 2024 11:05:13 +0100
Subject: [PATCH] Update container with pytorch (no CUDA yet).

---
 containerize.bash | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/containerize.bash b/containerize.bash
index 2544d69..dfc7309 100755
--- a/containerize.bash
+++ b/containerize.bash
@@ -26,7 +26,7 @@ bdeps base-devel rustup # Compiling suite.
 bdeps pacman-contrib    # For paccache.
 
 # Runtime dependencies.
-# deps <NONE>
+deps python-pytorch
 
 #=== Construction layers =======================================================
 read -r -d '' DEPENDENCIES_LAYER <<-EOF
@@ -47,11 +47,14 @@ read -r -d '' COMPILATION_LAYER <<-EOF
         --branch rust-implementation
     cd ${REPO}
 
-    # Test.
+    # Debug test.
     cargo test
 
-    # Compile
+    # Compile.
     cargo build --release
+
+    # Release test.
+    cargo test --release
     #---------------------------------------------------------------------------
 EOF
 
-- 
GitLab