Update utils/generator.py

Co-authored-by: Quentin Pradet <[email protected]>
miguelgrinberg · Sep 9, 2024 · 8b8143d · 8b8143d
1 parent 83559ab
commit 8b8143d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/utils/generator.py b/utils/generator.py
@@ -210,6 +210,7 @@ def get_python_type(self, schema_type):
                 if schema_type["name"]["name"].endswith("RangeQuery"):
                     return '"wrappers.Range[Any]"', None
                 elif schema_type["name"]["name"].endswith("ScoreFunction"):
+                    # When dropping Python 3.8, use `removesuffix("Function")` instead
                     name = schema_type["name"]["name"][:-8]
                     return f'"function.{name}"', None
                 elif schema_type["name"]["name"].endswith("DecayFunction"):