Skip to content

Commit e044e5a

Browse files
committed
fix broken test
1 parent 1b1ae3b commit e044e5a

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

backends/apple/coreml/runtime/test/ETCoreMLModelManagerTests.mm

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -172,10 +172,10 @@ - (void)testAutoreleasepoolError {
172172
[ETCoreMLTestUtils inputsForModel:model repeatedValues:@[@(2), @(3)] error:&localError];
173173
XCTAssert(inputArrays);
174174

175-
std::vector<MultiArray> multiArrays;
175+
std::vector<executorchcoreml::MultiArray> multiArrays;
176176
multiArrays.reserve(inputArrays.count + model.orderedOutputNames.count);
177177
for (MLMultiArray *array in inputArrays) {
178-
auto dataTypeOpt = to_multiarray_data_type(array.dataType);
178+
auto dataTypeOpt = executorchcoreml::to_multiarray_data_type(array.dataType);
179179
XCTAssert(dataTypeOpt.has_value());
180180
auto dataType = dataTypeOpt.value();
181181

@@ -192,7 +192,7 @@ - (void)testAutoreleasepoolError {
192192
}
193193

194194
multiArrays.emplace_back(array.dataPointer,
195-
MultiArray::MemoryLayout(dataType, dims, strides));
195+
executorchcoreml::MultiArray::MemoryLayout(dataType, dims, strides));
196196
}
197197

198198
auto inputLayout = multiArrays[0].layout();
@@ -205,15 +205,15 @@ - (void)testAutoreleasepoolError {
205205
auto originalLayout = multiArrays[0].layout();
206206
auto corruptedDims = originalLayout.shape();
207207
corruptedDims[0] += 1;
208-
multiArrays[0] = MultiArray(multiArrays[0].data(),
209-
MultiArray::MemoryLayout(originalLayout.dataType(),
208+
multiArrays[0] = executorchcoreml::MultiArray(multiArrays[0].data(),
209+
executorchcoreml::MultiArray::MemoryLayout(originalLayout.dataType(),
210210
corruptedDims,
211211
originalLayout.strides()));
212212
}
213213

214214
BOOL success = [self.modelManager executeModelWithHandle:modelHandle
215215
argsVec:multiArrays
216-
loggingOptions:ModelLoggingOptions()
216+
loggingOptions:executorchcoreml::ModelLoggingOptions()
217217
eventLogger:nullptr
218218
error:&localError];
219219
XCTAssertFalse(success);

0 commit comments

Comments
 (0)