fix loss imports

This commit is contained in:
shahules786 2022-09-26 12:25:31 +05:30
parent 23051041ce
commit 21cee225c2
1 changed files with 1 additions and 1 deletions

View File

@ -2,7 +2,7 @@ from asyncio import base_tasks
import torch import torch
import pytest 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()] loss_functions = [mean_absolute_error(), mean_squared_error()]