Checkpoint

databricks-academy · Nov 25, 2023 · 17f2cc8 · 17f2cc8
1 parent bef705b
commit 17f2cc8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/dbacademy/dbbuild/build_config.py b/src/dbacademy/dbbuild/build_config.py
@@ -104,7 +104,7 @@ def __init__(self,
                  *,
                  name: str,
                  version: str,
-                 supported_dbrs: List[str] = None,
+                 supported_dbrs: Union[str, List[str]] = None,
                  spark_version: str = None,
                  cloud: str = None,
                  instance_pool_id: str = None,
@@ -143,7 +143,8 @@ def __init__(self,
             self.__username = "[email protected]"  # When unit testing
         validate(username=self.__username).required.str()
 
-        self.__supported_dbrs = validate(supported_dbrs=supported_dbrs).list(str, auto_create=True)
+        supported_dbrs = supported_dbrs if isinstance(supported_dbrs, list) else [supported_dbrs]
+        self.__supported_dbrs = validate(supported_dbrs=supported_dbrs).required.list(str)
 
         self.__language_options = None
         self.__ignoring = validate(ignoring=ignoring).list(str, auto_create=True)