Skip to content

Commit

Permalink
Relax pytorch_lightning to '>=1.3.8,<1.5.10'
Browse files Browse the repository at this point in the history
Signed-off-by: Enrico Minack <github@enrico.minack.dev>
  • Loading branch information
EnricoMi committed Mar 22, 2022
1 parent bfc4694 commit 58bc4bb
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions setup.py
Expand Up @@ -162,7 +162,7 @@ def build_extensions(self):
tensorflow_gpu_require_list = ['tensorflow-gpu']
keras_require_list = ['keras>=2.0.8,!=2.0.9,!=2.1.0,!=2.1.1']
# pytorch-lightning 1.3.8 is a stable version to work with horovod
pytorch_require_list = ['torch', 'pytorch_lightning==1.3.8']
pytorch_require_list = ['torch', 'pytorch_lightning>=1.3.8,<1.5.10']
mxnet_require_list = ['mxnet>=1.4.1']
pyspark_require_list = ['pyspark>=2.3.2;python_version<"3.8"',
'pyspark>=3.0.0;python_version>="3.8"']
Expand All @@ -189,7 +189,7 @@ def build_extensions(self):
'keras==2.3.1',
'torch==1.4.0',
'torchvision==0.5.0',
'pytorch_lightning>=1.3.8',
'pytorch_lightning>=1.3.8,<1.5.10',
'mxnet==1.5.0',
'pyspark==3.0.1'] + spark_require_list
# torchvision 0.5.0 depends on torch==1.4.0
Expand Down

0 comments on commit 58bc4bb

Please sign in to comment.