Return PyUnionMarker for wrapped type in IntoPy impl

smithy-lang · Mar 17, 2023 · 1fc0a92 · 1fc0a92
1 parent c9e1610
commit 1fc0a92
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/.../amazon/smithy/rust/codegen/server/python/smithy/generators/PythonServerUnionGenerator.kt b/.../amazon/smithy/rust/codegen/server/python/smithy/generators/PythonServerUnionGenerator.kt
@@ -89,12 +89,7 @@ class PythonServerUnionGenerator(
     private fun renderPyObjectConverters() {
         writer.rustBlockTemplate("impl #{pyo3}::IntoPy<#{pyo3}::PyObject> for ${unionSymbol.name}", "pyo3" to pyo3) {
             rustBlockTemplate("fn into_py(self, py: #{pyo3}::Python<'_>) -> #{pyo3}::PyObject", "pyo3" to pyo3) {
-                rustBlock("match self") {
-                    sortedMembers.forEach { member ->
-                        val variantName = symbolProvider.toMemberName(member)
-                        rust("${unionSymbol.name}::$variantName(variant) => variant.into_py(py),")
-                    }
-                }
+                rust("PyUnionMarker${unionSymbol.name}(self).into_py(py)")
             }
         }
         writer.rustBlockTemplate("impl<'source> #{pyo3}::FromPyObject<'source> for ${unionSymbol.name}", "pyo3" to pyo3) {