diff --git a/src/gpuhunt/__init__.py b/src/gpuhunt/__init__.py index d6eb374..1af82bd 100644 --- 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