Update cli_wrapper.py (#572)

gabrieldemarmiesse · Mar 22, 2024 · a8714e4 · a8714e4
1 parent 070bf6e
commit a8714e4
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/python_on_whales/components/compose/cli_wrapper.py b/python_on_whales/components/compose/cli_wrapper.py
@@ -107,6 +107,14 @@ def build(
         else:
             run(full_cmd, capture_stdout=False)
 
+    @overload
+    def config(self, return_json: Literal[False] = ...) -> ComposeConfig:
+        ...
+
+    @overload
+    def config(self, return_json: Literal[True] = ...) -> Dict[str, Any]:
+        ...
+
     def config(self, return_json: bool = False) -> Union[ComposeConfig, Dict[str, Any]]:
         """Returns the configuration of the compose stack for further inspection.