Fix CI

Author: Acly

.github/workflows/test.yml

@@ -12,7 +12,7 @@ jobs:
       - name: Python
         uses: actions/setup-python@v5
         with:
-          python-version: '3.10'
+          python-version: '3.11'
           cache: 'pip'
       - name: Dependencies
         run: pip install -r requirements.txt          
@@ -35,9 +35,9 @@ jobs:
         with:
           submodules: true
       - name: Python
-        uses: actions/setup-python@v4
+        uses: actions/setup-python@v5
         with:
-          python-version: '3.10'
+          python-version: '3.11'
           cache: 'pip'
       - name: Dependencies
         run: pip install -r requirements.txt

ai_diffusion/ui/custom_workflow.py

@@ -14,7 +14,7 @@
 from ..custom_workflow import CustomGenerationMode
 from ..client import TextOutput
 from ..jobs import JobKind
-from ..model import Model, ApplyBehavior
+from ..model import Model
 from ..properties import Binding, Bind, bind, bind_combo
 from ..style import Styles
 from ..root import root