Update PyTorch script

lutzroeder · Nov 28, 2024 · 964d94d · 964d94d
1 parent e32209f
commit 964d94d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tools/pytorch b/tools/pytorch
@@ -55,7 +55,9 @@ schema() {
 metadata() {
     echo "pytorch metadata"
     [[ $(grep -U $'\x0D' ./source/pytorch-metadata.json) ]] && crlf=1
+    venv
     ${python} ./tools/pytorch_script.py
+    deactivate
     if [[ -n ${crlf} ]]; then
         unix2dos --quiet --newfile ./source/pytorch-metadata.json ./source/pytorch-metadata.json
     fi