From 320b2bfef861525491e35d042c7f19d33a9b3847 Mon Sep 17 00:00:00 2001
From: Ftps <ftpsflandre@gmail.com>
Date: Fri, 17 Nov 2023 18:24:37 +0900
Subject: [PATCH] Add type declarations

---
 rvc/configs/config.py | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/rvc/configs/config.py b/rvc/configs/config.py
index 73d0e38..ee3e43a 100644
--- a/rvc/configs/config.py
+++ b/rvc/configs/config.py
@@ -17,7 +17,7 @@ except (ImportError, Exception):
     pass
 import logging
 
-logger = logging.getLogger(__name__)
+logger: logging.Logger = logging.getLogger(__name__)
 
 
 version_config_list: list = [
@@ -35,13 +35,14 @@ class Config:
         return cls._instance
 
     def __init__(self):
-        self.device = "cuda:0"
-        self.is_half = True
-        self.use_jit = False
-        self.n_cpu = 0
-        self.gpu_name = None
+        self.device: str = "cuda:0"
+        self.is_half: bool = True
+        self.use_jit: bool = False
+        self.n_cpu: int = cpu_count()
+        self.gpu_name: str | None = None
         self.json_config = self.load_config_json()
-        self.gpu_mem = None
+        self.gpu_mem: int | None = None
+        self.instead: str | None = None
         (
             self.python_cmd,
             self.listen_port,
@@ -60,8 +61,7 @@ class Config:
 
     @staticmethod
     def arg_parse() -> tuple:
-        exe = sys.executable or "python"
-        parser = argparse.ArgumentParser()
+        parser: argparse.ArgumentParser = argparse.ArgumentParser()
         parser.add_argument("--port", type=int, default=7865, help="Listen port")
         parser.add_argument(
             "--pycmd",
@@ -82,7 +82,7 @@ class Config:
             action="store_true",
             help="torch_dml",
         )
-        cmd_opts = parser.parse_args()
+        cmd_opts: argparse.Namespace = parser.parse_args()
 
         cmd_opts.port = cmd_opts.port if 0 <= cmd_opts.port <= 65535 else 7865