formatting

defog/cli.py

@@ -283,7 +283,7 @@ def gen():
         table_name_list = re.split(r"\s+", table_names.strip())
     else:
         table_name_list = sys.argv[2:]
-    
+
     if table_name_list == [""] or table_name_list == []:
         print("No tables were registered. Exiting.")
         sys.exit(0)
@@ -295,7 +295,7 @@ def gen():
         scan = True
     else:
         scan = False
-    
+
     filename = df.generate_db_schema(table_name_list, scan=scan)
     pwd = os.getcwd()
     print(

defog/util.py

@@ -73,15 +73,17 @@ def identify_categorical_columns(
     # if it is a categorical variable, then we want to get the distinct values and their counts
     # we will then send this to the defog servers so that we can generate a column description
     # for each categorical variable
-    print(f"Identifying categorical columns in {table_name}. This might take a while if you have many rows in your table.")
+    print(
+        f"Identifying categorical columns in {table_name}. This might take a while if you have many rows in your table."
+    )
     for idx, row in enumerate(rows):
         if row["data_type"].lower() in [
             "character varying",
             "text",
             "character",
             "varchar",
             "char",
-            "string"
+            "string",
         ]:
             # get the total number of rows and number of distinct values in the table for this column
             cur.execute(