diff options
Diffstat (limited to 'python/pyarmnn/test/test_runtime.py')
-rw-r--r-- | python/pyarmnn/test/test_runtime.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/python/pyarmnn/test/test_runtime.py b/python/pyarmnn/test/test_runtime.py index a6c4e1dcc0..a37772c5df 100644 --- a/python/pyarmnn/test/test_runtime.py +++ b/python/pyarmnn/test/test_runtime.py @@ -156,8 +156,8 @@ def test_load_network_properties_provided(random_runtime): opt_network, _ = ann.Optimize(network, preferred_backends, runtime.GetDeviceSpec(), ann.OptimizerOptions()) - inputSource = ann.MemorySource_Undefined - outputSource = ann.MemorySource_Undefined + inputSource = ann.MemorySource_Malloc + outputSource = ann.MemorySource_Malloc properties = ann.INetworkProperties(False, inputSource, outputSource) net_id, messages = runtime.LoadNetwork(opt_network, properties) assert "" == messages |