From b1edd804878b9b57848b8fa5581673a162c5db69 Mon Sep 17 00:00:00 2001
From: Aarni Koskela <akx@iki.fi>
Date: Tue, 16 Jan 2024 11:11:49 +0200
Subject: [PATCH] Run attention generator tests without spaCy (#709)

* CI: use CPU wheel repo for PyTorch

* Run attention generator tests without spaCy

dynamicprompts v0.26.0+ doesn't require spaCy for attention, so we can test it in CI here too
---
 .github/workflows/test.yml                | 2 ++
 tests/prompts/test_attention_generator.py | 4 ----
 2 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 8dff4c57..40602d7b 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -23,6 +23,8 @@ jobs:
         with:
           python-version: "3.10"
           cache: "pip"
+      - name: Install CPU Torch
+        run: pip3 install torch torchaudio --index-url https://download.pytorch.org/whl/cpu
       - name: Install dependencies
         run: |
           python -m pip install pytest-cov
diff --git a/tests/prompts/test_attention_generator.py b/tests/prompts/test_attention_generator.py
index a7d593f4..8b6c654e 100644
--- a/tests/prompts/test_attention_generator.py
+++ b/tests/prompts/test_attention_generator.py
@@ -1,11 +1,7 @@
-import pytest
-
 from sd_dynamic_prompts.attention_generator import SpecialSyntaxAwareAttentionGenerator
 
 
-@pytest.mark.slow
 def test_default_generator():
-    pytest.importorskip("spacy")
     generator = SpecialSyntaxAwareAttentionGenerator()
     for prompt in generator.generate(
         "purple cat singing opera, artistic, painting "