(resolves #83) Probability in tools needs parameter type

src/__init__.py

@@ -2,3 +2,4 @@
 from .quantum_gate import gates
 from .visualize import *
 from .tools import *
+from .qubit import *

src/quantum_circuit/base/base_core.py

@@ -1,5 +1,5 @@
 import numpy as np
-from ...tools import qubit
+from ...qubit import qubit
 from ..interface import CoreInterface
 
 

src/quantum_circuit/gpu/gpu_core.py

@@ -1,6 +1,6 @@
 import cupy as cp
 from ..interface import CoreInterface
-from ...tools import qubit
+from ...qubit import qubit
 
 
 class GpuCore(CoreInterface):

src/quantum_circuit/gpu_sparse/gpu_sparse_core.py

@@ -2,7 +2,7 @@
 from cupyx.scipy.sparse import csr_matrix, csc_matrix
 from cupyx.scipy.sparse import kron
 from ..interface import CoreInterface
-from ...tools import qubit
+from ...qubit import qubit
 
 
 class GpuSparseCore(CoreInterface):

src/quantum_circuit/sparse/sparse_core.py

@@ -1,6 +1,6 @@
 from scipy import sparse as sp
 from ..interface import CoreInterface
-from ...tools import qubit
+from ...qubit import qubit
 
 
 class SparseCore(CoreInterface):

src/qubit/__init__.py

@@ -0,0 +1 @@
+from .qubit import qubit

src/tools/__init__.py

@@ -1,4 +1,3 @@
-from .qubit import qubit
 from .measure import measure
 from .probability import probability
 from .phase_angle import phase_angle as phaseangle

src/tools/probability.py

@@ -1,28 +1,33 @@
-from numpy import abs, around, multiply, square, log2
+from numpy import abs, around, multiply, square, log2, ndarray
 from .base import convert_state
+from typing import Union
+from ..quantum_circuit import QuantumCircuit
 
 
 def probability(
-    quantumstate, show_percent: bool = False, show_bit: int = -1, round: int = 3
+    quantum_state: Union[ndarray, QuantumCircuit],
+    show_percent: bool = False,
+    show_bit: int = -1,
+    round: int = 3,
 ):
 
-    circuit_size = int(log2(quantumstate.size))
-    quantumstate = convert_state(quantumstate)
+    circuit_size = int(log2(quantum_state.size))
+    quantum_state = convert_state(quantum_state)
     if round < 0:
         exit(
             f"Error: QuantumCircuit.tools.probabilities() -- round placement must be a value greater than 0."
         )
 
     if isinstance(show_bit, int) and show_bit < 0:
-        probability = abs(square(quantumstate))
+        probability = abs(square(quantum_state))
 
     elif isinstance(show_bit, str) or isinstance(show_bit, int):
 
         if 2**circuit_size <= show_bit:
             exit(
                 f"Error: QuantumCircuit.tools.probabilities() -- Called bit to find phase angle is not within range of possible values."
             )
-        probability = abs(square(quantumstate[show_bit]))
+        probability = abs(square(quantum_state[show_bit]))
 
     if show_percent:
         probability = multiply(probability, 100)