diff --git a/edgebench/config.py b/edgebench/config.py index 4949fd3..6cf6ff5 100644 --- a/edgebench/config.py +++ b/edgebench/config.py @@ -2,7 +2,11 @@ from pathlib import Path from typing import Any, Dict -import tomllib + +try: + import tomllib +except ModuleNotFoundError: # Python 3.10 fallback + import tomli as tomllib DEFAULT_COMPARE_THRESHOLDS: Dict[str, float] = { diff --git a/pyproject.toml b/pyproject.toml index 759ad85..00a0a63 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ version = "0.1.0" description = "Edge AI Inference Profiling Framework" authors = [{ name = "Your Name" }] readme = "README.md" -requires-python = ">=3.11" +requires-python = ">=3.10,<3.12" dependencies = [ "onnx", "onnxruntime",