relocate dependencies

Signed-off-by: Kosaku Kimura <[email protected]>

Author: kimusaku

pyproject.toml

@@ -8,10 +8,13 @@ maintainers = [
     "Akira Ura <[email protected]>",
 ]
 license = "Apache-2.0"
+packages = [{ include = "sapientml" }]
 
 [tool.poetry.dependencies]
 python = ">=3.10,<3.13"
 sapientml = "*"
+jinja2 = "^3.1.2"
+scikit-learn = "1.1.3"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^7.4.0"
@@ -28,7 +31,7 @@ requires = ["poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"
 
 [tool.poetry.plugins."code_block_generator"]
-loaddata = "sapientml_loaddata:LoadData"
+loaddata = "sapientml:plugins:loaddata:LoadData"
 
 [tool.pysen]
 version = "0.10"

sapientml_loaddata/__init__.py renamed to sapientml/generators/loaddata/__init__.py

@@ -13,15 +13,12 @@
 # limitations under the License.
 
 import os
-import pandas as pd
 
 from jinja2 import Environment, FileSystemLoader
-from sapientml.params import Dataset, Code, Task
 from sapientml.generator import CodeBlockGenerator
+from sapientml.params import Code, Dataset, Task
 
-template_env = Environment(
-    loader=FileSystemLoader(f"{os.path.dirname(__file__)}/templates"), trim_blocks=True
-)
+template_env = Environment(loader=FileSystemLoader(f"{os.path.dirname(__file__)}/templates"), trim_blocks=True)
 ROW_THRESHOLD_FOR_SAMPLING = 100000
 
 
@@ -31,7 +28,6 @@ def _render(tpl, *args, **kwargs):
 
 
 class LoadData(CodeBlockGenerator):
-
     def generate_code(self, dataset: Dataset, task: Task):
         code = Code()
         code.validation = code.test = code.train = code.predict = "# *** GENERATED PIPELINE ***\n\n"