revert pyright fix and ignore some typing issues

cbg-ethz · Nov 6, 2023 · 3e5b062 · 3e5b062
1 parent 1c36307
commit 3e5b062
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -36,7 +36,7 @@ pytest-cov = "^4.0.0"
 pytest-xdist = "^3.2.0"
 pre-commit = "^3.1.0"
 interrogate = "^1.5.0"
-pyright = "^1.1.309, <1.1.334" # 1.1.334 calls some false-positive errors in visualize/_mcmc.
+pyright = "^1.1.309"
 mkdocs-material = "^9.1.6"
 mkdocstrings = {extras = ["python"], version = "^0.21.2"}
 mkdocs-gen-files = "^0.4.0"

diff --git a/src/pyggdrasil/visualize/_mcmc.py b/src/pyggdrasil/visualize/_mcmc.py
@@ -46,7 +46,7 @@ def _ax_log_p_iteration(
 
     ax.set_xlabel("Iteration")
     ax.set_ylabel(r"$\log(P(D|T,\theta))$")
-    ax.plot(iterations, log_probs, color="black")
+    ax.plot(iterations, log_probs, color="black")  # type: ignore
     ax.tick_params(axis="y", labelcolor="black")
 
     return ax
@@ -108,7 +108,7 @@ def _ax_dist_iteration(
     ax.set_xlabel("Iteration")
     # get name of distance measure
     ax.set_ylabel(metric_name)
-    ax.plot(iteration, distances, color="black", label=metric_name)
+    ax.plot(iteration, distances, color="black", label=metric_name)  # type: ignore
     ax.tick_params(axis="y", labelcolor="black")
     # ax.legend(loc="upper right")