Fix types for calibration module

unitaryfund · Nov 1, 2023 · ec04601 · ec04601
1 parent ce5d849
commit ec04601
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/mitiq/calibration/calibrator.py b/mitiq/calibration/calibrator.py
@@ -68,7 +68,7 @@ def add_result(
         self.ideal[strategy.id, problem.id] = ideal_val
 
     @staticmethod
-    def _performance_str(noisy_error: float, mitigated_error: float):
+    def _performance_str(noisy_error: float, mitigated_error: float) -> str:
         """Get human readable performance representaion."""
         return (
             f"{'✔' if mitigated_error < noisy_error else '✘'}\n"
@@ -79,7 +79,7 @@ def _performance_str(noisy_error: float, mitigated_error: float):
 
     def _get_errors(
         self, strategy_id: int, problem_id: int
-    ) -> Tuple[str, float, float]:
+    ) -> Tuple[float, float]:
         """Get errors for a given strategy/problem combination.
 
         Returns:
@@ -293,7 +293,7 @@ def get_cost(self) -> Dict[str, int]:
             "ideal_executions": ideal,
         }
 
-    def run(self, log: OutputForm = None) -> None:
+    def run(self, log: Optional[OutputForm] = None) -> None:
         """Runs all the circuits required for calibration."""
         if not self.results.is_missing_data():
             self.results.reset_data()
@@ -331,7 +331,8 @@ def run(self, log: OutputForm = None) -> None:
                 self.results.log_results_cartesian()
             else:
                 raise ValueError(
-                    f"log parameter must be one of: {', '.join(OutputForm._member_names_)}"
+                    "log parameter must be one of: "
+                    f"{', '.join(OutputForm._member_names_)}"
                 )
 
     def best_strategy(self) -> Strategy: