Skip to content

GitLab

Unverified Commit 171c7aff authored by Casper's avatar Casper Committed by GitHub
Browse files

Fix importlib (#344)

parent 890127fa
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
setup.py
View file @ 171c7aff
...@@ -2,7 +2,7 @@ import os ...@@ -2,7 +2,7 @@ import os
import torch import torch
import platform import platform
import requests import requests
import importlib.util import importlib_metadata
from pathlib import Path from pathlib import Path
from setuptools import setup, find_packages from setuptools import setup, find_packages
...@@ -94,9 +94,9 @@ requirements = [ ...@@ -94,9 +94,9 @@ requirements = [
] ]
try: try:
importlib.metadata.version("autoawq-kernels") importlib_metadata.version("autoawq-kernels")
KERNELS_INSTALLED = True KERNELS_INSTALLED = True
except importlib.metadata.PackageNotFoundError: except importlib_metadata.PackageNotFoundError:
KERNELS_INSTALLED = False KERNELS_INSTALLED = False
# kernels can be downloaded from pypi for cuda+121 only # kernels can be downloaded from pypi for cuda+121 only
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment