modify: 优化包引用路径

loader/data/data_loader_mysql.py

@@ -49,7 +49,3 @@ class DataLoaderMysql(DataLoaderBase):
         cursor.close()
         self.close_connect()
         return df
-
-
-if __name__ == "__main__":
-    pass

metric/metric_by_sql_general.py

@@ -7,7 +7,7 @@
 import pandas as pd
 
 from loader import DataLoaderBase
-from metric.metric_base import MetricBase
+from .metric_base import MetricBase
 
 
 class MetricBySqlGeneral(MetricBase):
@@ -24,12 +24,3 @@ class MetricBySqlGeneral(MetricBase):
 
     def calculate(self, data_loader: DataLoaderBase, sql: str) -> pd.DataFrame:
         return self.load_data(data_loader, sql)
-
-
-if __name__ == "__main__":
-    from loader import DataLoaderMysql
-
-    data_loader = DataLoaderMysql(host="101.126.81.2", port=18001, user="root", passwd="Cqrcb2024", db="test")
-    metric_clzz = MetricBySqlGeneral(metric_name="auc", metric_code="auc")
-    metric = metric_clzz.calculate(data_loader, "select * from test.t1")
-    print(metric.head(5))

test.py

@@ -0,0 +1,18 @@
+# -*- coding: utf-8 -*-
+"""
+@author: yq
+@time: 2024/11/1
+@desc: 
+"""
+
+from metric import MetricBySqlGeneral
+
+
+if __name__ == "__main__":
+    from loader import DataLoaderMysql
+
+    data_loader = DataLoaderMysql(host="101.126.81.2", port=18001, user="root", passwd="Cqrcb2024", db="test")
+    metric_clzz = MetricBySqlGeneral(metric_name="auc", metric_code="auc")
+    metric = metric_clzz.calculate(data_loader, "select * from test.t1")
+    print(metric.head(5))
+