We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 23316c2 commit d5e8b7dCopy full SHA for d5e8b7d
src/pydvl/value/result.py
@@ -784,13 +784,17 @@ def zeros(
784
indices = np.arange(n_samples, dtype=np.int_)
785
else:
786
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
793
return cls(
794
algorithm=algorithm,
795
status=Status.Pending,
796
indices=indices,
- data_names=np.array(data_names, dtype=object)
- if data_names is not None
- else np.empty_like(indices, dtype=object),
797
+ data_names=data_names,
798
values=np.zeros(len(indices)),
799
variances=np.zeros(len(indices)),
800
counts=np.zeros(len(indices), dtype=np.int_),
0 commit comments