diff --git a/DashAI/back/exploration/explorers/density_heatmap.py b/DashAI/back/exploration/explorers/density_heatmap.py index ce6d22605..a62b51964 100644 --- a/DashAI/back/exploration/explorers/density_heatmap.py +++ b/DashAI/back/exploration/explorers/density_heatmap.py @@ -5,6 +5,8 @@ from DashAI.back.dependencies.database.models import Explorer, Notebook from DashAI.back.exploration.base_explorer import BaseExplorerSchema from DashAI.back.exploration.relationship_explorer import RelationshipExplorer +from DashAI.back.types.categorical import Categorical +from DashAI.back.types.value_types import Float, Integer if TYPE_CHECKING: from pathlib import Path @@ -99,7 +101,7 @@ class DensityHeatmapExplorer(RelationshipExplorer): SCHEMA = DensityHeatmapSchema metadata: Dict[str, Any] = { - "allowed_types": [], + "allowed_types": [Float, Integer, Categorical], "allowed_dtypes": [], "input_cardinality": {"exact": 2}, } diff --git a/DashAI/back/exploration/explorers/histogram_plot.py b/DashAI/back/exploration/explorers/histogram_plot.py index 987a8bcf3..601ca6097 100644 --- a/DashAI/back/exploration/explorers/histogram_plot.py +++ b/DashAI/back/exploration/explorers/histogram_plot.py @@ -13,6 +13,8 @@ from DashAI.back.dependencies.database.models import Explorer, Notebook from DashAI.back.exploration.base_explorer import BaseExplorerSchema from DashAI.back.exploration.distribution_explorer import DistributionExplorer +from DashAI.back.types.categorical import Categorical +from DashAI.back.types.value_types import Float, Integer if TYPE_CHECKING: from pathlib import Path @@ -183,7 +185,7 @@ class HistogramPlotExplorer(DistributionExplorer): SCHEMA = HistogramPlotSchema metadata: Dict[str, Any] = { - "allowed_types": [], + "allowed_types": [Float, Integer, Categorical], "allowed_dtypes": [], "input_cardinality": {"exact": 1}, }