fixed type annotation errors and formatting

BiomedSciAI · Jun 26, 2024 · 9091794 · 9091794
1 parent 7d2ebbc
commit 9091794
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 10 deletions.
diff --git a/fuse/eval/metrics/classification/metrics_classification_common.py b/fuse/eval/metrics/classification/metrics_classification_common.py
@@ -365,9 +365,9 @@ def __init__(
 
  def mcc_wrapper(
  self,
- pred: Optional[str] = None,
- target: Optional[str] = None,
- sample_weight: Optional[str] = None,
+ pred: Union[List, np.ndarray],
+ target: Union[List, np.ndarray],
+ sample_weight: Optional[Union[List, np.ndarray, None]] = None,
  **kwargs: dict,
  ) -> float:
  """
@@ -404,9 +404,9 @@ def __init__(
 
  def balanced_acc_wrapper(
  self,
- pred: Optional[str] = None,
- target: Optional[str] = None,
- sample_weight: Optional[str] = None,
+ pred: Union[List, np.ndarray],
+ target: Union[List, np.ndarray],
+ sample_weight: Optional[Union[List, np.ndarray, None]] = None,
  **kwargs: dict,
  ) -> float:
  """

diff --git a/fuse/eval/metrics/metrics_common.py b/fuse/eval/metrics/metrics_common.py
@@ -160,10 +160,7 @@ def collect(self, batch: Dict) -> None:
  batch_to_collect = {}
 
  for name, key in self._keys_to_collect.items():
- try:
- value = batch[key]
- except:
- print(self._keys_to_collect)
+ value = batch[key]
 
  # collect distributed
  if dist.is_initialized() and self._collect_distributed: