fix: some cleanup

qiboteam · Oct 22, 2024 · 123f053 · 123f053
1 parent a56a671
commit 123f053
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 311 deletions.
diff --git a/src/qiboml/models/#ansatze.py# b/src/qiboml/models/#ansatze.py#
diff --git a/src/qiboml/models/_pytorch.py b/src/qiboml/models/_pytorch.py
diff --git a/src/qiboml/models/_utils.py b/src/qiboml/models/_utils.py
diff --git a/src/qiboml/models/abstract.py b/src/qiboml/models/abstract.py
diff --git a/src/qiboml/models/decoding.py b/src/qiboml/models/decoding.py
@@ -38,6 +38,10 @@ def circuit(
     def set_backend(self, backend):
         self.backend = backend
 
+    @property
+    def output_shape(self):
+        raise_error(NotImplementedError)
+
 
 @dataclass
 class Probabilities(QuantumDecoding):
@@ -102,7 +106,7 @@ def output_shape(self):
 class Samples(QuantumDecoding):
 
     def forward(self, x: Circuit) -> ndarray:
-        return self.backend.cast(super().__call__(x).samples(), dtype=np.float64)
+        return self.backend.cast(super().__call__(x).samples(), self.backend.precision)
 
     @property
     def output_shape(self):

diff --git a/src/qiboml/models/encoding_decoding.py b/src/qiboml/models/encoding_decoding.py
diff --git a/src/qiboml/operations/differentiation.py b/src/qiboml/operations/differentiation.py
@@ -85,10 +85,10 @@ def evaluate(self, x: ndarray, encoding, training, decoding, backend, *parameter
         if binary:
             gradients = (
                 self._jax.numpy.zeros((decoding.output_shape[-1], x.shape[-1])),
-                self._jacobian_without_inputs(*parameters),
+                self._jacobian_without_inputs(*parameters),  # pylint: disable=no-member
             )
         else:
-            gradients = self._jacobian(x, *parameters)
+            gradients = self._jacobian(x, *parameters)  # pylint: disable=no-member
         decoding.set_backend(backend)
         return [
             backend.cast(self._jax.to_numpy(grad).tolist(), backend.precision)