Skip to content

Commit 40dfad2

Browse files
Maksym KholiavchenkoMaksym Kholiavchenko
authored andcommitted
Update install process
1 parent ff0bc46 commit 40dfad2

File tree

2 files changed

+2
-9
lines changed

2 files changed

+2
-9
lines changed

README.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,15 +8,16 @@ It demonstrates how a neural network with convolutional and fully connected laye
88
numpy
99
scipy
1010
scikit-tensor-py3
11+
absl-py
1112
flopco-pytorch
13+
tensorly-musco
1214
pytorch
1315
```
1416

1517
## Installation
1618
```
1719
pip install musco-pytorch
1820
```
19-
If something is going wrong, try ```pip install git+https://github.com/musco-ai/musco-pytorch```
2021

2122
## Quick Start
2223
```python

setup.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -2,13 +2,6 @@
22
from setuptools import setup, find_packages
33
from setuptools.command.install import install
44

5-
6-
class InstallLocalPackage(install):
7-
def run(self):
8-
install.run(self)
9-
subprocess.call("cd dependencies/tensorly && python setup.py install --force && cd ../..", shell=True)
10-
11-
125
try:
136
from pip._internal.req import parse_requirements
147
except ImportError:
@@ -30,6 +23,5 @@ def load_requirements(file_name):
3023
download_url="https://github.com/musco-ai/musco-pytorch/archive/v1.0.2.tar.gz",
3124
license="Apache-2.0",
3225
packages=find_packages(),
33-
cmdclass={"install": InstallLocalPackage},
3426
install_requires=load_requirements("requirements.txt")
3527
)

0 commit comments

Comments
 (0)