Rename DFATOOL_FIT_LINEAR_ONLY=1 to DFATOOL_SUBMODEL=fol

2 files changed, 2 insertions, 2 deletions

diff --git a/lib/functions.py b/lib/functions.py
index 501970e..13897bb 100644
--- a/lib/functions.py
+++ b/lib/functions.py
@@ -1842,7 +1842,7 @@ class analytic:
                 repr_str="β₀ + β₁ * safe_sqrt(x)",
             )
 
-        if bool(int(os.getenv("DFATOOL_FIT_LINEAR_ONLY", "0"))):
+        if os.getenv("DFATOOL_SUBMODEL", "uls") == "fol":
             functions = {"linear": functions["linear"]}
 
         return functions
diff --git a/lib/parameters.py b/lib/parameters.py
index 390420e..0e09610 100644
--- a/lib/parameters.py
+++ b/lib/parameters.py
@@ -598,7 +598,7 @@ class ModelAttribute:
 
         # There must be at least 3 distinct data values (≠ None) if an analytic model
         # is to be fitted. For 2 (or fewer) values, decision trees are better.
-        # Exceptions such as DFATOOL_FIT_LINEAR_ONLY=1 (2 values sufficient)
+        # Exceptions such as DFATOOL_SUBMODEL=fol (2 values sufficient)
         # can be handled via DFATOOL_ULS_MIN_DISTINCT_VALUES
         self.min_values_for_analytic_model = int(
             os.getenv("DFATOOL_ULS_MIN_DISTINCT_VALUES", "3")