diff --git a/src/arcaflow_plugin_sdk/schema.py b/src/arcaflow_plugin_sdk/schema.py
index c4a6b8a..5fe6fd3 100644
--- a/src/arcaflow_plugin_sdk/schema.py
+++ b/src/arcaflow_plugin_sdk/schema.py
@@ -5565,7 +5565,6 @@ def __init__(
             types=types,
             discriminator_inlined=discriminator_inlined,
             discriminator_field_name=discriminator_field_name,
-            # oneof_type="_discriminated_string_",
         )
         _OneOfType.__init__(
             self,
@@ -5606,7 +5605,6 @@ def __init__(
             self,
             types=types,
             discriminator_inlined=discriminator_inlined,
-            # oneof_type="_discriminated_int_",
             discriminator_field_name=discriminator_field_name,
         )
         _OneOfType.__init__(