Move serialization_version out of root

hugr-py/src/hugr/__init__.py

@@ -17,14 +17,8 @@
     "Kind",
     "Type",
     "Wire",
-    "get_serialisation_version",
 ]
 
 # This is updated by our release-please workflow, triggered by this
 # annotation: x-release-please-version
 __version__ = "0.5.0"
-
-
-def get_serialization_version() -> str:
-    """Return the current version of the serialization schema."""
-    return "live"

hugr-py/src/hugr/serialization/extension.py

@@ -4,7 +4,7 @@
 from pydantic_extra_types.semantic_version import SemanticVersion
 
 from .ops import Value
-from .serial_hugr import SerialHugr
+from .serial_hugr import SerialHugr, serialization_version
 from .tys import (
     ConfiguredBaseModel,
     ExtensionId,
@@ -74,9 +74,9 @@ class Extension(ConfiguredBaseModel):
 
     @classmethod
     def get_version(cls) -> str:
-        from hugr import get_serialization_version
+        from hugr.serialization.serial_hugr import serialization_version
 
-        return get_serialization_version()
+        return serialization_version()
 
 
 class Package(ConfiguredBaseModel):
@@ -85,6 +85,4 @@ class Package(ConfiguredBaseModel):
 
     @classmethod
     def get_version(cls) -> str:
-        from hugr import get_serialization_version
-
-        return get_serialization_version()
+        return serialization_version()

hugr-py/src/hugr/serialization/serial_hugr.py

@@ -12,15 +12,13 @@
 Edge = tuple[Port, Port]
 
 
-def get_serialization_version() -> str:
-    # Delay importing hugr to avoid circular imports
-    from hugr import get_serialization_version
-
-    return get_serialization_version()
+def serialization_version() -> str:
+    """Return the current version of the serialization schema."""
+    return "live"
 
 
 VersionField = Field(
-    default_factory=get_serialization_version,
+    default_factory=serialization_version,
     title="Version",
     description="Serialisation Schema Version",
     frozen=True,

hugr-py/tests/serialization/test_basic.py

@@ -1,11 +1,10 @@
-from hugr import get_serialization_version
-from hugr.serialization.serial_hugr import SerialHugr
+from hugr.serialization.serial_hugr import SerialHugr, serialization_version
 
 
 def test_empty():
     h = SerialHugr(nodes=[], edges=[])
     assert h.model_dump() == {
-        "version": get_serialization_version(),
+        "version": serialization_version(),
         "nodes": [],
         "edges": [],
         "metadata": None,

hugr-py/tests/serialization/test_extension.py

@@ -1,6 +1,5 @@
 from semver import Version
 
-from hugr import get_serialization_version
 from hugr.serialization.extension import (
     ExplicitBound,
     Extension,
@@ -9,7 +8,7 @@
     TypeDef,
     TypeDefBound,
 )
-from hugr.serialization.serial_hugr import SerialHugr
+from hugr.serialization.serial_hugr import SerialHugr, serialization_version
 from hugr.serialization.tys import (
     FunctionType,
     PolyFuncType,
@@ -75,7 +74,7 @@
 
 
 def test_extension():
-    assert get_serialization_version() == Extension.get_version()
+    assert serialization_version() == Extension.get_version()
     param = TypeParam(root=TypeTypeParam(b=TypeBound.Copyable))
 
     bound = TypeDefBound(root=ExplicitBound(bound=TypeBound.Copyable))
@@ -113,7 +112,7 @@ def test_extension():
 
 
 def test_package():
-    assert get_serialization_version() == Package.get_version()
+    assert serialization_version() == Package.get_version()
 
     ext = Extension(
         version=Version(0, 1, 0),