rename DockerHelper to DockerComposeHelper in MlopsManager and ruff

Author: b-yogesh

src/gaiaflow/managers/minikube_manager.py

@@ -392,9 +392,7 @@ def create_kube_config_inline(self):
         self.kube_helper.create_inline()
 
     def build_docker_image(self):
-        dockerfile_path = (
-            self.gaiaflow_path / "_docker" / "user-package" / "Dockerfile"
-        )
+        dockerfile_path = self.gaiaflow_path / "_docker" / "user-package" / "Dockerfile"
         self.docker_helper.build_image(dockerfile_path)
 
     def create_secrets(self, secret_name: str, secret_data: dict[str, Any]):

src/gaiaflow/managers/mlops_manager.py

@@ -72,7 +72,7 @@ class DockerResources:
     }
 
 
-class DockerHelper:
+class DockerComposeHelper:
     def __init__(self, gaiaflow_path: Path, is_prod_local: bool):
         self.gaiaflow_path = gaiaflow_path
         self.is_prod_local = is_prod_local
@@ -224,7 +224,7 @@ def __init__(
         self.user_env_name = user_env_name
         self.env_tool = env_tool
 
-        self.docker = DockerHelper(gaiaflow_path, prod_local)
+        self.docker = DockerComposeHelper(gaiaflow_path, prod_local)
         self.jupyter = JupyterHelper(
             jupyter_port, env_tool, user_env_name, gaiaflow_path
         )
@@ -405,9 +405,7 @@ def _create_gaiaflow_context(self):
         package_dir = Path(__file__).parent.parent.resolve()
         docker_dir = package_dir.parent / "_docker"
 
-        shutil.copytree(
-            docker_dir, self.gaiaflow_path / "_docker", dirs_exist_ok=True
-        )
+        shutil.copytree(docker_dir, self.gaiaflow_path / "_docker", dirs_exist_ok=True)
         log_info(f"Gaiaflow context created at {self.gaiaflow_path}")
 
     def _collect_volumes(self, compose_data: dict) -> list[str]:
@@ -471,10 +469,7 @@ def _update_files(self):
         yaml.preserve_quotes = True
 
         compose_path = (
-            self.gaiaflow_path
-            / "_docker"
-            / "docker-compose"
-            / "docker-compose.yml"
+            self.gaiaflow_path / "_docker" / "docker-compose" / "docker-compose.yml"
         )
 
         with open(compose_path) as f:

tests/managers/test_mlops_manager.py

@@ -11,7 +11,7 @@
 from gaiaflow.managers.mlops_manager import (
     MlopsManager,
     JupyterHelper,
-    DockerHelper,
+    DockerComposeHelper,
     DockerResources,
 )
 
@@ -316,22 +316,22 @@ def test_start_jupyter_env_not_exists(self, mock_popen, mock_env):
         mock_popen.assert_not_called()
 
     def test_docker_helper_builds_command(self):
-        helper = DockerHelper(self.manager.gaiaflow_path, is_prod_local=False)
+        helper = DockerComposeHelper(self.manager.gaiaflow_path, is_prod_local=False)
         cmd = helper._base_cmd()
         self.assertIn("docker", cmd)
         self.assertIn("compose", cmd)
         self.assertEqual(cmd.count("-f"), 1)
 
     def test_docker_helper_builds_command_prod_local(self):
-        helper = DockerHelper(self.manager.gaiaflow_path, is_prod_local=True)
+        helper = DockerComposeHelper(self.manager.gaiaflow_path, is_prod_local=True)
         cmd = helper._base_cmd()
         self.assertIn("docker", cmd)
         self.assertIn("compose", cmd)
         self.assertEqual(cmd.count("-f"), 2)
 
 
     def test_docker_services_for_known_and_unknown(self):
-        helper = DockerHelper(Path("/tmp"), False)
+        helper = DockerComposeHelper(Path("/tmp"), False)
         self.assertIn("mlflow", helper.docker_services_for("mlflow"))
         self.assertEqual(helper.docker_services_for("unknown"), [])