Skip to content

Commit 3615dab

Browse files
committed
Preparing the package for pypi.
1 parent 414f03d commit 3615dab

5 files changed

Lines changed: 12 additions & 2 deletions

File tree

.gitignore

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,8 @@
88
data/*
99
.idea/*
1010
checkpoint/*
11+
dist/*
12+
build/*
1113

1214
**/__pycache__/**
1315

File renamed without changes.

README.md

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,8 @@ The dataset contains 423,624 unique neural networks exhaustively generated and e
1212
# Usage
1313
You need to have PyTorch installed.
1414

15+
You can install the package by running `pip install nasbench_pytorch`. The second possibility is to install from source code:
16+
1517
1. Clone this repo
1618
```
1719
git clone https://github.com/romulus0914/NASBench-PyTorch
@@ -63,5 +65,5 @@ Example architecture (picture from the original repository)
6365
Modified from [NASBench: A Neural Architecture Search Dataset and Benchmark](https://github.com/google-research/nasbench).
6466
*graph_util.py* and *model_spec.py* are directly copied from the original repo. Original license can be found [here](https://github.com/google-research/nasbench/blob/master/LICENSE).
6567

66-
<a href="#note"></a>
68+
<a id="note"></a>
6769
**Please note that this repo is only used to train one possible architecture in the search space, not to generate all possible graphs and train them.

setup.cfg

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
[metadata]
2+
long_description = file: README.md
3+
long_description_content_type = text/markdown
4+
license_files=LICENSE.txt

setup.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22

33
setuptools.setup(
44
name='nasbench_pytorch',
5-
version='0.1',
5+
version='0.5',
6+
license='Apache License 2.0',
7+
author='Romulus Hong, Gabriela Suchopárová',
68
packages=setuptools.find_packages()
79
)

0 commit comments

Comments
 (0)