diff --git a/lib/crewai/tests/test_dependency_constraints.py b/lib/crewai/tests/test_dependency_constraints.py index f4dd02324..cda34505a 100644 --- a/lib/crewai/tests/test_dependency_constraints.py +++ b/lib/crewai/tests/test_dependency_constraints.py @@ -8,9 +8,15 @@ to defend against so future maintainers understand why a bound is required. from __future__ import annotations -import tomllib +import sys from pathlib import Path + +if sys.version_info >= (3, 11): + import tomllib +else: + import tomli as tomllib + import pytest from packaging.requirements import Requirement from packaging.specifiers import SpecifierSet