Fix import snafu

(Relative imports work differently depending whether you are already in a package or not... I should know this)
swagnercarena · Sep 16, 2022 · 4a35df6 · 4a35df6
1 parent 81fb4af
commit 4a35df6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/paltas/generate.py b/paltas/generate.py
@@ -21,8 +21,7 @@
 from tqdm import tqdm
 import pandas as pd
 from paltas.Configs.config_handler import ConfigHandler
-
-from .Utils.cli_maker import make_cli
+from paltas.Utils.cli_maker import make_cli
 
 
 def generate_from_config(config_path, save_folder, n=1, save_png_too=False, tf_record=False):

diff --git a/paltas/robustness_test.py b/paltas/robustness_test.py
@@ -7,7 +7,8 @@
 import string
 import os
 
-from .Utils.cli_maker import make_cli
+import numpy as np
+from paltas.Utils.cli_maker import make_cli
 
 config_header = """\
 import sys
@@ -62,9 +63,7 @@ def robustness_test(
         cleanup_results: delete all created files after a successful run.
             Will only delete robustness_test_xxx folder if they are empty.
     """
-    # Delayed imports, to make sure --help calls finish quickly
-    import numpy as np
-    import paltas
+    # Delayed import to speed up --help call
     import paltas.Analysis
 
     # Use default validation set config if no config given