Skip to content

Commit 1562032

Browse files
authored
Expose find_accelerators (#240)
1 parent 056a7ef commit 1562032

2 files changed

Lines changed: 27 additions & 23 deletions

File tree

pyproject.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,3 +73,4 @@ ignore = [
7373

7474
[tool.ruff.lint.isort]
7575
known-first-party = ["gpuhunt"]
76+
combine-as-imports = true

src/gpuhunt/__init__.py

Lines changed: 26 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -1,31 +1,34 @@
1-
# ruff: noqa: F401
21
import warnings
32

4-
from gpuhunt._internal.catalog import Catalog
3+
from gpuhunt._internal.catalog import Catalog as Catalog
54
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,
1519
)
16-
from gpuhunt._internal.default import default_catalog, query
1720
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,
2932
)
3033

3134
# Deprecated aliases

0 commit comments

Comments
 (0)