Adds MapKeyToFloat Transform to registry (TRANSFORM_REGISTRY)

Differential Revision: D70900626
facebook · Mar 10, 2025 · 1fcb3a2 · 1fcb3a2
1 parent fa42528
commit 1fcb3a2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ax/storage/transform_registry.py b/ax/storage/transform_registry.py
@@ -23,6 +23,7 @@
 from ax.modelbridge.transforms.log import Log
 from ax.modelbridge.transforms.log_y import LogY
 from ax.modelbridge.transforms.logit import Logit
+from ax.modelbridge.transforms.map_key_to_float import MapKeyToFloat
 from ax.modelbridge.transforms.map_unit_x import MapUnitX
 from ax.modelbridge.transforms.merge_repeated_measurements import (
     MergeRepeatedMeasurements,
@@ -96,6 +97,7 @@
     TransformToNewSQ: 28,
     FillMissingParameters: 29,
     LogIntToFloat: 30,
+    MapKeyToFloat: 31,
 }
 
 """