Fix cv import (#132)

* fix cv import issue

* remove debug message

* update packages

Author: suzhoum

pyproject.toml

@@ -36,24 +36,22 @@ classifiers = [
 ]
 dependencies = [
     "autogluon.common >=1, <2",
-    "aws-cdk-lib >=2.0.0,<2.179",
+    "aws-cdk-lib >=2.0.0,<2.190",
     "awscliv2 >=2.2,<2.4",
     "boto3 >=1.26.0,<2",
     "constructs >=10.0.0,<10.5",
-    "fsspec >=2023.5.0,<2025.3",
     "matplotlib >=3.4,<3.11",
     "pandas >=2.0.0,<3",
     "pyarrow >=19.0.0,<20",
     "pyyaml >=5.4,<7",
     "ray[default] >=2.6.3,<3",
-    "s3fs >=2023.5.0,<2025.3",
     "tqdm >=4.64.0,<=5",
     "typer >=0.9.0,<1.0.0",
     "wheel >0.38.0,<0.46",
 ]
 
 [project.optional-dependencies]
-tests = ["pytest", "pytest-mock", "tox", "black>=23.1.9, <=25.1.0", "isort>=5.11.0, <=6.0.0"]
+tests = ["pytest", "pytest-mock", "tox", "black>=23.1.9, <=25.1.0", "isort>=5.11.0, <=6.1.0"]
 
 [project.urls]
 "Homepage" = "https://github.com/autogluon/autogluon-bench"
@@ -97,7 +95,7 @@ legacy_tox_ini = """
     [testenv:lint]
     deps = 
         black>=23.1.9, <=25.1.0
-        isort>=5.11.0, <=6.0.0
+        isort>=5.11.0, <=6.1.0
     commands = 
         black --check --diff src/ tests/
         isort --check --diff src/ tests/

src/autogluon/bench/Dockerfile

@@ -10,7 +10,7 @@ RUN apt-get update && \
     apt-get install -y software-properties-common build-essential && \
     add-apt-repository ppa:deadsnakes/ppa && \
     apt-get update && \
-    apt-get install -y python3.9 python3.9-dev python3.9-distutils python3.9-venv && \
+    apt-get install -y python3.9 python3.9-dev python3.9-distutils python3.9-venv ffmpeg libsm6 libxext6 && \
     update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.9 1
 
 # Install utilities and AWS CLI

src/autogluon/bench/utils/hardware_utilization.sh

@@ -13,8 +13,6 @@ while true; do
       GPU_UTILIZATION=$(echo $line | awk -F ', ' '{printf "%.2f", $2}')
       aws cloudwatch put-metric-data --region $CDK_DEPLOY_REGION --namespace EC2 --storage-resolution 1 --unit Percent --dimensions InstanceId=$INSTANCE_ID --metric-name GPUUtilization_$GPU_INDEX --value $GPU_UTILIZATION
     done
-  else
-    echo "GPU not detected or nvidia-smi not found. Skipping GPU metrics."
   fi
 
   CPU_UTILIZATION=$(top -bn1 | grep "Cpu(s)" | awk -F'id,' '{ split($1, vs, ","); v=vs[length(vs)]; sub("%", "", v); printf "%.2f", 100 - v }')