Skip to content

Commit

Permalink
Update docs
Browse files Browse the repository at this point in the history
Signed-off-by: shiyu22 <[email protected]>
  • Loading branch information
shiyu22 committed Sep 14, 2022
1 parent 99e6179 commit 194c769
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 11 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ Towhee compiler is a Python JIT compiler that speeds up AI-related codes by nati

### Install with pip

> Some environments are not yet supported (like m1 etc.), please try installing from source.
> If install towhee.compiler failed with pip, please install it from source.
```bash
$ pip install towhee.compiler
Expand Down
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,6 @@ nebullvm<=0.4.0
numpy
recordclass
tabulate
torch
torch>=1.12.0
torchvision
typeguard
11 changes: 2 additions & 9 deletions setup.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,4 @@
#!/usr/bin/env python
from typing import List
from pathlib import Path

from setuptools import Extension
Expand All @@ -10,13 +9,6 @@
HERE = Path(__file__).parent
README = (HERE / 'README.md').read_text()

def parse_requirements(file_name: str) -> List[str]:
with open(HERE / file_name) as f:
return [
require.strip() for require in f
if require.strip() and not require.startswith('#')
]

setup(
name="towhee.compiler",
version="0.1.0rc1",
Expand All @@ -29,7 +21,8 @@ def parse_requirements(file_name: str) -> List[str]:
license="BSD-3",
keywords="pytorch machine learning compilers",
python_requires=">=3.7, <3.11",
install_requires=parse_requirements('requirements.txt'),
install_requires=["nebullvm<=0.4.0", "numpy", "recordclass", "tabulate",
"torch>=1.12.0", "torchvision", "typeguard"],
packages=find_packages(
include=[
"towhee",
Expand Down

0 comments on commit 194c769

Please sign in to comment.