File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 6
6
from unittest import TestCase
7
7
from unittest .mock import MagicMock , patch
8
8
9
- from aqua .model .constants import ModelTask
10
9
import pytest
11
10
from huggingface_hub .hf_api import HfApi , ModelInfo
12
11
from huggingface_hub .utils import GatedRepoError
13
12
from notebook .base .handlers import IPythonHandler
14
13
15
14
from ads .aqua .common .errors import AquaRuntimeError
15
+ from ads .aqua .common .utils import get_hf_model_info
16
16
from ads .aqua .extension .model_handler import (
17
+ AquaHuggingFaceHandler ,
17
18
AquaModelHandler ,
18
19
AquaModelLicenseHandler ,
19
- AquaHuggingFaceHandler ,
20
20
)
21
21
from ads .aqua .model import AquaModelApp
22
+ from ads .aqua .model .constants import ModelTask
22
23
from ads .aqua .model .entities import AquaModel , AquaModelSummary , HFModelSummary
23
- from ads .aqua .common .utils import get_hf_model_info
24
24
25
25
26
26
class ModelHandlerTestCase (TestCase ):
You can’t perform that action at this time.
0 commit comments