Update requirements-test.txt

nasa-nccs-hpda · Nov 21, 2024 · c3a9497 · c3a9497
1 parent 912fe75
commit c3a9497
Showing 1 changed file with 7 additions and 10 deletions.
diff --git a/requirements/requirements-test.txt b/requirements/requirements-test.txt
@@ -1,22 +1,14 @@
 torch>=2.0.0
 torchvision>=0.15
-pytorch-lightning
-omegaconf
+lightning
+datasets
 rasterio
 rioxarray
 xarray
 geopandas
-opencv-python
-opencv-python-headless
-opencv-contrib-python
-opencv-contrib-python-headless
 tifffile
 webcolors
-Pillow
-seaborn
-xgboost
 tiler
-segmentation-models
 pytest
 coveralls
 rtree
@@ -26,4 +18,9 @@ yacs
 termcolor
 numba
 segmentation-models-pytorch
+joblib
+GDAL>=3.3.0
 coverage
+deepspeed
+webdataset
+torchgeo