[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
GuillermoFidalgo · Sep 9, 2024 · bfbe89d · bfbe89d
1 parent 2008a54
commit bfbe89d
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 19 deletions.
diff --git a/notebooks/BB84.ipynb b/notebooks/BB84.ipynb
@@ -37,13 +37,16 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "from __future__ import annotations\n",
+    "\n",
+    "import matplotlib.pyplot as plt\n",
+    "import mplhep\n",
+    "import numpy as np\n",
     "from qcrypto.agents import Agent\n",
-    "from qcrypto.qstates import QstateUnEnt\n",
     "from qcrypto.gates import H_gate\n",
-    "import numpy as np\n",
-    "import matplotlib.pyplot as plt\n",
+    "from qcrypto.qstates import QstateUnEnt\n",
     "from scipy.optimize import curve_fit\n",
-    "import mplhep\n",
+    "\n",
     "plt.style.use(mplhep.styles.CMS)"
    ]
   },
@@ -79,7 +82,7 @@
     "        Eve.set_qstate(qstate=public_qstate, qstate_type=\"public\")\n",
     "\n",
     "    # Protocol\n",
-    "    \n",
+    "\n",
     "    Alice_base_choice = np.random.choice([True, False], size=(numqubits))\n",
     "    Alice_to_gate = np.nonzero(Alice_base_choice)[0]\n",
     "    Alice.apply_gate(qstate_type=\"public\", gate=H_gate, qubit_idx=Alice_to_gate)\n",
@@ -105,10 +108,10 @@
     "    Alice.keys[\"public\"] = Alice.keys[\"public\"][diff_base_mask]\n",
     "    Bob.keys[\"public\"] = Bob.keys[\"public\"][diff_base_mask]\n",
     "\n",
-    "    # Alice and Bob communicate check bits \n",
+    "    # Alice and Bob communicate check bits\n",
     "    Alice_checkbits = Alice.keys[\"public\"][:numcheckbits]\n",
     "    Bob_checkbits = Bob.keys[\"public\"][:numcheckbits]\n",
-    "    \n",
+    "\n",
     "    Alice.keys[\"public\"] = Alice.keys[\"public\"][numcheckbits:]\n",
     "    Bob.keys[\"public\"] = Bob.keys[\"public\"][numcheckbits:]\n",
     "\n",
@@ -129,7 +132,7 @@
    "source": [
     "numqubits = 30\n",
     "nchecks_lst = np.arange(numqubits//2)\n",
-    "probs = [] \n",
+    "probs = []\n",
     "numtrials = 1000\n",
     "\n",
     "for nchecks in nchecks_lst:\n",
@@ -138,7 +141,7 @@
     "        detected = run_trial(nchecks, numqubits, eve_present=True)\n",
     "        detections += detected\n",
     "    prob = detections / numtrials\n",
-    "    probs.append(prob)   \n",
+    "    probs.append(prob)\n",
     "with open(\"data/Eve_present.txt\",'w') as f:\n",
     "    f.writelines(str(probs).strip(\"[]\").split(','))\n"
    ]
@@ -161,8 +164,8 @@
     "param, param_cov = curve_fit(prob_func, nchecks_lst, probs)\n",
     "\n",
     "plt.errorbar(y=probs, x=nchecks_lst, yerr=[1/np.sqrt(numtrials)]*len(probs), fmt=\".\", capsize=3, label=\"Computed Probability\")\n",
-    "plt.plot(nchecks_lst, prob_func(nchecks_lst, param), label=r\"P = 1 - ({})$^n$\".format(round(param[0], 3)))\n",
-    "plt.plot(nchecks_lst, prob_func(nchecks_lst, 3/4), label=r\"P = 1 - $(3/4)^n$\".format(round(param[0], 3)))\n",
+    "plt.plot(nchecks_lst, prob_func(nchecks_lst, param), label=rf\"P = 1 - ({round(param[0], 3)})$^n$\")\n",
+    "plt.plot(nchecks_lst, prob_func(nchecks_lst, 3/4), label=r\"P = 1 - $(3/4)^n$\".format())\n",
     "\n",
     "plt.title(r\"Probability of Eve being detected (Eve)\")\n",
     "plt.xlabel(r\"Number of checkbits ($n_{\\mathrm{check}}$)\")\n",
@@ -183,7 +186,7 @@
    "source": [
     "numqubits = 30\n",
     "nchecks_lst = np.arange(numqubits//3)\n",
-    "probs = [] \n",
+    "probs = []\n",
     "numtrials = 1000\n",
     "\n",
     "for nchecks in nchecks_lst:\n",

diff --git a/notebooks/profiling.ipynb b/notebooks/profiling.ipynb
@@ -6,11 +6,12 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "from __future__ import annotations\n",
+    "\n",
+    "import numpy as np\n",
     "from qcrypto.agents import Agent\n",
-    "from qcrypto.qstates import QstateEnt, QstateUnEnt\n",
     "from qcrypto.gates import H_gate\n",
-    "import numpy as np\n",
-    "\n",
+    "from qcrypto.qstates import QstateUnEnt\n",
     "\n",
     "\n",
     "def Agent_profile():\n",
@@ -55,7 +56,7 @@
     "        Eve.set_qstate(qstate=public_qstate, qstate_type=\"public\")\n",
     "\n",
     "    # Protocol\n",
-    "    \n",
+    "\n",
     "    Alice_base_choice = np.random.choice([True, False], size=(numqubits))\n",
     "    Alice_to_gate = np.nonzero(Alice_base_choice)[0]\n",
     "    Alice.apply_gate(qstate_type=\"public\", gate=H_gate, qubit_idx=Alice_to_gate)\n",
@@ -81,10 +82,10 @@
     "    Alice.keys[\"public\"] = Alice.keys[\"public\"][diff_base_mask]\n",
     "    Bob.keys[\"public\"] = Bob.keys[\"public\"][diff_base_mask]\n",
     "\n",
-    "    # Alice and Bob communicate check bits \n",
+    "    # Alice and Bob communicate check bits\n",
     "    Alice_checkbits = Alice.keys[\"public\"][:numcheckbits]\n",
     "    Bob_checkbits = Bob.keys[\"public\"][:numcheckbits]\n",
-    "    \n",
+    "\n",
     "    Alice.keys[\"public\"] = Alice.keys[\"public\"][numcheckbits:]\n",
     "    Bob.keys[\"public\"] = Bob.keys[\"public\"][numcheckbits:]\n",
     "\n",
@@ -140,7 +141,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "%lprun -u 0.000_001 -T data/qstate_measure_all_profile.txt -f public_qstate.measure_all public_qstate.measure_all() "
+    "%lprun -u 0.000_001 -T data/qstate_measure_all_profile.txt -f public_qstate.measure_all public_qstate.measure_all()"
    ]
   },
   {

diff --git a/notebooks/qCryptoShowcase.ipynb b/notebooks/qCryptoShowcase.ipynb
@@ -14,6 +14,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "from __future__ import annotations\n",
+    "\n",
     "from qcrypto.qstates import QstateEnt, QstateUnEnt"
    ]
   },