Update R_CMD_check_main_weekly.yaml

add pynvml
OHDSI · Jul 23, 2024 · f33b761 · f33b761
1 parent 64c4555
commit f33b761
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/.github/workflows/R_CMD_check_main_weekly.yaml b/.github/workflows/R_CMD_check_main_weekly.yaml
@@ -49,7 +49,6 @@ jobs:
       WEBAPI_TEST_SECURE_WEBAPI_URL: ${{ secrets.WEBAPI_TEST_SECURE_WEBAPI_URL }}
       WEBAPI_TEST_ADMIN_USER_NAME: ${{ secrets.WEBAPI_TEST_ADMIN_USER_NAME }}
       WEBAPI_TEST_ADMIN_USER_PASSWORD: ${{ secrets.WEBAPI_TEST_ADMIN_USER_PASSWORD }}
-      TORCH_INSTALL: '1'
 
     steps:
       - uses: actions/checkout@v3
@@ -75,7 +74,7 @@ jobs:
         shell: Rscript {0}
         run: |
           python_packages <- 
-            c("polars", "tqdm", "connectorx", "pyarrow", "numpy==1.26.4")
+            c("polars", "tqdm", "connectorx", "pyarrow", "pynvml", "numpy==1.26.4")
           
           library(reticulate)
           virtualenv_create("r-reticulate", Sys.which("python"), packages=python_packages)