Skip to content

Commit 8560117

Browse files
committed
1 parent 804f2bf commit 8560117

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/tiledbsoma_ml/io_batch_iterable.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -64,9 +64,9 @@ def __iter__(self) -> Iterator[IOBatch]:
6464
if "soma_joinid" in self.obs_column_names
6565
else ["soma_joinid", *self.obs_column_names]
6666
)
67-
# Round-trip though tuple avoids `TypeError: IntIndexer only supports array of type int64`.
68-
# TODO: debug / work around that error; serde'ing the ndarray apparently results in a second np.int64 instance, that fails reference equality check vs. the version from the worker-process.
69-
var_joinids = np.array(tuple(self.var_joinids))
67+
# NOTE: `.astype("int64")` works around the `np.int64` singleton failing reference-equality after cross-process
68+
# serde.
69+
var_joinids = self.var_joinids.astype("int64")
7070
var_indexer = IntIndexer(var_joinids, context=context)
7171

7272
for obs_coords in self.io_batch_ids:

0 commit comments

Comments
 (0)