Move imports to inside of CACHE_MODELS=true condition (#1133)

allenai · Apr 25, 2018 · a869e0b · a869e0b
1 parent 53ae735
commit a869e0b
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/scripts/cache_models.py b/scripts/cache_models.py
@@ -1,14 +1,13 @@
 #!/usr/bin/env python
 
 import os
-import sys
-
-sys.path.insert(0, os.path.dirname(os.path.abspath(os.path.join(__file__, os.pardir))))
-from allennlp.commands.serve import DEFAULT_MODELS
-from allennlp.common.file_utils import cached_path
 
 value = os.environ.get('CACHE_MODELS', 'false')
 if value.lower() == "true":
+    import sys
+    sys.path.insert(0, os.path.dirname(os.path.abspath(os.path.join(__file__, os.pardir))))
+    from allennlp.commands.serve import DEFAULT_MODELS
+    from allennlp.common.file_utils import cached_path
     models = DEFAULT_MODELS.items()
     print("CACHE_MODELS is '%s'.  Downloading %i models." % (value, len(models)))
     for i, (model, url) in enumerate(models):