From bb2d2baebe3adba275921af9926f00e7fe67748a Mon Sep 17 00:00:00 2001 From: Marcel Rieger 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