From bb2d2baebe3adba275921af9926f00e7fe67748a Mon Sep 17 00:00:00 2001
From: Marcel Rieger <marcel.rieger@rwth-aachen.de>
Date: Fri, 22 Feb 2019 18:58:17 +0100
Subject: [PATCH] Fix installation to preserve one-file project.

---
 lbn.py   | 6 ++++--
 setup.py | 1 +
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/lbn.py b/lbn.py
index 8331d6a..0410c0b 100644
--- a/lbn.py
+++ b/lbn.py
@@ -16,10 +16,12 @@ __version__ = "1.0.2"
 __all__ = ["LBN", "FeatureFactoryBase", "FeatureFactory"]
 
 
+import os
 import functools
 
-import numpy as np
-import tensorflow as tf
+if os.environ.get("LBN_SETUP_RUNNING") != "1":
+    import numpy as np
+    import tensorflow as tf
 
 
 class LBN(object):
diff --git a/setup.py b/setup.py
index e7909bb..3e06738 100644
--- a/setup.py
+++ b/setup.py
@@ -4,6 +4,7 @@
 import os
 import setuptools
 
+os.environ["LBN_SETUP_RUNNING"] = "1"
 import lbn
 
 
-- 
GitLab