Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix(ci) ONC-114: reduce the ci load by only installing lmdb in tests (#14581) 

reduce the ci load by only installing lmdb in tests
This commit is contained in:
Rick Izzo 2022-09-07 11:55:03 -04:00 committed by GitHub
parent d2459df2ff
commit 12d4b2a5f5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/tests_app_examples/custom_work_dependencies/app.py
View File

@ -10,17 +10,13 @@ class CustomBuildConfig(BuildConfig):
class WorkWithCustomDeps(LightningWork): class WorkWithCustomDeps(LightningWork):
def __init__(self, cloud_compute: CloudCompute = CloudCompute(), **kwargs): def __init__(self, cloud_compute: CloudCompute = CloudCompute(), **kwargs):
build_config = CustomBuildConfig(requirements=["numpy", "pandas", "py"]) build_config = CustomBuildConfig(requirements=["py"])
super().__init__(parallel=True, **kwargs, cloud_compute=cloud_compute, cloud_build_config=build_config) super().__init__(parallel=True, **kwargs, cloud_compute=cloud_compute, cloud_build_config=build_config)
def run(self): def run(self):
# installed by the build commands and by requirements in the build config # installed by the build commands and by requirements in the build config
import lmdb import lmdb
import numpy as np
import pandas as pd
print("installed numpy version:", np.__version__)
print("installed pandas version:", pd.__version__)
print("installed lmdb version:", lmdb.__version__) print("installed lmdb version:", lmdb.__version__)