|
1 | | -# ruff: noqa: F401 |
2 | 1 | import warnings |
3 | 2 |
|
4 | | -from gpuhunt._internal.catalog import Catalog |
| 3 | +from gpuhunt._internal.catalog import Catalog as Catalog |
5 | 4 | from gpuhunt._internal.constraints import ( |
6 | | - KNOWN_ACCELERATORS, |
7 | | - KNOWN_AMD_GPUS, |
8 | | - KNOWN_INTEL_ACCELERATORS, |
9 | | - KNOWN_NVIDIA_GPUS, |
10 | | - KNOWN_TENSTORRENT_ACCELERATORS, |
11 | | - KNOWN_TPUS, |
12 | | - correct_gpu_memory_gib, |
13 | | - is_nvidia_superchip, |
14 | | - matches, |
| 5 | + KNOWN_ACCELERATORS as KNOWN_ACCELERATORS, |
| 6 | + KNOWN_AMD_GPUS as KNOWN_AMD_GPUS, |
| 7 | + KNOWN_INTEL_ACCELERATORS as KNOWN_INTEL_ACCELERATORS, |
| 8 | + KNOWN_NVIDIA_GPUS as KNOWN_NVIDIA_GPUS, |
| 9 | + KNOWN_TENSTORRENT_ACCELERATORS as KNOWN_TENSTORRENT_ACCELERATORS, |
| 10 | + KNOWN_TPUS as KNOWN_TPUS, |
| 11 | + correct_gpu_memory_gib as correct_gpu_memory_gib, |
| 12 | + find_accelerators as find_accelerators, |
| 13 | + is_nvidia_superchip as is_nvidia_superchip, |
| 14 | + matches as matches, |
| 15 | +) |
| 16 | +from gpuhunt._internal.default import ( |
| 17 | + default_catalog as default_catalog, |
| 18 | + query as query, |
15 | 19 | ) |
16 | | -from gpuhunt._internal.default import default_catalog, query |
17 | 20 | from gpuhunt._internal.models import ( |
18 | | - AcceleratorInfo, |
19 | | - AcceleratorVendor, |
20 | | - AMDGPUInfo, |
21 | | - CatalogItem, |
22 | | - CPUArchitecture, |
23 | | - IntelAcceleratorInfo, |
24 | | - NvidiaGPUInfo, |
25 | | - QueryFilter, |
26 | | - RawCatalogItem, |
27 | | - TenstorrentAcceleratorInfo, |
28 | | - TPUInfo, |
| 21 | + AcceleratorInfo as AcceleratorInfo, |
| 22 | + AcceleratorVendor as AcceleratorVendor, |
| 23 | + AMDGPUInfo as AMDGPUInfo, |
| 24 | + CatalogItem as CatalogItem, |
| 25 | + CPUArchitecture as CPUArchitecture, |
| 26 | + IntelAcceleratorInfo as IntelAcceleratorInfo, |
| 27 | + NvidiaGPUInfo as NvidiaGPUInfo, |
| 28 | + QueryFilter as QueryFilter, |
| 29 | + RawCatalogItem as RawCatalogItem, |
| 30 | + TenstorrentAcceleratorInfo as TenstorrentAcceleratorInfo, |
| 31 | + TPUInfo as TPUInfo, |
29 | 32 | ) |
30 | 33 |
|
31 | 34 | # Deprecated aliases |
|
0 commit comments