Skip to content

Commit e890da5

Browse files
Fix to metrics.
1 parent 4e2e75a commit e890da5

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/fmcore/framework/_algorithm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -687,7 +687,7 @@ def postprocess(batch: Predictions, **kwargs) -> Predictions:
687687
def evaluate(
688688
self,
689689
dataset: Any,
690-
metrics: Optional[Union[Union[Metric, Dict, str], List[Union[Metric, Dict, str]]]] = None,
690+
metrics: Optional[List[Union[Metric, Dict, str]]] = None,
691691
**kwargs,
692692
) -> List[Metric]:
693693
if metrics is None and "metric" in kwargs:

src/fmcore/framework/_metric.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def _pre_registration_hook(cls):
123123
@model_validator(mode="before")
124124
@classmethod
125125
def _metric_convert_params(cls, params: Dict):
126-
# print(f'Metric params: {params}')
126+
print(f'Metric params: {params}')
127127
params["params"] = super(Metric, cls)._convert_params(cls.Params, params.get("params"))
128128
params["name"] = cls.class_name
129129
if "value" in params:

0 commit comments

Comments
 (0)