Skip to content

Commit 60a56af

Browse files
committed
Fix test
1 parent 225be8e commit 60a56af

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

rust/worker/src/segment/test.rs

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ use chroma_types::{
1010
use crate::log::test::{LogGenerator, TEST_EMBEDDING_DIMENSION};
1111

1212
use super::{
13-
metadata_segment::MetadataSegmentWriter, record_segment::RecordSegmentWriter, LogMaterializer,
13+
materialize_logs, metadata_segment::MetadataSegmentWriter, record_segment::RecordSegmentWriter,
1414
SegmentFlusher, SegmentWriter,
1515
};
1616

@@ -26,12 +26,10 @@ pub struct TestSegment {
2626
impl TestSegment {
2727
// WARN: The size of the log chunk should not be too large
2828
async fn compact_log(&mut self, logs: Chunk<LogRecord>, offset: usize) {
29-
let materializer =
30-
LogMaterializer::new(None, logs, Some(AtomicU32::new(offset as u32).into()));
31-
let materialized_logs = materializer
32-
.materialize()
33-
.await
34-
.expect("Should be able to materialize log.");
29+
let materialized_logs =
30+
materialize_logs(&None, &logs, Some(AtomicU32::new(offset as u32).into()))
31+
.await
32+
.expect("Should be able to materialize log.");
3533

3634
let mut metadata_writer =
3735
MetadataSegmentWriter::from_segment(&self.metadata_segment, &self.blockfile_provider)

0 commit comments

Comments
 (0)