diff options
Diffstat (limited to 'src/mlia/backend/tosa_checker/compat.py')
-rw-r--r-- | src/mlia/backend/tosa_checker/compat.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/mlia/backend/tosa_checker/compat.py b/src/mlia/backend/tosa_checker/compat.py index 81f3015..1c410d3 100644 --- a/src/mlia/backend/tosa_checker/compat.py +++ b/src/mlia/backend/tosa_checker/compat.py @@ -5,12 +5,12 @@ from __future__ import annotations import sys from dataclasses import dataclass +from pathlib import Path from typing import Any from typing import cast from typing import Protocol from mlia.backend.errors import BackendUnavailableError -from mlia.core.typing import PathOrFileLike from mlia.utils.logging import capture_raw_output @@ -45,7 +45,7 @@ class TOSACompatibilityInfo: def get_tosa_compatibility_info( - tflite_model_path: PathOrFileLike, + tflite_model_path: str | Path, ) -> TOSACompatibilityInfo: """Return list of the operators.""" # Capture the possible exception in running get_tosa_checker @@ -100,7 +100,7 @@ def get_tosa_compatibility_info( ) -def get_tosa_checker(tflite_model_path: PathOrFileLike) -> TOSAChecker | None: +def get_tosa_checker(tflite_model_path: str | Path) -> TOSAChecker | None: """Return instance of the TOSA checker.""" try: import tosa_checker as tc # pylint: disable=import-outside-toplevel |