fix return type in BaseTensorFlowModel

This commit is contained in:
robcaulk 2022-07-26 16:01:54 +02:00
parent ad25a4cb56
commit 3f149c4067

View File

@ -1,5 +1,5 @@
import logging import logging
from typing import Tuple from typing import Any
from pandas import DataFrame from pandas import DataFrame
@ -27,7 +27,7 @@ class BaseTensorFlowModel(IFreqaiModel):
def train( def train(
self, unfiltered_dataframe: DataFrame, pair: str, dk: FreqaiDataKitchen self, unfiltered_dataframe: DataFrame, pair: str, dk: FreqaiDataKitchen
) -> Tuple[DataFrame, DataFrame]: ) -> Any:
""" """
Filter the training data and train a model to it. Train makes heavy use of the datakitchen Filter the training data and train a model to it. Train makes heavy use of the datakitchen
for storing, saving, loading, and analyzing the data. for storing, saving, loading, and analyzing the data.