Skip to content
Snippets Groups Projects
Commit 90453ad1 authored by Anjith GEORGE's avatar Anjith GEORGE
Browse files

small fix in unit tests

parent 8ab4559f
No related branches found
No related tags found
1 merge request!22Cross validation
Pipeline #27526 passed
......@@ -394,7 +394,7 @@ def test_extractors():
from ..extractor.image import FASNetExtractor
extractor = FASNetExtractor()
# this architecture expects RGB images of size 3x224x224 channel images
data = numpy.random.rand(3, 224, 224).astype("float32")
data = numpy.random.rand(3, 224, 224).astype("uint8")
output = extractor(data)
assert output.shape[0] == 1
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment