diff options
| author | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2023-02-02 09:07:02 +0000 |
|---|---|---|
| committer | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2023-02-06 11:51:52 +0000 |
| commit | c6cfc78d5245c550016b9709686d2b32ab3fcd5b (patch) | |
| tree | 42786a3755370d0fefea659d7e532a5483955239 /tests/test_target_tosa_data_collection.py | |
| parent | f1eaff3c9790464bed3183ff76555cf815166f50 (diff) | |
| download | mlia-c6cfc78d5245c550016b9709686d2b32ab3fcd5b.tar.gz | |
MLIA-461 Add parameter for the output directory
- Add CLI parameter --output-dir
- Rename ExecutionContext property working_dir into output_dir
- Remove logic for default command as it is no longer needed
Change-Id: I6387f6b688520ba1fc69a80167587297353620f6
Diffstat (limited to 'tests/test_target_tosa_data_collection.py')
| -rw-r--r-- | tests/test_target_tosa_data_collection.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/test_target_tosa_data_collection.py b/tests/test_target_tosa_data_collection.py index 9d590ca..75192fa 100644 --- a/tests/test_target_tosa_data_collection.py +++ b/tests/test_target_tosa_data_collection.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates. +# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 """Tests for TOSA data collection module.""" from pathlib import Path @@ -19,7 +19,7 @@ def test_tosa_data_collection( "mlia.target.tosa.data_collection.get_tosa_compatibility_info", MagicMock(return_value=TOSACompatibilityInfo(True, [])), ) - context = ExecutionContext(working_dir=tmpdir) + context = ExecutionContext(output_dir=tmpdir) collector = TOSAOperatorCompatibility(test_tflite_model) collector.set_context(context) |