Merge pull request #13 from Harsh-br0/ditch-packaging

ditch packaging

Author: comfuture

function_schema/core.py

@@ -1,23 +1,20 @@
 import enum
 import inspect
-import platform
-import packaging.version
 from typing import (
     Annotated,
-    Optional,
-    Union,
+    Any,
     Callable,
     Literal,
-    Any,
+    Optional,
+    Union,
     get_args,
     get_origin,
 )
-from .types import FunctionSchema
 
-current_version = packaging.version.parse(platform.python_version())
-py_310 = packaging.version.parse("3.10")
+from .types import FunctionSchema
+from .utils import is_py310_atleast
 
-if current_version >= py_310:
+if is_py310_atleast():
     from types import UnionType
 else:
     UnionType = Union  # type: ignore

function_schema/utils.py

@@ -0,0 +1,6 @@
+import platform
+
+
+def is_py310_atleast():
+    version_tuple = tuple(map(int, platform.python_version_tuple()))
+    return version_tuple >= (3, 10)

test/test_guess_type.py

@@ -1,9 +1,9 @@
-import platform
-import packaging.version
 import typing
 
 import pytest
+
 from function_schema.core import guess_type
+from function_schema.utils import is_py310_atleast
 
 
 def test_primitive():
@@ -67,12 +67,8 @@ def test_union():
     }
 
 
-current_version = packaging.version.parse(platform.python_version())
-py_310 = packaging.version.parse("3.10")
-
-
 @pytest.mark.skipif(
-    current_version < py_310, reason="Union type is only available in Python 3.10+"
+    not is_py310_atleast(), reason="Union type is only available in Python 3.10+"
 )
 def test_union_type():
     """Test union types in Python 3.10+"""