[DPE-6431] Reformat the charm.py

charmcraft.yaml

@@ -12,13 +12,13 @@ parts:
     source: .
     override-build: |
       # Ship the charm contents
-      # curl -sSL https://code.launchpad.net/~pguimaraes/openmessaging-benchmark/+git/openmessaging-benchmark/+artifact/280249/+files/openmessaging-benchmark-0.0.1-ubuntu0-20241119152607-linux-x64.tar.gz | tar -zxvf -
+      # curl -sSL https://code.launchpad.net/~pguimaraes/openmessaging-benchmark/+git/openmessaging-benchmark/+artifact/308300/+files/openmessaging-benchmark-0.0.1-ubuntu0-20250219140257-linux-x64.tar.gz | tar -zxvf -
 
       # For some reason, getting 23 write error in this curl command.
       # moving away to wget
       apt update
       apt install -y wget
-      wget https://code.launchpad.net/~pguimaraes/openmessaging-benchmark/+git/openmessaging-benchmark/+artifact/280249/+files/openmessaging-benchmark-0.0.1-ubuntu0-20241119152607-linux-x64.tar.gz
+      wget https://code.launchpad.net/~pguimaraes/openmessaging-benchmark/+git/openmessaging-benchmark/+artifact/308300/+files/openmessaging-benchmark-0.0.1-ubuntu0-20250219140257-linux-x64.tar.gz
       tar -zxvf openmessaging-benchmark-*.tar.gz
 
       mv openmessaging-benchmark-0.0.1-SNAPSHOT/ openmessaging-benchmark/

pyproject.toml

@@ -141,7 +141,7 @@ convention = "google"
 
 [tool.pyright]
 include = ["src"]
-extraPaths = ["./lib"]
+extraPaths = ["./lib", "src"]
 pythonVersion = "3.10"
 pythonPlatform = "All"
 typeCheckingMode = "basic"

src/benchmark/wrapper/main.py

@@ -7,10 +7,11 @@
 import logging
 import signal
 
-from core import WorkloadCLIArgsModel
-from process import WorkloadToProcessMapping
 from prometheus_client import start_http_server
 
+from benchmark.wrapper.core import WorkloadCLIArgsModel
+from benchmark.wrapper.process import WorkloadToProcessMapping
+
 
 class MainWrapper:
     """Main class to manage the benchmark tool."""

src/benchmark/wrapper/process.py

@@ -14,14 +14,15 @@
 import time
 from abc import ABC, abstractmethod
 
-from core import (
+from pydantic import BaseModel
+
+from benchmark.wrapper.core import (
     BenchmarkCommand,
     BenchmarkMetrics,
     ProcessModel,
     ProcessStatus,
     WorkloadCLIArgsModel,
 )
-from pydantic import BaseModel
 
 VALID_LOG_LEVELS = ["info", "debug", "warning", "error", "critical"]