Pass DeepSource

iterative · May 1, 2020 · b254d10 · b254d10
1 parent 194ed2d
commit b254d10
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/dvc/config.py b/dvc/config.py
@@ -219,7 +219,8 @@ class Config(dict):
     CONFIG = "config"
     CONFIG_LOCAL = "config.local"
 
-    def __init__(self, dvc_dir=None, validate=True):
+    def __init__(self, dvc_dir=None, validate=True, **kwargs):
+        super().__init__(**kwargs)
         self.dvc_dir = dvc_dir
 
         if not dvc_dir:
@@ -238,12 +239,14 @@ def __init__(self, dvc_dir=None, validate=True):
     def get_dir(cls, level):
         from appdirs import user_config_dir, site_config_dir
 
-        assert level in ("global", "system")
-
         if level == "global":
             return user_config_dir(cls.APPNAME, cls.APPAUTHOR)
-        if level == "system":
+        elif level == "system":
             return site_config_dir(cls.APPNAME, cls.APPAUTHOR)
+        else:
+            raise ConfigError(
+                "This method only used in global or system level."
+            )
 
     @cached_property
     def files(self):