diff --git a/pyproject.toml b/pyproject.toml index ed99885..644c986 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -12,7 +12,8 @@ dependencies = [ "pyyaml>=6.0.2", "matplotlib>=3.9.2", "plotly>=5.23.0", - "deprecated>=1.2.14" + "deprecated>=1.2.14", + "scikit-learn>=1.5.1" ] requires-python = ">=3.10" authors = [ diff --git a/src/tensortrade/__init__.py b/src/tensortrade/__init__.py index 06ec4b1..413cb91 100644 --- a/src/tensortrade/__init__.py +++ b/src/tensortrade/__init__.py @@ -1,5 +1,19 @@ -from . import core -from . import feed +# Copyright 2024 The TensorTrade-NG Authors. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License +from tensortrade import core +from tensortrade import env +from tensortrade import feed from tensortrade.oms import ( orders, wallets, @@ -7,6 +21,6 @@ exchanges, services ) -from . import env +from tensortrade import pipeline -from .__about__ import __version__ +from tensortrade.__about__ import __version__ diff --git a/src/tensortrade/pipeline/__init__.py b/src/tensortrade/pipeline/__init__.py index 1b82aed..0f4bf72 100644 --- a/src/tensortrade/pipeline/__init__.py +++ b/src/tensortrade/pipeline/__init__.py @@ -11,4 +11,5 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License -from tensortrade.pipeline.pipeline import DataPipeline \ No newline at end of file +from tensortrade.pipeline import transformers +from tensortrade.pipeline.pipeline import DataPipeline