Linter

ml_qem/data/core.py

@@ -1,4 +1,5 @@
 """Dataclasses module."""
+
 from abc import abstractmethod
 from dataclasses import dataclass
 from typing import List, Any

ml_qem/data/dataio/__init__.py

@@ -1,2 +1,3 @@
 """Data io module."""
+
 from .dataio import ExpValDataWriter, ExpValDataReader, Reader, Writer

ml_qem/data/encoders/backend.py

@@ -7,8 +7,8 @@
 from qiskit.providers import BackendV1, BackendV2
 from torch_geometric.data import Data
 
-from blackwater.data.core import BackendEncoder
-from blackwater.data.encoders.graph_utils import backend_to_json_graph
+from ml_qem.data.core import BackendEncoder
+from ml_qem.data.encoders.graph_utils import backend_to_json_graph
 
 N_QUBIT_PROPERTIES = 2
 ALL_INSTRUCTIONS = list(get_standard_gate_name_mapping().keys())

ml_qem/data/encoders/circuit.py

@@ -8,8 +8,8 @@
 from qiskit.providers import BackendV2
 from torch_geometric.data import Data
 
-from blackwater.data.core import CircuitEncoder
-from blackwater.data.encoders.graph_utils import (
+from ml_qem.data.core import CircuitEncoder
+from ml_qem.data.encoders.graph_utils import (
     DefaultNodeEncoder,
     circuit_to_json_graph,
     BackendNodeEncoder,

ml_qem/data/encoders/graph_utils.py

@@ -106,9 +106,7 @@ def extract_properties_from_backend(
         return BackendProperties(qubit_properties_map)
 
     else:
-        raise MLQEMException(
-            f"Backend of type [{type(backend)}] is not supported yet."
-        )
+        raise MLQEMException(f"Backend of type [{type(backend)}] is not supported yet.")
 
 
 class BackendNodeEncoder(NodeEncoder):

ml_qem/data/encoders/operator.py

@@ -3,8 +3,8 @@
 import numpy as np
 from qiskit.quantum_info import Operator
 
-from blackwater.data.core import OperatorEncoder
-from blackwater.data.encoders.utils import encode_operator
+from ml_qem.data.core import OperatorEncoder
+from ml_qem.data.encoders.utils import encode_operator
 
 
 class DefaultOperatorEncoder(OperatorEncoder):

ml_qem/data/encoders/primtives_utils.py

@@ -4,9 +4,9 @@
 from qiskit import QuantumCircuit
 from qiskit.opflow import PauliSumOp
 
-from blackwater.data.core import DataEncoder
-from blackwater.data.encoders.circuit import DefaultCircuitEncoder
-from blackwater.data.encoders.operator import DefaultOperatorEncoder
+from ml_qem.data.core import DataEncoder
+from ml_qem.data.encoders.circuit import DefaultCircuitEncoder
+from ml_qem.data.encoders.operator import DefaultOperatorEncoder
 
 
 class DefaultNumpyEstimatorInputEncoder(DataEncoder):

ml_qem/data/encoders/utils.py

@@ -1,4 +1,5 @@
 """Operator encoders."""
+
 from dataclasses import dataclass
 from typing import Optional, List, Dict, Union, Any
 

ml_qem/data/generators/exp_val.py

@@ -9,7 +9,7 @@
 from qiskit.providers import BackendV1
 from torch_geometric.data import Data
 
-from blackwater.data.encoders.utils import (
+from ml_qem.data.encoders.utils import (
     get_backend_properties_v1,
     circuit_to_graph_data_json,
     generate_random_pauli_sum_op,

ml_qem/data/generators/rb.py

@@ -7,14 +7,14 @@
 from qiskit.quantum_info import Operator
 from qiskit_experiments.library import StandardRB
 
-from blackwater.data.encoders.utils import (
+from ml_qem.data.encoders.utils import (
     get_backend_properties_v1,
     circuit_to_graph_data_json,
     generate_random_pauli_sum_op,
     create_estimator_meas_data,
     encode_pauli_sum_op,
 )
-from blackwater.data.generators.exp_val import ExpValueEntry
+from ml_qem.data.generators.exp_val import ExpValueEntry
 
 
 def generate_rb_circuit(

ml_qem/data/loaders/dataclasses.py

@@ -5,7 +5,7 @@
 from torch_geometric.data import Dataset
 from torch_geometric.transforms import BaseTransform
 
-from blackwater.data.dataio import ExpValDataReader
+from ml_qem.data.dataio import ExpValDataReader
 
 # pylint: disable=abstract-method
 class ExpValDataSet(Dataset):

ml_qem/data/loaders/exp_val.py

@@ -6,7 +6,7 @@
 from torch_geometric.data import Dataset
 from torch_geometric.transforms import BaseTransform
 
-from blackwater.data.generators.exp_val import ExpValueEntry
+from ml_qem.data.generators.exp_val import ExpValueEntry
 
 
 # pylint: disable=abstract-method

ml_qem/metrics/improvement_factor.py

@@ -2,6 +2,7 @@
 
 https://arxiv.org/pdf/2210.07194.pdf
 """
+
 import typing
 from dataclasses import dataclass
 from math import sqrt

tests/data/encoders/test_graph.py

@@ -4,7 +4,7 @@
 from qiskit import QuantumCircuit, transpile
 from qiskit.providers.fake_provider import FakeLimaV2
 
-from blackwater.data.encoders.graph_utils import (
+from ml_qem.data.encoders.graph_utils import (
     circuit_to_json_graph,
     GraphData,
     BackendNodeEncoder,

tests/data/encoders/test_numpy.py

@@ -5,8 +5,8 @@
 import numpy as np
 from qiskit.circuit.random import random_circuit
 
-from blackwater.data import DefaultNumpyEstimatorInputEncoder
-from blackwater.data.encoders.utils import generate_random_pauli_sum_op
+from ml_qem.data import DefaultNumpyEstimatorInputEncoder
+from ml_qem.data.encoders.utils import generate_random_pauli_sum_op
 
 
 class TestNumpyEncoders(TestCase):

tests/data/io_tests/test_io.py

@@ -8,8 +8,8 @@
 from qiskit.providers.fake_provider import FakeLimaV2
 from qiskit.quantum_info import SparsePauliOp
 
-from blackwater.data.dataio import ExpValDataWriter, ExpValDataReader
-from blackwater.data.encoders.graph_utils import ExpValData
+from ml_qem.data.dataio import ExpValDataWriter, ExpValDataReader
+from ml_qem.data.encoders.graph_utils import ExpValData
 from tests.data.encoders.test_graph import create_bell_circuit
 
 

tests/data/loaders/test_dataclasses_loader.py

@@ -9,9 +9,9 @@
 from qiskit.quantum_info import SparsePauliOp
 from torch_geometric.data import Data
 
-from blackwater.data.dataio import ExpValDataWriter
-from blackwater.data.encoders.graph_utils import ExpValData
-from blackwater.data.loaders.dataclasses import ExpValDataSet
+from ml_qem.data.dataio import ExpValDataWriter
+from ml_qem.data.encoders.graph_utils import ExpValData
+from ml_qem.data.loaders.dataclasses import ExpValDataSet
 from tests.data.encoders.test_graph import create_bell_circuit
 
 

tests/data/loaders/test_exp_val_dataloader.py

@@ -6,8 +6,8 @@
 from qiskit.providers.fake_provider import FakeLima
 from torch_geometric.data import Data
 
-from blackwater.data.generators.exp_val import exp_value_generator
-from blackwater.data.loaders.exp_val import CircuitGraphExpValMitigationDataset
+from ml_qem.data.generators.exp_val import exp_value_generator
+from ml_qem.data.loaders.exp_val import CircuitGraphExpValMitigationDataset
 
 
 class TestExpValDataloader(TestCase):

tests/data/test_dataclasses.py

@@ -7,7 +7,7 @@
 from qiskit.quantum_info import SparsePauliOp
 from torch_geometric.data import Data
 
-from blackwater.data.encoders.graph_utils import ExpValData
+from ml_qem.data.encoders.graph_utils import ExpValData
 from tests.data.encoders.test_graph import create_bell_circuit
 
 

tests/data/test_utils.py

@@ -4,7 +4,7 @@
 from qiskit import QuantumCircuit
 from torch_geometric.data import Data
 
-from blackwater.data.encoders.utils import circuit_to_pyg_data
+from ml_qem.data.encoders.utils import circuit_to_pyg_data
 
 
 class TestDataUtils(TestCase):

tests/metrics/test_improvement_factor.py

@@ -1,7 +1,7 @@
 """Tests for improvement factor metric."""
 from unittest import TestCase
 
-from blackwater.metrics.improvement_factor import improvement_factor, Problem, Trial
+from ml_qem.metrics.improvement_factor import improvement_factor, Problem, Trial
 
 
 class TestImprovementFactor(TestCase):

tests/primitives/test_learning_estimator.py

@@ -10,10 +10,10 @@
 from sklearn.datasets import make_regression
 from sklearn.ensemble import RandomForestRegressor
 
-from blackwater.data import DefaultNumpyEstimatorInputEncoder
-from blackwater.data.encoders.graph_utils import DefaultPyGEstimatorEncoder
-from blackwater.data.encoders.utils import generate_random_pauli_sum_op
-from blackwater.primitives.learning_esimator import (
+from ml_qem.data import DefaultNumpyEstimatorInputEncoder
+from ml_qem.data.encoders.graph_utils import DefaultPyGEstimatorEncoder
+from ml_qem.data.encoders.utils import generate_random_pauli_sum_op
+from ml_qem.primitives.learning_esimator import (
     learning_estimator,
     ScikitLearnEstimatorModel,
     TorchGeometricEstimatorModel,