fix mamba

Kaggle · May 24, 2024 · cc40287 · cc40287
1 parent 86a15be
commit cc40287
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
@@ -112,7 +112,7 @@ RUN conda config --add channels nvidia && \
 # b/341938540: unistall grpc-cpp to allow >=v24.4 cudf and cuml to be installed.
 {{ if eq .Accelerator "gpu" }}
 RUN pip uninstall -y pyarrow && \
-    mamba remove grpc-cpp && \
+    mamba remove --force grpc-cpp && \
     mamba install -y -c conda-forge spacy cudf>=24.4 cuml>=24.4 cupy cuda-version=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION && \
     /tmp/clean-layer.sh
 {{ else }}

diff --git a/tests/common.py b/tests/common.py
@@ -7,7 +7,7 @@
 def getAcceleratorName():
     try:
         deviceName = subprocess.check_output(['nvidia-smi', '--query-gpu=name', '--format=csv,noheader'])
-        return deviceName.decode('utf-8').strip();
+        return deviceName.decode('utf-8').strip()
     except FileNotFoundError:
         return("nvidia-smi not found.")