use validators from ValidatorCache

Author: bmaranville

plotly/figure_factory/_annotated_heatmap.py

@@ -2,7 +2,8 @@
 from plotly import exceptions, optional_imports
 from plotly.figure_factory import utils
 from plotly.graph_objs import graph_objs
-from plotly.validators.heatmap import ColorscaleValidator
+from plotly.validator_cache import ValidatorCache
+# from _plotly_utils.basevalidators import ColorscaleValidator
 
 # Optional imports, may be None for users that only use our core functionality.
 np = optional_imports.get_module("numpy")
@@ -102,7 +103,7 @@ def create_annotated_heatmap(
     validate_annotated_heatmap(z, x, y, annotation_text)
 
     # validate colorscale
-    colorscale_validator = ColorscaleValidator()
+    colorscale_validator = ValidatorCache.get_validator("layout", "colorscale")
     colorscale = colorscale_validator.validate_coerce(colorscale)
 
     annotations = _AnnotatedHeatmap(