Fix '--where' experiment index filter on info

Fixes a bug where filtering using '--where' on the experiment index does not work correctly
GoogleCloudPlatform · Sep 11, 2024 · 60c87ce · 60c87ce
1 parent 95fd326
commit 60c87ce
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/lib/ramble/ramble/cmd/workspace.py b/lib/ramble/ramble/cmd/workspace.py
@@ -662,6 +662,19 @@ def workspace_info(args):
                 print_experiment_set.set_experiment_context(experiment_context)
                 print_experiment_set.build_experiment_chains()
 
+                # Reindex the experiments in the print set to match the overall set
+                for exp_name, _, _ in print_experiment_set.all_experiments():
+                    app_inst = experiment_set.get_experiment(exp_name)
+                    experiment_index = app_inst.expander.expand_var_name(
+                        app_inst.keywords.experiment_index
+                    )
+
+                    print_app_inst = print_experiment_set.get_experiment(exp_name)
+                    print_app_inst.define_variable(
+                        print_app_inst.keywords.experiment_index,
+                        experiment_index
+                    )
+
                 print_header = True
                 # Define variable printing groups.
                 var_indent = "        "