From 21cee225c2938b6b52966f0ae887c60e45dc4de5 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Mon, 26 Sep 2022 12:25:31 +0530 Subject: [PATCH] fix loss imports --- tests/loss_function_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/loss_function_test.py b/tests/loss_function_test.py index 637d6f3..fbc982c 100644 --- a/tests/loss_function_test.py +++ b/tests/loss_function_test.py @@ -2,7 +2,7 @@ from asyncio import base_tasks import torch import pytest -from enhancer.utils.loss import mean_absolute_error, mean_squared_error +from enhancer.loss import mean_absolute_error, mean_squared_error loss_functions = [mean_absolute_error(), mean_squared_error()]