Install ai extras in test workflow

Author: OutlyingWest

.github/workflows/test.yml

@@ -18,7 +18,7 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install -e .[test]
+        pip install -e .[test,ai]
         pip install -e ./jumper_wrapper_kernel
 
     - name: Run tests with coverage