replace lint with observable matching

Author: m-philipps

petab/v1/core.py

@@ -15,7 +15,6 @@
 
 from . import yaml
 from .C import *  # noqa: F403
-from .visualize.lint import validate_visualization_df
 
 logger = logging.getLogger(__name__)
 __all__ = [
@@ -300,8 +299,14 @@ def flatten_timepoint_specific_output_overrides(
     petab_problem.observable_df.index.name = OBSERVABLE_ID
     petab_problem.measurement_df = pd.concat(new_measurement_dfs)
 
-    # remove visualization df if it is invalid
-    if validate_visualization_df(petab_problem):
+    # remove visualization df if it uses observables that are not in the
+    # flattened PEtab problem
+    assert petab_problem.observable_df.index.name == OBSERVABLE_ID
+    if not all(
+        petab_problem.observable_df.index.isin(
+            petab_problem.visualization_df[Y_VALUES]
+        )
+    ):
         petab_problem.visualization_df = None
         logger.warning(
             "Removing visualization table from flattened PEtab problem."