Move config to be public facing

Author: XkunW

tests/test_imports.py

@@ -19,12 +19,12 @@ def test_imports(self):
             # Client imports
             import vec_inf.client
             import vec_inf.client._client_vars  # noqa: F401
-            import vec_inf.client._config
             import vec_inf.client._exceptions
             import vec_inf.client._helper
             import vec_inf.client._slurm_script_generator
             import vec_inf.client._utils
             import vec_inf.client.api
+            import vec_inf.client.config
             import vec_inf.client.models
             import vec_inf.client.slurm_vars  # noqa: F401
 

vec_inf/client/__init__.py

@@ -5,8 +5,8 @@
 users direct control over the lifecycle of inference servers via python scripts.
 """
 
-from vec_inf.client._config import ModelConfig
 from vec_inf.client.api import VecInfClient
+from vec_inf.client.config import ModelConfig
 from vec_inf.client.models import (
     LaunchOptions,
     LaunchOptionsDict,

vec_inf/client/_helper.py

@@ -20,14 +20,14 @@
     REQUIRED_FIELDS,
     SRC_DIR,
 )
-from vec_inf.client._config import ModelConfig
 from vec_inf.client._exceptions import (
     MissingRequiredFieldsError,
     ModelConfigurationError,
     ModelNotFoundError,
     SlurmJobError,
 )
 from vec_inf.client._slurm_script_generator import SlurmScriptGenerator
+from vec_inf.client.config import ModelConfig
 from vec_inf.client.models import (
     LaunchResponse,
     ModelInfo,

vec_inf/client/_utils.py

@@ -15,7 +15,7 @@
 import yaml
 
 from vec_inf.client._client_vars import MODEL_READY_SIGNATURE
-from vec_inf.client._config import ModelConfig
+from vec_inf.client.config import ModelConfig
 from vec_inf.client.models import ModelStatus
 from vec_inf.client.slurm_vars import CACHED_CONFIG
 

vec_inf/client/api.py

@@ -14,7 +14,6 @@
 import warnings
 from typing import Any, Optional, Union
 
-from vec_inf.client._config import ModelConfig
 from vec_inf.client._exceptions import (
     ServerError,
     SlurmJobError,
@@ -26,6 +25,7 @@
     PerformanceMetricsCollector,
 )
 from vec_inf.client._utils import run_bash_command
+from vec_inf.client.config import ModelConfig
 from vec_inf.client.models import (
     LaunchOptions,
     LaunchResponse,