Skip to content

Commit d505d2a

Browse files
minor updates per review
1 parent a1cbf47 commit d505d2a

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

covid_xprize/examples/predictors/conditional_lstm/train_predictor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ def train_predictor(training_data: pd.DataFrame,
6666
:param context_column: Which column in the data df to use as context and outcome.
6767
:param arch: Which predictor architecture to use.
6868
Current options are 'conditional' and 'independent'.
69-
:param return_all_trials: If set to True, then this function returns all trials as a list.
69+
:param return_all_trials: If set to True, then this function returns all trials as a list of trained models.
7070
:param verbose: Verbosity level for model.fit() when training the predictor.
7171
:returns: (best_model, results_df)
7272
"""

setup.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
import sys
88
from setuptools import setup, find_packages
99

10-
LIBRARY_VERSION = '2.0.2'
10+
LIBRARY_VERSION = '2.0.3'
1111

1212
CURRENT_PYTHON = sys.version_info[:2]
1313
REQUIRED_PYTHON = (3, 10)
@@ -61,14 +61,14 @@ def read(fname):
6161
]
6262
},
6363
install_requires=[
64-
'keras==2.11.0',
6564
'neat-python==0.92',
6665
'numpy==1.24.2',
6766
'pandas==1.5.3',
6867
'scikit-learn==1.2.2',
6968
'scipy==1.10.1',
7069
'setuptools==67.6.0',
71-
'tensorflow==2.11.1',
70+
'tensorflow==2.13.0',
71+
'keras==2.13.1',
7272
'h5py==3.8.0'
7373
],
7474
description='Contains sample code and notebooks '

0 commit comments

Comments
 (0)