fix databricks config

vadimooo · vadimooo · commit be1d59e6146b · 2024-04-25T14:01:33.000-07:00
diff --git a/cli/dataforge/databricks_sql.py b/cli/dataforge/databricks_sql.py
@@ -62,4 +62,4 @@ def validate(self):
         validate_value(self.config,'access_token')
         validate_value(self.config,'catalog')
         validate_value(self.config, 'schema')
-        self.execute("SELECT 1 as i;", mode='test')  # execute test query
+        self.execute("SELECT 1 as i", mode='test')  # execute test query
diff --git a/cli/dataforge/main.py b/cli/dataforge/main.py
@@ -1,5 +1,3 @@
-import sys
-
 from .importProject import ImportProject
 from .mainConfig import MainConfig
 
diff --git a/pyproject.toml b/pyproject.toml
@@ -16,7 +16,8 @@ dependencies = [
     "pyspark",
     "importlib_resources",
     "databricks-sql-connector",
-    "sql-formatter"
+    "sql-formatter",
+    "psutil"
     ]
 [tool.setuptools.packages.find]
 # All the following settings are optional:

Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,3 @@`
`1`		`-import sys`
`2`		`-`
`3`	`1`	`from .importProject import ImportProject`
`4`	`2`	`from .mainConfig import MainConfig`
`5`	`3`