Skip to content

Commit dd78209

Browse files
committed
Fixes unit tests
1 parent 32c31e8 commit dd78209

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/unitary/with_extras/aqua/test_model_handler.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,21 +6,21 @@
66
from unittest import TestCase
77
from unittest.mock import MagicMock, patch
88

9-
from aqua.model.constants import ModelTask
109
import pytest
1110
from huggingface_hub.hf_api import HfApi, ModelInfo
1211
from huggingface_hub.utils import GatedRepoError
1312
from notebook.base.handlers import IPythonHandler
1413

1514
from ads.aqua.common.errors import AquaRuntimeError
15+
from ads.aqua.common.utils import get_hf_model_info
1616
from ads.aqua.extension.model_handler import (
17+
AquaHuggingFaceHandler,
1718
AquaModelHandler,
1819
AquaModelLicenseHandler,
19-
AquaHuggingFaceHandler,
2020
)
2121
from ads.aqua.model import AquaModelApp
22+
from ads.aqua.model.constants import ModelTask
2223
from ads.aqua.model.entities import AquaModel, AquaModelSummary, HFModelSummary
23-
from ads.aqua.common.utils import get_hf_model_info
2424

2525

2626
class ModelHandlerTestCase(TestCase):

0 commit comments

Comments
 (0)