1 files changed, 12 insertions, 2 deletions
diff --git a/lib/cli.py b/lib/cli.py
index 89f0106..4fc39b6 100644
--- a/lib/cli.py
+++ b/lib/cli.py
@@ -17,8 +17,18 @@ def sanity_check(args):
         )
         sys.exit(1)
     if args.skip_param_stats and not args.force_tree:
-        print("--skip-param-stats requires --force-tree", file=sys.stderr)
-        sys.exit(1)
+        if bool(int(os.getenv("DFATOOL_FIT_FOL", "0"))):
+            print(
+                "Note: DFATOOL_FIT_FOL=1 relies on param stats to skip useless features.",
+                file=sys.stderr,
+            )
+            print(
+                "Disabling it via --skip-param-stats will likely lead to unsatisfactory results.",
+                file=sys.stderr,
+            )
+        else:
+            print("--skip-param-stats requires --force-tree", file=sys.stderr)
+            sys.exit(1)
 
 
 def print_static(model, static_model, name, attribute, with_dependence=False):