Skip to content

Commit f92afd3

Browse files
style: apply pre-commit fixes (black, isort)
1 parent 756ebc6 commit f92afd3

File tree

16 files changed

+57
-50
lines changed

16 files changed

+57
-50
lines changed

cli/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
1-
import click
21
import subprocess
2+
3+
import click
4+
35
from llm_utils.tools import set_gms_server
46

57

data_utils/datahub_source.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,18 @@
1-
from datahub.metadata.schema_classes import DatasetPropertiesClass, SchemaMetadataClass
2-
from datahub.emitter.rest_emitter import DatahubRestEmitter
3-
from datahub.ingestion.graph.client import DatahubClientConfig, DataHubGraph
4-
from datahub.metadata.schema_classes import UpstreamLineageClass
51
from collections import defaultdict
2+
63
import requests
4+
from datahub.emitter.rest_emitter import DatahubRestEmitter
5+
from datahub.ingestion.graph.client import DatahubClientConfig, DataHubGraph
6+
from datahub.metadata.schema_classes import (
7+
DatasetPropertiesClass,
8+
SchemaMetadataClass,
9+
UpstreamLineageClass,
10+
)
11+
712
from data_utils.queries import (
8-
ROOT_GLOSSARY_NODES_QUERY,
913
GLOSSARY_NODE_QUERY,
1014
LIST_QUERIES_QUERY,
15+
ROOT_GLOSSARY_NODES_QUERY,
1116
)
1217

1318

evaluation/gen_answer.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,9 @@
1+
import uuid
12
from argparse import ArgumentParser
2-
from langchain_core.messages import HumanMessage
3-
4-
from utils import load_question_json, save_answer_json
53

4+
from langchain_core.messages import HumanMessage
65
from tqdm import tqdm
7-
import uuid
6+
from utils import load_question_json, save_answer_json
87

98
from llm_utils.graph import builder
109

evaluation/gen_persona.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
11
import os
2+
from argparse import ArgumentParser
23

3-
from utils import save_persona_json, pretty_print_persona
4+
from langchain_core.prompts import ChatPromptTemplate
5+
from langchain_openai.chat_models import ChatOpenAI
46
from persona_class import PersonaList
7+
from utils import pretty_print_persona, save_persona_json
58

69
from llm_utils.tools import _get_table_info
7-
from langchain_openai.chat_models import ChatOpenAI
8-
from langchain_core.prompts import ChatPromptTemplate
9-
from argparse import ArgumentParser
1010

1111

1212
def get_table_des_string(tables_desc):

evaluation/gen_question.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
1-
from utils import load_persona_json, save_question_json
1+
import os
2+
from argparse import ArgumentParser
3+
24
from langchain_core.prompts import ChatPromptTemplate
35
from langchain_openai.chat_models import ChatOpenAI
46
from tqdm import tqdm
5-
6-
from argparse import ArgumentParser
7-
import os
7+
from utils import load_persona_json, save_question_json
88

99

1010
def get_persona_prompt(persona):

evaluation/persona_class.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
1-
from pydantic import BaseModel
21
from typing import List
32

3+
from pydantic import BaseModel
4+
45

56
class Persona(BaseModel):
67
name: str

evaluation/utils.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
11
import json
2-
from persona_class import PersonaList
3-
from glob import glob
42
import os
3+
from glob import glob
4+
5+
from persona_class import PersonaList
56

67

78
def save_persona_json(data, filepath):

interface/lang2sql.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,15 @@
1-
import streamlit as st
2-
from langchain_core.messages import HumanMessage
3-
from llm_utils.graph import builder
4-
from langchain.chains.sql_database.prompt import SQL_PROMPTS
51
import os
62
from typing import Union
7-
import pandas as pd
83

4+
import pandas as pd
5+
import streamlit as st
96
from clickhouse_driver import Client
10-
from llm_utils.connect_db import ConnectDB
117
from dotenv import load_dotenv
8+
from langchain.chains.sql_database.prompt import SQL_PROMPTS
9+
from langchain_core.messages import HumanMessage
1210

11+
from llm_utils.connect_db import ConnectDB
12+
from llm_utils.graph import builder
1313

1414
# Clickhouse 연결
1515
db = ConnectDB()

interface/viz_eval.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
1-
import streamlit as st
2-
import json
31
import glob
4-
import pandas as pd
2+
import json
53
import os
64

5+
import pandas as pd
6+
import streamlit as st
77

88
st.set_page_config(layout="wide", page_title="Lang2SQL 평가 시각화")
99

llm_utils/chains.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,16 @@
11
import os
2+
3+
from dotenv import load_dotenv
24
from langchain_core.prompts import (
35
ChatPromptTemplate,
46
MessagesPlaceholder,
57
SystemMessagePromptTemplate,
68
)
79

8-
from .llm_factory import get_llm
9-
10-
from dotenv import load_dotenv
1110
from prompt.template_loader import get_prompt_template
1211

12+
from .llm_factory import get_llm
13+
1314
env_path = os.path.join(os.getcwd(), ".env")
1415

1516
if os.path.exists(env_path):

0 commit comments

Comments
 (0)