CI: fix tox tests (#29)

* CI: fix tox tests
* Lint: fix linter

.pylintrc

@@ -254,7 +254,7 @@ ignore-mixin-members=yes
 # (useful for modules/projects where namespaces are manipulated during runtime
 # and thus existing member attributes cannot be deduced by static analysis. It
 # supports qualified module names, as well as Unix pattern matching.
-ignored-modules=
+ignored-modules=torch_geometric
 
 # List of class names for which member attributes should not be checked (useful
 # for classes with dynamically set attributes). This supports the use of

blackwater/data/loaders/exp_val.py

@@ -53,7 +53,9 @@ def __init__(
                     pass
 
     def len(self):
+        """Number of entries."""
         return len(self.entries)
 
     def get(self, idx):
+        """Get specific entry."""
         return self.entries[idx]

requirements-graph.txt

@@ -0,0 +1,3 @@
+torch-geometric>=2.1.0.post1
+torch-sparse>=0.6.16
+torch-scatter>=2.1.0

requirements.txt

@@ -1,6 +1,5 @@
-torch>=1.12.1+cpu
-torch-geometric>=2.1.0.post1
-qiskit-aer==0.11.0
+torch>=1.12.1
+qiskit-aer>=0.11.0
 qiskit-ibmq-provider>=0.19.2
 qiskit-terra>=0.23.1
 qiskit-experiments>=0.4.0

setup.py

@@ -8,6 +8,10 @@
 with open("requirements.txt") as f:
     install_requires = f.read().splitlines()
 
+with open("requirements-graph.txt") as f:
+    graph_requires = f.read().splitlines()
+
+
 version_path = os.path.abspath(
     os.path.join(os.path.dirname(__file__), "blackwater", "VERSION.txt")
 )
@@ -23,6 +27,9 @@
     keywords="quantum ml qiskit error correction mitigation",
     packages=setuptools.find_packages(),
     install_requires=install_requires,
+    extra_requires={
+        "graph": graph_requires
+    },
     python_requires=">=3.7",
     version=version,
     classifiers=[

tox.ini

@@ -14,6 +14,7 @@ deps = -rrequirements.txt
        -rrequirements-dev.txt
 commands =
   pip check
+  pip install -rrequirements-graph.txt
   python -m unittest -v
 
 [testenv:lint]