From 2fead53a49f364eadda6db8592b0fe18ff8dc3f8 Mon Sep 17 00:00:00 2001 From: xvdp Date: Tue, 15 Mar 2022 14:20:06 -0700 Subject: [PATCH 1/2] fixed setup so that can be called from anywhere --- .gitignore | 2 ++ setup.py | 11 ++++++++--- 2 files changed, 10 insertions(+), 3 deletions(-) diff --git a/.gitignore b/.gitignore index 2d97991d7e..f098a90aab 100644 --- a/.gitignore +++ b/.gitignore @@ -1,3 +1,5 @@ .DS_Store __pycache__/ +build/ +*.egg-info/ diff --git a/setup.py b/setup.py index 461626e1e9..1c95fda53e 100644 --- a/setup.py +++ b/setup.py @@ -1,7 +1,12 @@ -from setuptools import setup +""" @xvdp fixed setup for simple import as: +>>> import improved_diffusion +""" +from setuptools import setup, find_packages setup( name="improved-diffusion", - py_modules=["improved_diffusion"], - install_requires=["blobfile>=1.0.5", "torch", "tqdm"], + packages = find_packages(), + install_requires=["blobfile>=1.0.5", "torch", "tqdm", "mpi4py"], + license=('LICENSE'), + version='0.0.1' ) From 4a40855c75a9331e38c988326b9d274b84bc75d6 Mon Sep 17 00:00:00 2001 From: xvdp Date: Tue, 15 Mar 2022 14:24:55 -0700 Subject: [PATCH 2/2] remove fix comments --- setup.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/setup.py b/setup.py index 1c95fda53e..8e46902553 100644 --- a/setup.py +++ b/setup.py @@ -1,6 +1,3 @@ -""" @xvdp fixed setup for simple import as: ->>> import improved_diffusion -""" from setuptools import setup, find_packages setup(