Method test_namespaced

axlearn/cloud/common/utils_test.py:528–644

(self)

        self.assertIsNone(cfg_no_inner.inner)

    def test_namespaced(self):
        # pylint: disable=missing-class-docstring
        fv = flags.FlagValues()

        class GrandChild(utils.FlagConfigurable):
            @config_class
            class Config(utils.FlagConfigurable.Config):
                grandchild: Optional[str] = None
                child_default: Optional[str] = None
                parent_default: Optional[str] = None

            @classmethod
            def define_flags(cls, fv: flags.FlagValues):
                flags.DEFINE_string("grandchild", None, "", flag_values=fv, allow_override=True)
                flags.DEFINE_string("child_default", None, "", flag_values=fv, allow_override=True)
                flags.DEFINE_string("parent_default", None, "", flag_values=fv, allow_override=True)

        @utils.namespaced("inner")
        class Child(utils.FlagConfigurable):
            @config_class
            class Config(utils.FlagConfigurable.Config):
                inner: Required[dict[str, ConfigBase]] = REQUIRED
                child: Optional[str] = None
                child_default: Optional[str] = None
                parent_default: Optional[str] = None

            @classmethod
            def default_config(cls) -> Config:
                return super().default_config().set(inner={"c": GrandChild.default_config()})

            @classmethod
            def define_flags(cls, fv: flags.FlagValues):
                flags.DEFINE_string("child", None, "", flag_values=fv, allow_override=True)
                flags.DEFINE_string(
                    "child_default", "from_child", "", flag_values=fv, allow_override=True
                )

        @utils.namespaced("inner")
        class Parent(utils.FlagConfigurable):
            @config_class
            class Config(utils.FlagConfigurable.Config):
                inner: Required[dict[str, ConfigBase]] = REQUIRED
                parent: Optional[str] = None
                parent_default: Optional[str] = None

            @classmethod
            def default_config(cls):
                return (
                    super()
                    .default_config()
                    .set(inner={"a": Child.default_config(), "b": Child.default_config()})
                )

            @classmethod
            def define_flags(cls, fv: flags.FlagValues):
                flags.DEFINE_string("parent", None, "", flag_values=fv, allow_override=True)
                flags.DEFINE_string(
                    "parent_default", "from_parent", "", flag_values=fv, allow_override=True