Fix example generator wrapping named number types in non-existent constructors

.generator/src/generator/formatter.py

@@ -277,7 +277,7 @@ def format_data_with_schema(
     imports = imports or defaultdict(set)
     nullable = schema.get("nullable", False)
 
-    if schema.get("type") not in {"string", "integer", "boolean"} or schema.get("enum"):
+    if schema.get("type") not in {"string", "integer", "boolean", "number"} or schema.get("enum"):
         name, imports = get_name_and_imports(schema, version, imports)
     if schema.get("oneOf"):
         name = None

.generator/tests/test_formatter.py

@@ -91,3 +91,33 @@ def test_dict_with_valid_identifiers_uses_constructor(self):
         assert result.endswith(")")
         assert "normal_key='value1'" in result
         assert "another_key='value2'" in result
+
+
+class SchemaWithReference(dict):
+    """Dict subclass that carries a __reference__ attribute, mimicking a JsonRef proxy."""
+    def __init__(self, data, ref_name):
+        super().__init__(data)
+        self.__reference__ = {"$ref": f"#/components/schemas/{ref_name}"}
+
+
+class TestFormatDataWithSchemaNamedNumberType:
+    def test_named_number_type_returns_raw_float(self):
+        schema = SchemaWithReference({"type": "number", "format": "double"}, "RumRetentionFilterSampleRate")
+        result, imports = format_data_with_schema(50.0, schema, version="v2")
+        assert result == "50.0"
+        assert "RumRetentionFilterSampleRate" not in result
+        assert all("rum_retention_filter_sample_rate" not in k for k in imports)
+
+    def test_named_integer_type_still_works(self):
+        schema = SchemaWithReference({"type": "integer", "format": "int64"}, "SomeNamedInteger")
+        result, imports = format_data_with_schema(42, schema, version="v2")
+        assert result == "42"
+        assert "SomeNamedInteger" not in result
+
+    def test_named_number_enum_still_uses_constructor(self):
+        schema = SchemaWithReference(
+            {"type": "number", "enum": [0.5, 1.0], "x-enum-varnames": ["HALF", "ONE"]},
+            "SomeNumberEnum",
+        )
+        result, imports = format_data_with_schema(0.5, schema, version="v2")
+        assert result == "SomeNumberEnum.HALF"

examples/v2/rum-retention-filters/CreateRetentionFilter.py

@@ -8,7 +8,6 @@
 from datadog_api_client.v2.model.rum_retention_filter_create_data import RumRetentionFilterCreateData
 from datadog_api_client.v2.model.rum_retention_filter_create_request import RumRetentionFilterCreateRequest
 from datadog_api_client.v2.model.rum_retention_filter_event_type import RumRetentionFilterEventType
-from datadog_api_client.v2.model.rum_retention_filter_sample_rate import RumRetentionFilterSampleRate
 from datadog_api_client.v2.model.rum_retention_filter_type import RumRetentionFilterType
 
 body = RumRetentionFilterCreateRequest(
@@ -18,7 +17,7 @@
             name="Test creating retention filter",
             event_type=RumRetentionFilterEventType.SESSION,
             query="custom_query",
-            sample_rate=RumRetentionFilterSampleRate(50.0),
+            sample_rate=50.0,
             enabled=True,
         ),
     ),

examples/v2/rum-retention-filters/UpdateRetentionFilter.py

@@ -5,7 +5,6 @@
 from datadog_api_client import ApiClient, Configuration
 from datadog_api_client.v2.api.rum_retention_filters_api import RumRetentionFiltersApi
 from datadog_api_client.v2.model.rum_retention_filter_event_type import RumRetentionFilterEventType
-from datadog_api_client.v2.model.rum_retention_filter_sample_rate import RumRetentionFilterSampleRate
 from datadog_api_client.v2.model.rum_retention_filter_type import RumRetentionFilterType
 from datadog_api_client.v2.model.rum_retention_filter_update_attributes import RumRetentionFilterUpdateAttributes
 from datadog_api_client.v2.model.rum_retention_filter_update_data import RumRetentionFilterUpdateData
@@ -19,7 +18,7 @@
             name="Test updating retention filter",
             event_type=RumRetentionFilterEventType.VIEW,
             query="view_query",
-            sample_rate=RumRetentionFilterSampleRate(100.0),
+            sample_rate=100.0,
             enabled=True,
         ),
     ),