Skip to content

Commit ee2abf9

Browse files
MirjalilovSskshetry
authored andcommitted
Addressed review feedback regarding ExpState
1 parent a2fc897 commit ee2abf9

File tree

1 file changed

+7
-12
lines changed

1 file changed

+7
-12
lines changed

dvc/repo/experiments/collect.py

Lines changed: 7 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -303,18 +303,13 @@ def collect(
303303
baseline_names = describe(
304304
repo.scm, baseline_revs, refs=cached_refs, logger=logger
305305
)
306-
result: list[ExpState] = []
307-
if not hide_workspace:
308-
workspace_data = collect_rev(repo, "workspace", **kwargs)
309-
result.append(
310-
ExpState(
311-
rev="workspace",
312-
name=baseline_names.get("workspace"),
313-
data=workspace_data.data,
314-
error=workspace_data.error,
315-
experiments=None,
316-
)
317-
)
306+
workspace_data = collect_rev(repo, "workspace", **kwargs)
307+
result: list[ExpState]
308+
if hide_workspace:
309+
result = []
310+
else:
311+
result = [workspace_data]
312+
318313
queued = collect_queued(repo, baseline_revs, **kwargs) if not hide_queued else {}
319314
active = collect_active(repo, baseline_revs, **kwargs)
320315
failed = collect_failed(repo, baseline_revs, **kwargs) if not hide_failed else {}

0 commit comments

Comments
 (0)