avoid using dataset classes for indexing, just use strings

janelia-cellmap · Nov 12, 2024 · 64a508b · 64a508b
1 parent f0a7a39
commit 64a508b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dacapo/experiments/validation_scores.py b/dacapo/experiments/validation_scores.py
@@ -234,7 +234,7 @@ def to_xarray(self) -> xr.DataArray:
                 "iterations": [
                     iteration_score.iteration for iteration_score in self.scores
                 ],
-                "datasets": self.datasets,
+                "datasets": [d.name for d in self.datasets],
                 "parameters": self.parameters,
                 "criteria": self.criteria,
             },

diff --git a/dacapo/plot.py b/dacapo/plot.py
@@ -427,7 +427,7 @@ def plot_runs(
             )
             colors_val = itertools.cycle(plt.cm.tab20.colors)
             for dataset in run.validation_scores.datasets:
-                dataset_data = validation_score_data.sel(datasets=dataset)
+                dataset_data = validation_score_data.sel(datasets=dataset.name)
                 include_validation_figure = True
                 x = [score.iteration for score in run.validation_scores.scores]
                 for i, cc in zip(range(dataset_data.data.shape[1]), colors_val):