lint

Author: holtskinner

genai/controlled_generation/test_controlled_generation_samples.py

@@ -19,8 +19,8 @@
 import os
 
 import ctrlgen_with_class_schema
-import ctrlgen_with_enum_schema
 import ctrlgen_with_enum_class_schema
+import ctrlgen_with_enum_schema
 import ctrlgen_with_nested_class_schema
 import ctrlgen_with_nullable_schema
 import ctrlgen_with_resp_schema
@@ -35,14 +35,14 @@ def test_ctrlgen_with_class_schema() -> None:
     assert ctrlgen_with_class_schema.generate_content()
 
 
-def test_ctrlgen_with_enum_schema() -> None:
-    assert ctrlgen_with_enum_schema.generate_content()
-
-
 def test_ctrlgen_with_enum_class_schema() -> None:
     assert ctrlgen_with_enum_class_schema.generate_content()
 
 
+def test_ctrlgen_with_enum_schema() -> None:
+    assert ctrlgen_with_enum_schema.generate_content()
+
+
 def test_ctrlgen_with_nested_class_schema() -> None:
     assert ctrlgen_with_nested_class_schema.generate_content()