Merge remote-tracking branch 'team/impl/sky_lauch' into impl/demo/100724

Author: chris-aeviator

src/pymc_server/cli.py

@@ -8,7 +8,7 @@
 from pymc_server.cli_factory import setup_launch_factory, setup_status_factory
 from sky.usage import usage_lib
 from sky.cli import _get_shell_complete_args, _get_click_major_version, _complete_cluster_name, _NaturalOrderGroup, _DocumentedCodeCommand
-from pymc_server.utils.cli_ex import jobs_launch as ex_launch
+
 
 
 from sky.cli import (

src/pymc_server/utils/yaml.py

@@ -125,7 +125,7 @@ def get_pymc_config_yaml(module_name, import_from="config", file_name="base.yaml
 
     # check that we have the config and support the module
     is_valid_module = file_exists and module_name in supported_modules
-    assert is_valid_module , f'Not Implemented: the only supported module are {supported_modules} but we may have config for additional modules: {list}'
+    assert is_valid_module , f'{colorama.Fore.RED}Not Implemented: {colorama.Style.RESET_ALL}the only supported module are {colorama.Fore.YELLOW}{supported_modules}{colorama.Style.RESET_ALL} but we may have config for additional modules: {colorama.Fore.YELLOW}{list}{colorama.Style.RESET_ALL}'
 
     return f'{base_path}/{import_from}/{module_name}/{file_name}'
 
@@ -155,15 +155,18 @@ def get_config_from_yaml(entrypoint: Tuple[str, ...],module_name:Optional[str],b
         exists(entrypoint[0])
     userYaml, isValid = _check_and_return_yaml(getUserYaml(entrypoint))
 
+    current_module = module_name if module_name is not None else get_module_name_from_yaml()
     if base_config_path is not None:
+        if(current_module is None):
+            raise Exception(f'{colorama.Fore.RED}Please define a module for: '
+                            f'{colorama.Fore.YELLOW}{base_config_path}{colorama.Style.RESET_ALL}')
         base_file =  f'{base_config_path}/{module_name}/base.yaml'
         exists(base_file)
         #customBaseYaml, isValid_ = _check_and_return_yaml(getUserYaml(base_file))
         module_config_path = base_file
 
 
     if module_config_path is None:
-        current_module = module_name if module_name is not None else get_module_name_from_yaml()
         base_file = get_pymc_config_yaml(current_module if current_module is not None else "pymc-marketing")
         exists(base_file)
         module_config_path = base_file