moving files

tests/metagpt/tools/functions/__init__.py

@@ -1,6 +0,0 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
-# @Time    : 2023/11/17 10:24
-# @Author  : lidanyang
-# @File    : __init__.py
-# @Desc    :

tests/metagpt/tools/libs/test_data_preprocess.py

@@ -5,7 +5,7 @@ import numpy.testing as npt
 import pandas as pd
 import pytest
 
-from metagpt.tools.functions.libs.data_preprocess import (
+from metagpt.tools.libs.data_preprocess import (
     FillMissingValue,
     LabelEncode,
     MaxAbsScale,

tests/metagpt/tools/libs/test_feature_engineering.py

@@ -3,7 +3,7 @@ import pandas as pd
 import pytest
 from sklearn.datasets import fetch_california_housing, load_breast_cancer, load_iris
 
-from metagpt.tools.functions.libs.feature_engineering import (
+from metagpt.tools.libs.feature_engineering import (
     CatCount,
     CatCross,
     ExtractTimeComps,
@@ -147,6 +147,7 @@ def test_general_selection(mock_dataset):
     assert "cat2" not in transformed.columns
 
 
+@pytest.mark.skip  # skip because TreeBasedSelection needs lgb as dependency
 def test_tree_based_selection(mock_dataset):
     # regression
     data = load_sklearn_data("housing")

tests/metagpt/tools/libs/test_sd.py

@@ -4,7 +4,7 @@
 # @Desc    :
 import pytest
 
-from metagpt.tools.sd_engine import SDEngine
+from metagpt.tools.libs.sd_engine import SDEngine
 
 
 def test_sd_tools():

tests/metagpt/tools/libs/test_udf.py

@@ -3,7 +3,7 @@ import json
 import yaml
 
 from metagpt.logs import logger
-from metagpt.tools.functions.libs.udf import UDFS, UDFS_YAML, docstring_to_yaml
+from metagpt.tools.libs.udf import UDFS, UDFS_YAML, docstring_to_yaml
 
 
 def test_udfs():