From 0e4d7fd2ef57e38d634c7121633b42d6b811c0d6 Mon Sep 17 00:00:00 2001 From: Adam Lewis <23342526+Adam-D-Lewis@users.noreply.github.com> Date: Sat, 9 Mar 2024 14:02:12 -0600 Subject: [PATCH] allow each InputSchema to exclude certain parts of itself from the config --- src/nebari/plugins.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/src/nebari/plugins.py b/src/nebari/plugins.py index 8332e2adf2..63aa9762f1 100644 --- a/src/nebari/plugins.py +++ b/src/nebari/plugins.py @@ -125,7 +125,9 @@ def ordered_stages(self): @property def ordered_schemas(self): - return [schema.Main] + [_.input_schema for _ in self.ordered_stages if _.input_schema is not None] + return [schema.Main] + [ + _.input_schema for _ in self.ordered_stages if _.input_schema is not None + ] @property def config_schema(self): @@ -139,14 +141,14 @@ def write_config(self): config_exclude = config_exclude.union(new_exclude) return self.dict(exclude=config_exclude) - return type( - "ConfigSchema", - tuple(ordered_schemas), + "ConfigSchema", + tuple(ordered_schemas), { "_ordered_schemas": ordered_schemas, "write_config": write_config, - }) + }, + ) nebari_plugin_manager = NebariPluginManager()