feat: Adds _config.__all__

vega · Sep 8, 2024 · 0af1cd1 · 0af1cd1
1 parent 2d89ad7
commit 0af1cd1
Show file tree

Hide file tree

Showing 2 changed files with 66 additions and 0 deletions.
diff --git a/altair/vegalite/v5/schema/_config.py b/altair/vegalite/v5/schema/_config.py
@@ -10,6 +10,69 @@
     from ._typing import *  # noqa: F403
 
 
+__all__ = [
+    "AreaConfigKwds",
+    "AutoSizeParamsKwds",
+    "AxisConfigKwds",
+    "BarConfigKwds",
+    "BindCheckboxKwds",
+    "BindDirectKwds",
+    "BindInputKwds",
+    "BindRadioSelectKwds",
+    "BindRangeKwds",
+    "BoxPlotConfigKwds",
+    "BrushConfigKwds",
+    "CompositionConfigKwds",
+    "DateTimeKwds",
+    "DerivedStreamKwds",
+    "ErrorBandConfigKwds",
+    "ErrorBarConfigKwds",
+    "FeatureGeometryGeoJsonPropertiesKwds",
+    "FormatConfigKwds",
+    "GeoJsonFeatureCollectionKwds",
+    "GeoJsonFeatureKwds",
+    "GeometryCollectionKwds",
+    "GradientStopKwds",
+    "HeaderConfigKwds",
+    "IntervalSelectionConfigKwds",
+    "IntervalSelectionConfigWithoutTypeKwds",
+    "LegendConfigKwds",
+    "LegendStreamBindingKwds",
+    "LineConfigKwds",
+    "LineStringKwds",
+    "LinearGradientKwds",
+    "LocaleKwds",
+    "MarkConfigKwds",
+    "MergedStreamKwds",
+    "MultiLineStringKwds",
+    "MultiPointKwds",
+    "MultiPolygonKwds",
+    "NumberLocaleKwds",
+    "OverlayMarkDefKwds",
+    "PointKwds",
+    "PointSelectionConfigKwds",
+    "PointSelectionConfigWithoutTypeKwds",
+    "PolygonKwds",
+    "ProjectionConfigKwds",
+    "RadialGradientKwds",
+    "RangeConfigKwds",
+    "RectConfigKwds",
+    "ScaleConfigKwds",
+    "ScaleInvalidDataConfigKwds",
+    "SelectionConfigKwds",
+    "StyleConfigIndexKwds",
+    "ThemeConfig",
+    "TickConfigKwds",
+    "TimeIntervalStepKwds",
+    "TimeLocaleKwds",
+    "TitleConfigKwds",
+    "TooltipContentKwds",
+    "TopLevelSelectionParameterKwds",
+    "VariableParameterKwds",
+    "ViewConfigKwds",
+]
+
+
 class AreaConfigKwds(TypedDict, total=False):
     """Placeholder doc."""
 

diff --git a/tools/generate_schema_wrapper.py b/tools/generate_schema_wrapper.py
@@ -836,6 +836,9 @@ def generate_config_typed_dicts(fp: Path, /) -> Iterator[str]:
         )
         for info in relevant.values()
     )
+    _all = [f"{nm}{KWDS}" for nm in relevant]
+    _all.append("ThemeConfig")
+    yield f"__all__ = {_all}\n\n"
     yield "\n".join(sub_dicts)
 
     for prop, prop_info in config.properties.items():