fix typing

AllenInstitute · May 16, 2024 · 3163041 · 3163041
1 parent b2e0a2d
commit 3163041
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/src/aibs_informatics_cdk_lib/project/config.py b/src/aibs_informatics_cdk_lib/project/config.py
@@ -189,9 +189,6 @@ def load_stage_config(
         return proj_config.get_stage_config(env_type=env_type)
 
 
-# ProjectConfig = BaseProjectConfig[GlobalConfig, StageConfig]
-
-
 class ProjectConfig(BaseProjectConfig[GlobalConfig, StageConfig]):
     pass
 
@@ -203,6 +200,6 @@ def get_stage_config(
         env_type: Union[str, EnvType],
         path: Optional[Union[str, Path]] = None,
         project_config_cls: Type[BaseProjectConfig[G, S]] = ProjectConfig,
-    ) -> StageConfig:
+    ) -> S:
         proj_config = project_config_cls.load_config(path)
         return proj_config.get_stage_config(env_type=env_type)