Updating capabilities dictionary, requirements.txt and black CI check

.github/workflows/build.yml

@@ -54,7 +54,10 @@ jobs:
 
       - name: Get required Python packages
         run: |
-          pip install numpy pybind11 pytest pytest-cov pytest-mock flaky
+          cd main
+          python -m pip install --upgrade pip
+          pip install git+https://github.com/PennyLaneAI/pennylane.git
+          pip install -r requirements.txt
 
       - name: Install lightning.qubit device
         run: |

.github/workflows/format.yml

@@ -6,8 +6,22 @@ jobs:
   black:
     runs-on: ubuntu-latest
     steps:
-    - uses: actions/checkout@v1
-    - name: Black Code Formatter
-      uses: lgeiger/black-action@v1.0.1
-      with:
-        args: "-l 100 pennylane_lightning/ --check"
+      - name: Cancel Previous Runs
+        uses: styfle/cancel-workflow-action@0.4.1
+        with:
+          access_token: ${{ github.token }}
+
+      - name: Set up Python
+        uses: actions/setup-python@v2
+        with:
+          python-version: 3.8
+
+      - name: Install dependencies
+        run:
+            python -m pip install --upgrade pip
+            pip install black
+
+      - uses: actions/checkout@v2
+
+      - name: Run Black
+        run: black -l 100 pennylane_lightning/ --check

.github/workflows/wheels.yml

@@ -70,6 +70,8 @@ jobs:
 
       - name: Install cibuildwheel
         run: |
+          python -m pip install --upgrade pip
+          pip install git+https://github.com/PennyLaneAI/pennylane.git
           python -m pip install cibuildwheel==1.5.5
 
       - name: Install Eigen on Windows

pennylane_lightning/lightning_qubit.py

@@ -56,7 +56,6 @@ class LightningQubit(DefaultQubit):
     pennylane_requires = ">=0.11"
     version = __version__
     author = "Xanadu Inc."
-    _capabilities = {"inverse_operations": False}  # we should look at supporting
 
     operations = {
         "BasisState",
@@ -94,9 +93,22 @@ def __init__(self, wires, *, shots=1000, analytic=True):
 
         if self.num_wires > self._MAX_WIRES:
             warnings.warn(
-                "The number of wires exceeds 16, reverting to NumPy-based evaluation.", UserWarning,
+                "The number of wires exceeds 16, reverting to NumPy-based evaluation.",
+                UserWarning,
             )
 
+    @classmethod
+    def capabilities(cls):
+        capabilities = super().capabilities().copy()
+        capabilities.update(
+            model="qubit",
+            supports_reversible_diff=False,
+            supports_inverse_operations=False,
+            supports_analytic_computation=True,
+            returns_state=True,
+        )
+        return capabilities
+
     def apply(self, operations, rotations=None, **kwargs):
 
         if self.num_wires > self._MAX_WIRES:

requirements.txt

@@ -1,5 +1,7 @@
+flaky
 numpy
 pennylane>=0.11
+pybind11
 pytest
 pytest-cov
 pytest-mock

setup.py

@@ -168,7 +168,8 @@ def build_extensions(self):
 
 requirements = [
     "numpy",
-    "pennylane>=0.11.0",
+    "pennylane @ git+https://github.com/PennyLaneAI/pennylane.git",
+    "pybind11",
 ]