From 82d512f50bf0ede81be3fb281fbf8706ff6f2372 Mon Sep 17 00:00:00 2001
From: Jeroen Oudshoorn <oudshoorn.jeroen@gmail.com>
Date: Fri, 3 Nov 2023 21:44:23 +0100
Subject: [PATCH] v2.5.1

Signed-off-by: Jeroen Oudshoorn <oudshoorn.jeroen@gmail.com>
---
 builder/raspberrypi32.yml | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/builder/raspberrypi32.yml b/builder/raspberrypi32.yml
index 03b16129..ef02d389 100644
--- a/builder/raspberrypi32.yml
+++ b/builder/raspberrypi32.yml
@@ -350,8 +350,7 @@
     - name: install 32-bit pwnagotchi wheel and dependencies with 32-bit torch wheels
       pip:
         name:
-          - "{{ /usr/local/src/torch-2.1.0a0+gitunknown-cp39-cp39-linux_armv6l.whl }}"
-          - "{{ /usr/local/src/torchvision-0.16.0a0-cp39-cp39-linux_armv6l.whl }}"
+          - "{{ lookup('fileglob', '/usr/local/src/torch*.whl') }}"
           - "{{ lookup('fileglob', '/usr/local/src/pwnagotchi/dist/pwnagotchi*.whl') }}"
         extra_args: "--no-cache-dir"
       environment:
@@ -369,6 +368,11 @@
         state: absent
         path: /usr/local/src/pwnagotchi
 
+    - name: remove torch whls
+      file:
+        state: absent
+        path: "{{ lookup('fileglob', '/usr/local/src/torch*.whl') }}"
+
     ##########################################
     #
     # pwngrid, bettercap