forktown/Av1an
1
0
Fork 0
mirror of https://github.com/master-of-zen/Av1an.git synced 2024-11-24 18:18:06 +00:00
Code Issues Projects Releases Packages Wiki Activity

requirements fix

Browse source
This commit is contained in:
Zen 2021-05-13 08:56:14 +03:00
parent 67a0028819
commit ed20b91fe5
2 changed files with 14 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
requirements.txt
View file

@ -5,3 +5,4 @@ tqdm
psutil
scipy
matplotlib
setuptools_rust

15
setup.py
View file

@ -1,8 +1,19 @@
#!/usr/bin/env python3
import setuptools
from setuptools_rust import Binding, RustExtension
import sys
try:
from setuptools_rust import Binding, RustExtension
except ImportError:
import subprocess
errno = subprocess.call([sys.executable, "-m", "pip", "install", "setuptools-rust"])
if errno:
print("Please install setuptools-rust package")
raise SystemExit(errno)
else:
from setuptools_rust import Binding, RustExtension
# TODO: rewrite it in rust
REQUIRES = [
"numpy",