Skip to content

Commit d5e8b7d

Browse files
committed
Don't leave names empty in values factory
1 parent 23316c2 commit d5e8b7d

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

src/pydvl/value/result.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -784,13 +784,17 @@ def zeros(
784784
indices = np.arange(n_samples, dtype=np.int_)
785785
else:
786786
indices = np.array(indices, dtype=np.int_)
787+
788+
if data_names is None:
789+
data_names = np.array(indices)
790+
else:
791+
data_names = np.array(data_names)
792+
787793
return cls(
788794
algorithm=algorithm,
789795
status=Status.Pending,
790796
indices=indices,
791-
data_names=np.array(data_names, dtype=object)
792-
if data_names is not None
793-
else np.empty_like(indices, dtype=object),
797+
data_names=data_names,
794798
values=np.zeros(len(indices)),
795799
variances=np.zeros(len(indices)),
796800
counts=np.zeros(len(indices), dtype=np.int_),

0 commit comments

Comments
 (0)