diff --git a/conda/meta.yaml b/conda/meta.yaml index 771b4a3feed6194c6ccb1599ba52ed2f02f4504f..74ab9d7b21af02d6c2d1e74780fa9ba6d906d213 100644 --- a/conda/meta.yaml +++ b/conda/meta.yaml @@ -36,7 +36,7 @@ requirements: - torchvision {{ torchvision }} - tqdm {{ tqdm }} - tensorboard {{ tensorboard }} - - pytorch-lightning {{ pytorch_lightning }} + - lightning {{ lightning }} - clapper run: - python >=3.9 @@ -53,7 +53,7 @@ requirements: - {{ pin_compatible('torchvision') }} - {{ pin_compatible('tqdm') }} - {{ pin_compatible('tensorboard') }} - - {{ pin_compatible('pytorch-lightning') }} + - {{ pin_compatible('lightning') }} - clapper test: diff --git a/pyproject.toml b/pyproject.toml index 47820ed74c5103622503d734f73e199b8e8c38d3..342d68c30241dac247c5a2fd8d1a6674b3a1f843 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -41,7 +41,7 @@ dependencies = [ "pillow", "torch>=1.8", "torchvision>=0.10", - "pytorch-lightning", + "lightning", "tensorboard", ]