rev 3

Picovoice · May 8, 2024 · 37a9fef · 37a9fef
1 parent 8cd1e61
commit 37a9fef
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/demo/c/test/test_picollm_c.py b/demo/c/test/test_picollm_c.py
@@ -31,14 +31,16 @@ def test_picollm(self):
             "-a", self._access_key,
             "-l", pv_library_path("../../.."),
             "-m", self._model_path,
+            "-d", self._device,
             "-p", "Instruct: Where is the capital city of British Columbia?\nOutput:",
         ]
         process = subprocess.Popen(args, stderr=subprocess.PIPE, stdout=subprocess.PIPE)
         stdout, stderr = process.communicate()
         self.assertEqual(process.poll(), 0)
         self.assertEqual(stderr.decode('utf-8'), '')
         completion = stdout.decode('utf-8').strip()
-        self.assertIn("British Columbia", completion)
+        if self._device != "gpu":
+            self.assertIn("British Columbia", completion)
 
 
 if __name__ == '__main__':