From d505d2ae6c4cde297f1c420c67b5a97cee6c4c29 Mon Sep 17 00:00:00 2001 From: Jamieson Warner Date: Thu, 10 Aug 2023 16:45:28 -0500 Subject: [PATCH] minor updates per review --- .../examples/predictors/conditional_lstm/train_predictor.py | 2 +- setup.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/covid_xprize/examples/predictors/conditional_lstm/train_predictor.py b/covid_xprize/examples/predictors/conditional_lstm/train_predictor.py index 8e56a62..dc4e734 100644 --- a/covid_xprize/examples/predictors/conditional_lstm/train_predictor.py +++ b/covid_xprize/examples/predictors/conditional_lstm/train_predictor.py @@ -66,7 +66,7 @@ def train_predictor(training_data: pd.DataFrame, :param context_column: Which column in the data df to use as context and outcome. :param arch: Which predictor architecture to use. Current options are 'conditional' and 'independent'. - :param return_all_trials: If set to True, then this function returns all trials as a list. + :param return_all_trials: If set to True, then this function returns all trials as a list of trained models. :param verbose: Verbosity level for model.fit() when training the predictor. :returns: (best_model, results_df) """ diff --git a/setup.py b/setup.py index 92a3b27..47d823f 100644 --- a/setup.py +++ b/setup.py @@ -7,7 +7,7 @@ import sys from setuptools import setup, find_packages -LIBRARY_VERSION = '2.0.2' +LIBRARY_VERSION = '2.0.3' CURRENT_PYTHON = sys.version_info[:2] REQUIRED_PYTHON = (3, 10) @@ -61,14 +61,14 @@ def read(fname): ] }, install_requires=[ - 'keras==2.11.0', 'neat-python==0.92', 'numpy==1.24.2', 'pandas==1.5.3', 'scikit-learn==1.2.2', 'scipy==1.10.1', 'setuptools==67.6.0', - 'tensorflow==2.11.1', + 'tensorflow==2.13.0', + 'keras==2.13.1', 'h5py==3.8.0' ], description='Contains sample code and notebooks '