From b1985e997f6ce8067882076f6aabbd29052458de Mon Sep 17 00:00:00 2001
From: Tan Nian Wei <tannianwei@aggienetwork.com>
Date: Thu, 4 Nov 2021 15:43:36 +0800
Subject: [PATCH] make pkg available globally when installed

---
 setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 204ee2cc0..9cc707a87 100644
--- a/setup.py
+++ b/setup.py
@@ -1,7 +1,8 @@
-from setuptools import setup
+from setuptools import setup, find_packages
 
 setup(
     name="guided-diffusion",
     py_modules=["guided_diffusion"],
     install_requires=["blobfile>=1.0.5", "torch", "tqdm"],
+    packages=find_packages()
 )