Export KNOWN_GPUS (#35)

dstackai · Jan 30, 2024 · c74d4f3 · c74d4f3
1 parent b2fd6ab
commit c74d4f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gpuhunt/__init__.py b/src/gpuhunt/__init__.py
@@ -1,4 +1,4 @@
 from gpuhunt._internal.catalog import Catalog
-from gpuhunt._internal.constraints import matches
+from gpuhunt._internal.constraints import KNOWN_GPUS, GPUInfo, matches
 from gpuhunt._internal.default import default_catalog, query
 from gpuhunt._internal.models import CatalogItem, QueryFilter, RawCatalogItem