truefoundry 0.5.10rc1__py3-none-any.whl → 0.5.11rc1__py3-none-any.whl

truefoundry/deploy/builder/__init__.py

@@ -1,6 +1,6 @@
 from typing import Any, Dict, List, Optional, Union
 
-from truefoundry.deploy.auto_gen.models import (
+from truefoundry.deploy._autogen.models import (
     DockerFileBuild,
     PythonBuild,
     TaskDockerFileBuild,

truefoundry/deploy/builder/builders/dockerfile.py

@@ -1,7 +1,7 @@
 import os
 from typing import Dict, List, Optional
 
-from truefoundry.deploy.auto_gen.models import DockerFileBuild
+from truefoundry.deploy._autogen.models import DockerFileBuild
 from truefoundry.deploy.builder.docker_service import build_docker_image
 from truefoundry.logger import logger
 

truefoundry/deploy/builder/builders/tfy_notebook_buildpack/__init__.py

@@ -2,7 +2,7 @@ import os
 from tempfile import TemporaryDirectory
 from typing import List, Optional
 
-from truefoundry.deploy.auto_gen.models import DockerFileBuild
+from truefoundry.deploy._autogen.models import DockerFileBuild
 from truefoundry.deploy.builder.builders import dockerfile
 from truefoundry.deploy.builder.builders.tfy_notebook_buildpack.dockerfile_template import (
     NotebookImageBuild,

truefoundry/deploy/builder/builders/tfy_python_buildpack/__init__.py

@@ -2,7 +2,7 @@ import os
 from tempfile import TemporaryDirectory
 from typing import List, Optional
 
-from truefoundry.deploy.auto_gen.models import DockerFileBuild, PythonBuild
+from truefoundry.deploy._autogen.models import DockerFileBuild, PythonBuild
 from truefoundry.deploy.builder.builders import dockerfile
 from truefoundry.deploy.builder.builders.tfy_python_buildpack.dockerfile_template import (
     generate_dockerfile_content,

truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py

@@ -4,7 +4,7 @@ from typing import Dict, List, Optional
 from mako.template import Template
 
 from truefoundry.common.constants import ENV_VARS, PythonPackageManager
-from truefoundry.deploy.auto_gen.models import PythonBuild
+from truefoundry.deploy._autogen.models import PythonBuild
 from truefoundry.deploy.builder.constants import (
     PIP_CONF_BUILDKIT_SECRET_MOUNT,
     PIP_CONF_SECRET_MOUNT_AS_ENV,

truefoundry/deploy/io/output_callback.py

@@ -1,23 +1,30 @@
+from typing import Any, List, Optional
+
+
 class OutputCallBack:
-    def print_header(self, line):
+    def print_header(self, line: Any) -> None:
         print(line)
 
-    def _print_separator(self):
+    def _print_separator(self) -> None:
         print("-" * 80)
 
-    def print_line(self, line):
+    def print_line(self, line: str) -> None:
         print(line)
 
-    def print_lines_in_panel(self, lines, header=None):
+    def print_lines_in_panel(
+        self, lines: List[str], header: Optional[str] = None
+    ) -> None:
         self.print_header(header)
         self._print_separator()
         for line in lines:
             self.print_line(line)
         self._print_separator()
 
-    def print_code_in_panel(self, lines, header=None, lang="python"):
+    def print_code_in_panel(
+        self, lines: List[str], header: Optional[str] = None
+    ) -> None:
         self.print_lines_in_panel(lines, header)
 
-    def print(self, line):
+    def print(self, line: Any) -> None:
         # just an alias
         self.print_line(line)

truefoundry/deploy/io/rich_output_callback.py

@@ -1,3 +1,5 @@
+from typing import Any, List, Optional
+
 from rich.console import Console
 from rich.highlighter import ReprHighlighter
 from rich.panel import Panel
@@ -14,14 +16,16 @@ class RichOutputCallBack(OutputCallBack):
     console = Console(soft_wrap=True)
     highlighter = ReprHighlighter()
 
-    def print_header(self, line):
+    def print_header(self, line: Any) -> None:
         self.console.rule(_text(line), style="cyan")
 
-    def print_line(self, line):
+    def print_line(self, line: Any) -> None:
         self.console.print(_text(line))
 
-    def print_lines_in_panel(self, lines, header=None):
+    def print_lines_in_panel(
+        self, lines: List[str], header: Optional[str] = None
+    ) -> None:
         self.console.print(Panel(self.highlighter("\n".join(lines)), title=header))
 
-    def print_code_in_panel(self, lines, header=None, lang="python"):
+    def print_code_in_panel(self, lines: List[str], header: Optional[str] = None):
         self.print_lines_in_panel(lines, header)